AI Research Scientist · Optimization Algorithms & Schedules
Gradient descent optimization
Gradient descent optimization
Gradient descent update rule
Gradient descent updates a parameter by stepping opposite the gradient of the loss: Here is the learning rate controlling step size.How the learning rate shapes the trajectory
- 1Too small: steps are tiny, progress toward the minimum is slow but stable.
- 2Too large: steps overshoot, loss oscillates or diverges instead of shrinking.
- 3Well-tuned: steps shrink as the gradient flattens, converging steadily.
- Averages all 10,000 examples’ gradients each step: accurate but costly.
- Smooth convergence path since noise is minimal.
- Mini-batch of 32 examples costs far less per step but adds noise.
- Stochastic (batch=1) is cheapest yet noisiest, needing more steps to settle.
Recall check from the same lesson
For a training set with 50 million examples, computing the full-batch gradient every step is usually preferred over mini-batch or stochastic updates because it gives the most accurate gradient direction each step.
Review the explanation
Answer: False. Full-batch gradients are accurate, but the compute cost per step scales with dataset size, making it impractically slow at 50 million examples. Mini-batch or stochastic updates trade some gradient noise for much cheaper, more frequent steps, which is why they're preferred at that scale.
Sources
One sitting · 20–30 minutes
A focused session on your AI Research Scientist interview
LearnBench starts from what you already know — skip what you have, master what you’re missing.
Start now