about
Surprising Effectiveness of Masking Updates in Adaptive Optimizers (arxiv.org)
2 points by energy123 224 days ago | hide | past | pdf | discuss on HN

In plain words: Instead of updating every parameter each step, this trick randomly zeroes out some updates and favors those pointing the same way as recent momentum. For a 1-billion-parameter model, it cut perplexity, a prediction-error score, by over 19% versus Adam, the standard optimizer.

Abstract · On Surprising Effectiveness of Masking Updates in Adaptive Optimizers

Training large language models (LLMs) relies almost exclusively on dense adaptive optimizers with increasingly sophisticated preconditioners. We challenge this by showing that randomly masking parameter updates can be highly effective, with a masked variant of RMSProp consistently outperforming recent state-of-the-art optimizers. Our analysis reveals that the random masking induces a curvature-dependent geometric regularization that smooths the optimization trajectory. Motivated by this finding, we introduce Momentum-aligned gradient masking (Magma), which modulates the masked updates using momentum-gradient alignment. Extensive LLM pre-training experiments show that Magma is a simple drop-in replacement for adaptive optimizers with consistent gains and negligible computational overhead. Notably, for the 1B model size, Magma reduces perplexity by over 19\% and 9\% compared to Adam and Muon, respectively.

Taejong Joo, Wenhan Xia, Cheolmin Kim, Ming Zhang, Eugene Ie
arXiv:2602.15322 · cs.LG, cs.AI · submitted Feb 17, 2026
abstract · pdf · html · Preprint

add comment on HN
Also discussed: Feb 2026 (5 points, 0 comments)