Voltar às notícias
InovaçãoInstruções AI Understanding

DeltaMomentum paper proposes direction-aware optimizer updates for neural-network training

An arXiv preprint introduces DeltaMomentum, an optimizer update designed to forget frequently and rarely seen gradient directions at different rates, reporting faster training across language, image and vision benchmarks.

Por 5 min read
Unbranded GPU servers and cooling hardware in an academic machine-learning lab equipment room
A versão curta

An arXiv preprint introduces DeltaMomentum, an optimizer update designed to forget frequently and rarely seen gradient directions at different rates, reporting faster training across language, image and vision benchmarks.

O que aconteceu

An arXiv preprint by Euijin Hong and Guannan Qu proposes DeltaMomentum, a key-value-based momentum update for deep-network training. The method replaces the usual exponential moving average with a delta-rule update that adjusts forgetting according to how often an input direction appears.

The paper starts from a limitation of common momentum methods. They typically maintain an exponential moving average of past gradients and apply one forgetting rate to every direction. The authors argue that this is poorly matched to anisotropic training data, where some directions are encountered frequently while others are rare. Their proposed method makes the update itself direction-aware rather than adding processing around an unchanged momentum buffer.

DeltaMomentum treats the gradient of a linear layer as having a key-value structure. In the paper's description, the input acts as the key and an output-side error acts as the value. The momentum buffer is then updated with the canonical delta rule, so the effective forgetting rate depends on how often a direction appears. The authors say this gives the method an input-side curvature correction without requiring matrix inversion.

The paper presents DeltaMomentum as a drop-in replacement for the momentum buffer in any optimizer. It also claims that the coefficient transfers across model widths under the muP framework and that the method requires no persistent memory. The source estimates the additional computation at between 22.2% and 25.0% of the linear cost of a gated-MLP block. That is an overhead estimate from the paper, not an independently verified systems measurement.

The strongest results summarized on the source page concern FineWeb-Edu pretraining. AdamW using DeltaMomentum, called DeltaAdamW, reportedly reached AdamW's validation loss in up to 46.39 ± 4.32% fewer steps at 67 million parameters and 22.12 ± 0.80% fewer steps at 370 million parameters, measured over three seeds. The source says the gain persisted at 1 billion parameters on a Chinchilla-optimal budget. It also reports that a Muon baseline tuned under the same protocol remained above DeltaAdamW at both language-model scales, with additional gains reported for SGD, ResNet-18 and ViT-Tiny on CIFAR-10.

Leia a fonte primária: arxiv.org

Por que isso importa

If the reported results hold up, DeltaMomentum could reduce the number of training steps required to reach a given validation loss without requiring persistent memory or a wholesale replacement of existing optimizers. That could affect the cost and efficiency of training larger AI systems, although the evidence currently comes from one version of an arXiv preprint.

Training efficiency is a central constraint in modern AI development. Reaching the same validation loss in fewer optimization steps could reduce the amount of computation, time and energy needed for some training runs. The paper's most consequential claim is therefore not that it introduces another optimizer variant, but that changing the momentum update can materially shorten training across multiple scales and model families.

The reported results are potentially important because they span language-model pretraining and image-classification settings. The source describes experiments at 67 million, 370 million and 1 billion parameters, as well as tests involving SGD, ResNet-18, ViT-Tiny and CIFAR-10. That range suggests the authors are testing whether the proposed mechanism is more general than a single benchmark-specific adjustment. It does not, however, establish that the method will transfer to every architecture, objective or data distribution.

The paper also offers a mechanistic explanation for its results. Its diagnostics reportedly show better gradient tracking and healthier input directions, which the authors present as evidence for the predicted direction-aware behavior. If those diagnostics withstand replication, they could help researchers understand why a particular optimizer works, rather than relying only on end-point loss comparisons. The source page provides the claim but not the underlying plots, tables or statistical analysis beyond the figures summarized in the abstract.

There are practical tradeoffs. DeltaMomentum is described as a drop-in update, but the paper still reports additional compute of roughly one-quarter of a gated-MLP block's linear cost. Whether fewer optimization steps compensate for that overhead will depend on hardware, implementation and workload. The source does not provide total wall-clock savings, energy measurements, monetary costs, training throughput, code availability or evidence of use in deployed systems.

O que assistir a seguir

The main questions are whether independent researchers can reproduce the reported gains, how the method behaves across more architectures and data mixtures, and whether its extra computation offsets some of the claimed savings. The source does not establish peer review, production adoption, public software availability or results beyond the experiments it summarizes.

Replication should be the first test. The headline step reductions are reported by the paper's authors over three seeds at the smaller language-model scales, and the source says the effect persists at 1 billion parameters. Independent runs would clarify how sensitive the result is to random seeds, hyperparameter selection, data ordering, model width and the exact training budget. The source does not say whether the experiments or implementation have been independently reproduced.

Researchers should also compare total cost rather than steps alone. DeltaMomentum adds computation to each update, so the relevant measure is elapsed time, hardware utilization or energy to reach a specified validation loss. The abstract gives a relative extra-compute estimate but no end-to-end training-cost result. It is therefore not possible from this source to conclude that the method lowers real-world operating costs in every setting.

Broader evaluation will matter. The source mentions FineWeb-Edu, CIFAR-10, language-model scales, ResNet-18 and ViT-Tiny, but it does not describe results for other training objectives, modalities, optimizers, architectures or data regimes. It also does not establish behavior under distributed training, noisy or rapidly changing data, very large production models, or workloads where the input-direction distribution differs substantially from the tested settings.

Finally, readers should distinguish a research proposal from an available product. The source identifies the work as an arXiv version 1 submission dated August 19, 2026. It does not state that the paper has undergone peer review, that an implementation has been released, that major AI developers have adopted the method, or that the reported gains have been validated outside the authors' experiments. Those are open questions for follow-up coverage.

Guias e questionários relacionados

Modelos de IA explicadosTreinamento de IATransformadoresFuturo da IATeste o que você sabe – experimente um teste gratuito de IAProcure um termo de IA em nosso glossário
Achou isso útil?