DeltaMomentum: A Key-Value based Anisotropic Momentum Update via Delta Rule

arXiv:2608.19491 2026 Optimization 1 ideas extracted · analyzed Sep 1, 2026

What the math gives to ML

The paper identifies a structural mismatch in conventional momentum: for a linear layer, each per-example gradient is a key-value outer product, but EMA treats all input directions as if they had identical visitation frequency. Its transferable asset is the delta-rule interpretation of the momentum matrix as an associative memory, allowing the update to correct only the key direction currently being queried rather than globally decaying every direction. This suggests a drop-in optimizer update for linear, convolutional, attention-projection, and MLP weights that tracks drifting gradients more accurately without storing additional history. The most direct test is to replace only the momentum-buffer update in SGD or AdamW and compare loss-per-step, stale-direction recovery, and wall-clock overhead against EMA momentum.

Ideas from this paper

Failed on benchmark 2026

Key-Selective Delta Momentum

Replace the global EMA update for each linear-layer momentum matrix with a delta-rule update that learns the current output-side gradient value only along the current input-key direction. Frequently occurring directions are corrected repeatedly, while rarely visited directions are not unnecessarily overwritten or uniformly decayed. Use the resulting matrix as the ordinary momentum buffer in SGD, AdamW, or another optimizer.

Useful8/10
Difficulty5/10
Novelty7/10
Paper: DeltaMomentum: A Key-Value based Anisotropic Momentum Update via Delta Rule arXiv:2608.19491