The Moment That Speeds Up Neural Network Training express

Original: "Critical Damping as a Momentum Schedule: Multi-Seed Validation, a Hybrid Recipe, and an Exhaustive Negative Result on Surgical Layer Selection"
· Ivan Pasichnyk
arXiv:2603.28921 · 2026-03-30 · CC BY 4.0 · 2 min
A properly tuned momentum in the optimizer speeds up neural network training by 2.4 times without losing accuracy.

📄 Showing the "Simple" version — "Popular" is not ready yet. Add it to favorites to help prioritize it.

Training a neural network is like sledding down a hill: you need the right speed to avoid getting stuck or flying off the track. In the popular momentum method (like inertia), a constant value of 0.9 is typically used. The authors revisited the physics of damped oscillations and proposed changing momentum during training: initially large, then decreasing according to a formula where momentum equals one minus two times the square root of the learning rate.

The formula is derived from the condition that the system is "critically damped"—like a shock absorber that stops oscillations without excessive bouncing.

They tested it on a standard task: a ResNet-18 neural network learning to recognize images from the CIFAR-10 dataset. With the new momentum schedule, the model reaches 90% accuracy 2.34 times faster than with constant 0.9. However, the final accuracy is slightly lower — by 0.46 percentage points. So they applied a hybrid: critical damping until 90%, then switching to constant momentum. This eliminates the accuracy loss and retains the speedup: final accuracy is 95.45%, and it reaches 90% accuracy 2.4 times faster.

The speedup also works for other architectures: VGG-16 is 1.72 times faster, and on the more complex CIFAR-100 dataset the gain is modest but still present.

In the second part of the paper, the authors tested the idea of selecting layers for fine-tuning based on error gradients. It turned out that gradient-based selection is no better than random. But there is a useful signal: if a layer has the lowest gradient, it's definitely not worth touching — interfering with it can be very harmful. So gradients help avoid bad decisions, not find good ones.

Momentum in optimizers is named after the physical concept of moment of inertia: it helps overcome local minima, just as a sled glides over small bumps thanks to its accumulated speed.

\mu(t) = 1 - 2\sqrt{\alpha(t)}
Here μ(t) is the momentum at step t, and α(t) is the learning rate at the same step. The formula specifies how momentum should change during training to keep the system critically damped.
Concepts
learning rate xgboost