mlsanityData to 7 Oct 2026

Relative Generalization Invariance of LLM Pretraining

arXiv 2609.33016 ↗First posted 26 Sep 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Large Language Model (LLM) pretraining performance is jointly shaped by three components of the training triplet: the optimizer, model architecture, and training data stream. However, how these components influence performance in distinct ways remains unclear. We take a first step toward isolating their effects by studying relative generalization. We introduce Relative Generalization Invariance (RGI), the invariance of the validation-loss difference between any two tokens across models. We show that RGI approximately holds across a wide range of optimizers and moderate architectural variations, suggesting that these choices induce an approximately uniform shift in token-wise losses. In contrast, changing the training data stream can substantially alter relative generalization. We further show that RGI cannot be explained by the neural tangent kernel or mean-field regimes alone and prove that it can emerge in an overparameterized quadratic model. Overall, our work identifies RGI as a new phenomenon in LLM pretraining that helps distinguish the effects of optimizers and architectures from those of training data.

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

Builds on

85 of 103 references.

References · 85

  1. 2014Adam: A Method for Stochastic Optimizationcited by 36,960
  2. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  3. 2020Scaling Laws for Neural Language Modelscited by 6,411
  4. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
  5. 2016Gaussian Error Linear Units (GELUs)cited by 3,431
  6. 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
  7. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
  8. 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,085
  9. 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 948
  10. 2017Sigmoid-Weighted Linear Units for Neural Network Function Approximation in Reinforcement Learningcited by 935
  11. 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 896
  12. 2019On the Convergence of Adam and Beyondcited by 834
38 more
  1. 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 732
  2. 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
  3. 2018On Lazy Training in Differentiable Programmingcited by 671
  4. 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 655
  5. 2019On Exact Computation with an Infinitely Wide Neural Netcited by 572
  6. 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 562
  7. 2018A Mean Field View of the Landscape of Two-Layers Neural Networkscited by 463
  8. 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
  9. 2023Symbolic Discovery of Optimization Algorithmscited by 423
  10. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
  11. 2023DoReMi: Optimizing Data Mixtures Speeds Up Language Model Pretrainingcited by 385
  12. 2018Learning and Generalization in Overparameterized Neural Networks, Going Beyond Two Layerscited by 340
  13. 2023The Impact of Positional Encoding on Length Generalization in Transformerscited by 319
  14. 2017Stochastic Gradient Descent as Approximate Bayesian Inferencecited by 298
  15. 2017How to Escape Saddle Points Efficientlycited by 284
  16. 2020AdaBelief Optimizer: Adapting Stepsizes by the Belief in Observed Gradientscited by 282
  17. 2021Scaling Laws for Transfercited by 265
  18. 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 260
  19. 2019A Constructive Prediction of the Generalization Error Across Scalescited by 202
  20. 2019Mean-field theory of two-layers neural networks: dimension-free bounds and kernel limitcited by 187
  21. 2023Efficient Large Language Models: A Surveycited by 183
  22. 2024Data Mixing Laws: Optimizing Data Mixtures by Predicting Language Modeling Performancecited by 179
  23. 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 179
  24. 2024RedPajama: an Open Dataset for Training Large Language Modelscited by 178
  25. 2025Training Deep Learning Models with Norm-Constrained LMOscited by 177
  26. 2024RegMix: Data Mixture as Regression for Language Model Pre-trainingcited by 172
  27. 2019Neural Tangents: Fast and Easy Infinite Neural Networks in Pythoncited by 149
  28. 2024Adam-mini: Use Fewer Learning Rates To Gain Morecited by 134
  29. 2020On the linearity of large non-linear models: when and why the tangent kernel is constantcited by 129
  30. 2024Scaling Laws with Vocabulary: Larger Models Deserve Larger Vocabulariescited by 113
  31. 2018Mean Field Analysis of Neural Networks: A Central Limit Theoremcited by 107
  32. 2018On the Power of Over-parametrization in Neural Networks with Quadratic Activationcited by 102
  33. 2025Fantastic Pretraining Optimizers and Where to Find Themcited by 87
  34. 2016Approximation by Combinations of ReLU and Squared ReLU Ridge Functions with ℓ1 and ℓ0 Controlscited by 84
  35. 2024Understanding Warmup-Stable-Decay Learning Rates: A River Valley Loss Landscape Perspectivecited by 75
  36. 2025NorMuon: Making Muon more efficient and scalablecited by 74
  37. 2025Dion: Distributed Orthonormalized Updatescited by 64
  38. 2025Benchmarking Optimizers for Large Language Model Pretrainingcited by 53

50 most cited of 85 All 85 (JSON)

Built on by

Not cited yet.