mlsanityData to 14 Oct 2026

Backpropagated Output Momentum: Relocating Optimizer History from Parameters to Task Space

arXiv 2609.36738 ↗First posted 29 Sep 2026cs.LG cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Optimizer momentum is usually stored as a parameter-sized moving average of past gradients, which makes history costly and fixes each past signal in the coordinates in which it was computed. We introduce Backpropagated Output Momentum (BOM), which instead stores a compact moving average of prediction errors at the model output and reprojects that history through the current network at every step. A batch-level analysis characterizes the information retained and omitted by this relocation, while the implementation preserves the current supervised gradient and can replace the first-moment component of several adaptive optimizers. As a plug-in for momentum-based optimizers, including ones that already compress their state, BOM reduces parameter-shaped optimizer state by 49.7-99.8% in three compositions and, averaged over three language backbones, paired step time by 4.0%. It also improves mean validation performance across language and vision fine-tuning, by 1.42 points in the primary five-task comparison. Language and vision pretraining studies, together with matched mechanism controls, further test the construction across output spaces and model scales.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 7 of its 31 references that are in the map. First posted on arXiv 29 Sep 2026.

Builds on

7 of 31 references.

References · 7

  1. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
  2. 2025Qwen3 Technical Reportcited by 8,820
  3. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
  4. 2018Riemannian Adaptive Optimization Methodscited by 194
  5. 2024Adam-mini: Use Fewer Learning Rates To Gain Morecited by 137
  6. 2024A Kaczmarz-inspired approach to accelerate the optimization of neural network wavefunctionscited by 26
  7. 2026M+Adam: Low-Precision Training via Additive-Multiplicative Optimizationcited by 1

Built on by

Not cited yet.