AI/ optimizers · deep learning · muon · model training

A New Optimizer Learns When to Act Like Muon or Adam

A revised paper describes an optimizer that decides per layer whether to act like SGD, Muon, or Adam, matching or beating either across three tests.

A new training method lets a neural network's optimizer pick its own math on the fly, instead of locking in one fixed update rule for the whole job.

Researchers published an updated paper describing an adaptive optimizer that chooses, layer by layer, how aggressively to reshape weight updates - landing somewhere between plain gradient descent (SGD) and the newer Muon method, or recovering Adam-style updates when that fits better. The system reads gradient and activation statistics during training and runs a quick regression-based estimate to pick the best update "geometry" for each layer, rather than relying on one geometry chosen in advance by hand. Because recalculating this choice at every step could slow training down, the authors added computational shortcuts that keep the extra cost to roughly 3% over already-optimized runs. Across three different training setups, the adaptive approach matched or beat whichever of Muon or AdamW performed best on its own.

Optimizer choice has become a real arms race since Muon showed that matrix-aware updates can out-train Adam on some models; this result suggests the next gain may not be another single fixed-rule optimizer but one that adapts its rule per layer, per run. That matters for anyone training large models, since it cuts into the guesswork and expensive hyperparameter sweeps that go into picking between SGD, Adam, and Muon variants in the first place.

Whether a 3% overhead is worth paying depends on the model and the budget, and - as with most optimizer papers - the real test is whether labs training frontier-scale systems bother adopting it outside a benchmark table.

TR

The Revision

Written by an AI system from the public sources credited above. How we write →