Backward-State Policy Is Part of the Learning Algorithm
Not on the map. Why?
Abstract (arXiv)
Low-precision training rounds tensors that the backward pass reads again, often for several gradients; each use can read the forward's rounded value, the original, or a new random rounding. This backward-state policy looks like a memory and precision detail, settled by copy accuracy and final loss. We argue that it is part of the learning algorithm, and that neither check shows whether it is right. Copy accuracy does not decide the outcome: in three pairs of 390M runs with an emulated FP8 backward, training fails when attention's backward reuses the forward's rounded output and succeeds with a new rounding from the same distribution. Even the most accurate copy, the original itself, can be wrong by our reference: the gradient of the forward pass as it actually ran, with gradients passed through rounding unchanged. For example, a normalization output stored in low precision feeds two gradients: the gain's gradient needs the original, but the next layer's weight gradient needs the rounded value that layer multiplied. Final loss, the other check, does not rule out the error of reading the original for both: it persists in models trained with such a store, while planned loss comparisons stay within a margin fixed in advance. We therefore derive from this reference which value each use must read, or which substitute gives the same gradient on average with the forward held fixed, and check these per-use requirements on single operators, without training. In three tests using PyTorch and Transformer Engine, the requirements predicted beforehand whether reuse changes what the backward computes on average relative to an independent copy, and every prediction held. Backward-state policy is thus part of the learning algorithm: it should be specified and checked use by use, not settled by copy accuracy and final loss.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 39 of its 52 references that are in the map. First posted on arXiv 30 Sep 2026.
Builds on
39 of 52 references.
References · 39
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,708
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
- 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,031
- 2020GLU Variants Improve Transformercited by 1,724
- 2019Root Mean Square Layer Normalizationcited by 1,425
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,311
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,104
- 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
- 2015Deep Learning with Limited Numerical Precisioncited by 368
- 2019Understanding Straight-Through Estimator in Training Activation Quantized Neural Netscited by 225
27 more
- 2023Small-scale proxies for large-scale Transformer training instabilitiescited by 182
- 2024Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durationscited by 160
- 2018Flipout: Efficient Pseudo-Independent Weight Perturbations on Mini-Batchescited by 133
- 2018The Mechanics of n-Player Differentiable Gamescited by 118
- 2024Scaling Laws for Precisioncited by 92
- 2023FP8-LM: Training FP8 Large Language Modelscited by 75
- 2023Stable and low-precision training for large-scale vision-language modelscited by 74
- 2024Scaling FP8 training to trillion-token LLMscited by 63
- 2025SageAttention3: Microscaling FP4 Attention for Inference and An Exploration of 8-Bit Trainingcited by 63
- 2019Inverse Path Tracing for Joint Material and Lighting Estimationcited by 62
- 2021ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Trainingcited by 61
- 2025Quartet: Native FP4 Training Can Be Optimal for Large Language Modelscited by 37
- 2025FP4 All the Way: Fully Quantized Training of LLMscited by 37
- 2024COAT: Compressing Optimizer states and Activation for Memory-Efficient FP8 Trainingcited by 32
- 2022GACT: Activation Compressed Training for Generic Network Architecturescited by 31
- 2025Recipes for Pre-training LLMs with MXFP8cited by 18
- 2023Unit Scaling: Out-of-the-Box Low-Precision Trainingcited by 17
- 2026Quartet II: Accurate LLM Pre-Training in NVFP4 by Improved Unbiased Gradient Estimationcited by 16
- 2019Credit Assignment Techniques in Stochastic Computation Graphscited by 14
- 2025Why Low-Precision Transformer Training Fails: An Analysis on Flash Attentioncited by 11
- 2024Is Flash Attention Stable?cited by 10
- 2026SageBwd: A Trainable Low-bit Attentioncited by 7
- 2026Stable FP4 Training via Transposition-Invariant Block Quantizationcited by 3
- 2025Spectral Alignment as Predictor of Loss Explosion in Neural Network Trainingcited by 2
- 2026Hardware-Aware FP4 FlashAttention-4cited by 2
- 2026Understanding Quantization-Aware Training: Gradients at Quantized Weights Bias to the Low-Loss Basincited by 1
- 2026Explicit State and Resource Contracts for Low-Precision Pipeline Parallel Training under Captured Graphscited by 1
Built on by
Not cited yet.