3.2.6 · D5Training Deep Networks

Question bank — Learning rate scheduling

2,375 words11 min readBack to topic

A few symbols recur in every trap below. Fix them in your head before you start — that is half the battle:


True or false — justify

Constant learning rate can converge exactly to the true minimum if you just train long enough.
False — the steady-state variance is fixed by , not by time; more steps just keep jittering inside the same band. Only shrinks the band to zero.
A schedule must be monotonically decreasing to be valid.
False — warmup ramps up for the first steps, and warm restarts jump it back up periodically. Decay is the typical late-phase behaviour, not a universal rule. See Warm Restarts (SGDR).
Cosine annealing decays linearly over training.
False — cosine is flat near and (its slope there) and drops fastest in the middle; only the middle looks straight. That is exactly why it settles gently at the end.
Both Robbins–Monro conditions are about making the noise vanish.
False — is about being able to travel arbitrarily far to reach the minimum; only is about the noise dying out. You need both, and they pull in opposite directions.
The schedule satisfies both Robbins–Monro conditions.
False — ✓ but ✗, so the squared-sum condition fails; the noise never provably dies. The exact schedule is , the form.
Warmup and decay do the same job, so you only need one of them.
False — warmup protects the unstable early phase (random weights, tiny/unreliable Adam statistics); decay handles the late fine-tuning phase. They act at opposite ends of training, so modern schedules use both.
A larger constant learning rate always trains faster overall.
False — it travels faster early but leaves a larger noise floor , so final loss is worse; you trade speed for a worse plateau. The point of scheduling is to get speed early and precision late.
Halving the learning rate halves the residual jitter variance.
True (for small ) — since is linear in , halving halves . This is the numerical backbone of "smaller LR = smaller settling jitter."

Spot the error

"For a Transformer with Adam, I'll decay from step 0 — LR should only ever go down."
The error: at step 0 Adam's second-moment estimate , so effective steps blow up and the loss diverges. Decay theory assumes stable gradients; you need warmup first. See Adam and Adaptive Optimizers.
"My step-decay schedule drops every 30 epochs, but I wrote where is counted in steps."
The error: mixing units. If an epoch is 500 steps, " steps" makes collapse ~500× too fast. Always confirm whether (and hence ) is in steps or epochs.
"Cosine annealing with means the model stops learning at the end, which wastes the last epochs."
The error framing: the near-zero LR at the end is deliberate fine-tuning — it lets the parameters settle into the bottom of the basin rather than jitter. The last epochs polish the minimum, they don't waste time. See Loss Landscapes and Minima.
"I doubled the batch size but kept the same schedule; behaviour should be identical."
The error: larger batches have less gradient noise (smaller ), so the effective good learning rate scales up (roughly linearly). Keeping the old under-utilizes the cleaner gradient. See Batch Size and Learning Rate scaling.
"Warm restarts prove that decay theory is wrong, since they raise again."
The error: warm restarts don't violate convergence — each restart is a fresh decay phase whose purpose is to jump the ball out of one basin and let it settle into a possibly better one. It's an exploration strategy layered on top of decay, not a contradiction of it.
"Since satisfies Robbins–Monro, it must be the best practical schedule."
The error: it's the theoretically sufficient schedule, but it decays too aggressively early — LR shrinks before the model has traveled far, so training stalls. Practice prefers gentler shapes (cosine, step) that keep high longer.

Why questions

Why does the noise floor variance depend on but not on how long you train?
Because is a steady state of : each step's decay of old jitter, , exactly balances new injected noise , fixing regardless of step count. Time can't help once you're at equilibrium.
Why choose the cosine shape rather than a straight line from to ?
Its flat-high start lets the model travel fast while the landscape is still far from the minimum, and its flat-low end gives gentle settling — a linear ramp does neither well, cutting speed early and slamming the LR down abruptly late.
Why does exponential decay have a "half-life" and step decay does not?
Exponential decay multiplies by a fixed factor every step continuously, so like radioactive decay it has a constant half-life set by the decay rate . Step decay holds flat then drops by every epochs, so there's no smooth constant-ratio halving time.
Why can a sudden step-decay drop () reach a lower loss than smooth decay sometimes?
Training at a fixed LR lets SGD explore a basin until progress stalls; the abrupt drop then quenches the jitter and lets the model fall into a lower sub-basin it was previously bouncing over. See Stochastic Gradient Descent.
Why is warmup "essential for Transformers" specifically, and less critical for a small CNN with plain SGD?
Transformers rely on Adam, whose early second-moment estimate is unreliable and makes effective steps explode; plain SGD has no such adaptive denominator, so its early steps are already bounded and warmup matters less.
Why do the two Robbins–Monro conditions pull against each other?
wants LRs large/persistent enough to travel forever, while wants them shrinking fast enough that squared steps sum finitely — the schedule must decay slowly enough for the first yet fast enough for the second. See Robbins-Monro Stochastic Approximation.

Edge cases

What happens with cosine annealing at exactly and ?
At , gives ; at , gives . The endpoints are exact plateaus where the slope is zero, so barely moves right at the boundaries.
What does do to cosine annealing versus a small positive ?
With the model fully stops updating at (pure fine-tune-then-freeze); a small positive keeps a floor of exploration, useful when you plan warm restarts or continued training.
What happens to step decay when ?
It degenerates into a constant learning rate — every "drop" multiplies by 1, so nothing changes. You're back to the constant- noise floor with no scheduling benefit.
What happens at the handoff step where warmup ends and decay begins?
If designed correctly, both pieces equal at , so the curve is continuous — a mismatch there causes a visible LR jump that can destabilize training right at the transition.
What if you set the warmup length to zero?
There is no ramp — you start at full immediately, reintroducing the exact early-instability (exploding Adam steps) that warmup was meant to prevent. It's equivalent to "no warmup."
As under decay , does reach zero?
It approaches zero but never equals it — . That's exactly what Robbins–Monro wants: infinitely small yet with divergent sum, so the model can still creep the last distance to the minimum.
What is the effect of choosing so large that in the quadratic-bowl model?
The factor flips sign, so each step overshoots to the other side of the minimum; if the oscillations grow and training diverges. This is the "ball bouncing across the valley, never settling" picture made precise. See Hyperparameter Tuning.