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