Scaling Collapse Reveals Universal Dynamics in Compute-Optimally Trained Neural Networks
Not on the map. Why?
Abstract (arXiv)
What scaling limits govern neural network training dynamics when model size and training time grow in tandem? We show that despite the complex interactions between architecture, training algorithms, and data, compute-optimally trained models exhibit a remarkably precise universality. Specifically, loss curves from models of varying sizes collapse onto a single universal curve when training compute and loss are normalized to unity at the end of training. With learning rate decay, the collapse becomes so tight that differences in the normalized curves across models fall below the noise floor of individual loss curves across random seeds, a phenomenon we term supercollapse. We observe supercollapse across learning rate schedules, datasets, and architectures, including transformers trained on next-token prediction, and find it breaks down when hyperparameters are scaled suboptimally, providing a precise and practical indicator of good scaling. We explain these phenomena by connecting collapse to the power-law structure in typical neural scaling laws, and analyzing a simple yet surprisingly effective model of SGD noise dynamics that accurately predicts loss curves across various learning rate schedules and quantitatively explains the origin of supercollapse.
Key facts. Cited by 22 papers in the map (data through 14 Oct 2026). Builds on 41 of its 57 references that are in the map. First posted on arXiv 2 Jul 2025.
Builds on
41 of 57 references.
References · 41
- 2023GPT-4 Technical Reportcited by 18,396
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2016Gaussian Error Linear Units (GELUs)cited by 3,502
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2017Deep Learning Scaling is Predictable, Empiricallycited by 582
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
- 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
- 2018An Empirical Model of Large-Batch Trainingcited by 281
- 2020Feature Learning in Infinite-Width Neural Networkscited by 190
- 2023Small-scale proxies for large-scale Transformer training instabilitiescited by 183
- 2015Stochastic modified equations and adaptive stochastic gradient algorithmscited by 142
29 more
- 2019Which Algorithmic Choices Matter at Which Batch Sizes? Insights From a Noisy Quadratic Modelcited by 117
- 2024A Dynamical Model of Neural Scaling Lawscited by 111
- 2022Self-Consistent Dynamical Field Theory of Kernel Evolution in Wide Neural Networkscited by 105
- 2022On the SDEs and Scaling Rules for Adaptive Gradient Algorithmscited by 104
- 2022Adaptive Gradient Methods at the Edge of Stabilitycited by 101
- 2024Understanding Warmup-Stable-Decay Learning Rates: A River Valley Loss Landscape Perspectivecited by 81
- 2025Don't be lazy: CompleteP enables compute-efficient deep transformerscited by 78
- 2024Scaling Exponents Across Parameterizations and Optimizerscited by 76
- 20244+3 Phases of Compute-Optimal Neural Scaling Lawscited by 68
- 2023Depthwise Hyperparameter Transfer in Residual Networks: Dynamics and Scaling Limitcited by 66
- 2024Understanding Optimization in Deep Learning with Central Flowscited by 64
- 2024How Feature Learning Can Improve Neural Scaling Lawscited by 56
- 2024How to set AdamW's weight decay as you scale model and dataset sizecited by 43
- 2025Power Lines: Scaling Laws for Weight Decay and Batch Size in LLM Pre-trainingcited by 43
- 2023Feature-Learning Networks Are Consistent Across Widths At Realistic Scalescited by 41
- 2024Infinite Limits of Multi-head Transformer Dynamicscited by 39
- 2024Reconciling Kaplan and Chinchilla Scaling Lawscited by 39
- 2023Tensor Programs IVb: Adaptive Optimization in the Infinite-Width Limitcited by 38
- 2021SGD in the Large: Average-case Analysis, Asymptotics, and Stepsize Criticalitycited by 35
- 2025The Surprising Agreement Between Convex Optimization Theory and Learning-Rate Scheduling for Large Model Trainingcited by 33
- 2024Scaling Law with Learning Rate Annealingcited by 29
- 2022Homogenization of SGD in high-dimensions: Exact dynamics and generalization propertiescited by 27
- 2024Super Consistency of Neural Network Landscapes and Learning Rate Transfercited by 23
- 2024Rethinking Conventional Wisdom in Machine Learning: From Generalization to Scalingcited by 23
- 2025A Multi-Power Law for Loss Curve Prediction Across Learning Rate Schedulescited by 22
- 2025Gemstones: A Model Suite for Multi-Faceted Scaling Lawscited by 20
- 2024High dimensional analysis reveals conservative sharpening and a stochastic edge of stabilitycited by 18
- 2022Optimal learning rate schedules in high-dimensional non-convex optimization problemscited by 4
- 2023Universal Scaling Laws of Absorbing Phase Transitions in Artificial Deep Neural Networkscited by 4
Built on by
Citing papers per year: 2025: 7; 2026: 15.
22 papers.
Sort:
By area:
- 2025Completed Hyperparameter Transfer across Modules, Width, Depth, Batch and Durationcited by 27
- 2025Hyperparameter Transfer Enables Consistent Gains of Matrix-Preconditioned Optimizers Across Scalescited by 21
- 2026Hyperparameter Transfer with Mixture-of-Expert Layerscited by 13
- 2026Why Larger Models Learn More: Effects of Capacity, Interference, and Rare-Task Retentioncited by 10
- 2025Scaling with Collapse: Efficient and Predictable Training of LLM Familiescited by 9
- 2026Theory of Optimal Learning Rate Schedules and Scaling Laws for a Random Feature Modelcited by 7
- 2026Fast Catch-Up, Late Switching: Optimal Batch Size Scheduling via Functional Scaling Lawscited by 6
- 2025Efficient Hyperparameter Tuning via Trajectory Invariance Principlecited by 4
- 2026Decoupled DiLoCo for Resilient Distributed Pre-trainingcited by 4
- 2025Evaluating the Robustness of Chinchilla Compute-Optimal Scalingcited by 2
- 2025Predicting Training Re-evaluation Curves Enables Effective Data Curriculums for LLMscited by 2
- 2025Per-example gradients: a new frontier for understanding and improving optimizerscited by 2
10 more
- 2026Configuration-to-Performance Scaling Law with Neural Ansatzcited by 2
- 2026Statistical Properties of Training & Generalizationcited by 2
- 2026Explaining Data Mixing Scaling Lawscited by 1
- 2026Towards joint scaling laws with optimal batch size schedulescited by 1
- 2026Abra: Scaling Diffusion Image Trainingcited by 1
- 2026What do near-optimal learning rate schedules look like?cited by 0
- 2026Predicting Large Model Test Losses with a Noisy Quadratic Systemcited by 0
- 2026When is Warmstarting Effective for Scaling Language Models?cited by 0
- 2026Understanding the Subspace Stabilization of the Hessian and Gradient Covariance Matrixcited by 0
- 2026Simulating Training Spikes of Deep Neural Networks by Layer-wise Random Feature Modelscited by 0
- 2026Simulating Training Spikes of Deep Neural Networks by Layer-wise Random Feature Models9 Oct 2026
- 2026Understanding the Subspace Stabilization of the Hessian and Gradient Covariance Matrix25 Sep 2026
- 2026Abra: Scaling Diffusion Image Training18 Aug 2026
- 2026Towards joint scaling laws with optimal batch size schedules30 Jul 2026
- 2026Statistical Properties of Training & Generalization18 Jun 2026
- 2026Explaining Data Mixing Scaling Laws6 Jun 2026
- 2026Why Larger Models Learn More: Effects of Capacity, Interference, and Rare-Task Retention28 May 2026
- 2026When is Warmstarting Effective for Scaling Language Models?13 May 2026
- 2026Predicting Large Model Test Losses with a Noisy Quadratic System9 May 2026
- 2026Decoupled DiLoCo for Resilient Distributed Pre-training23 Apr 2026
- 2026What do near-optimal learning rate schedules look like?11 Mar 2026
- 2026Fast Catch-Up, Late Switching: Optimal Batch Size Scheduling via Functional Scaling Laws15 Feb 2026
10 more
- 2026Configuration-to-Performance Scaling Law with Neural Ansatz10 Feb 2026
- 2026Theory of Optimal Learning Rate Schedules and Scaling Laws for a Random Feature Model4 Feb 2026
- 2026Hyperparameter Transfer with Mixture-of-Expert Layers28 Jan 2026
- 2025Completed Hyperparameter Transfer across Modules, Width, Depth, Batch and Duration26 Dec 2025
- 2025Hyperparameter Transfer Enables Consistent Gains of Matrix-Preconditioned Optimizers Across Scales5 Dec 2025
- 2025Per-example gradients: a new frontier for understanding and improving optimizers30 Sep 2025
- 2025Predicting Training Re-evaluation Curves Enables Effective Data Curriculums for LLMs29 Sep 2025
- 2025Scaling with Collapse: Efficient and Predictable Training of LLM Families29 Sep 2025
- 2025Efficient Hyperparameter Tuning via Trajectory Invariance Principle29 Sep 2025
- 2025Evaluating the Robustness of Chinchilla Compute-Optimal Scaling28 Sep 2025
- 2026Abra: Scaling Diffusion Image Trainingcited by 1
- 2026Abra: Scaling Diffusion Image Training18 Aug 2026
- 2026Hyperparameter Transfer with Mixture-of-Expert Layerscited by 13
- 2026Hyperparameter Transfer with Mixture-of-Expert Layers28 Jan 2026
- 2025Completed Hyperparameter Transfer across Modules, Width, Depth, Batch and Durationcited by 27
- 2025Hyperparameter Transfer Enables Consistent Gains of Matrix-Preconditioned Optimizers Across Scalescited by 21
- 2026Why Larger Models Learn More: Effects of Capacity, Interference, and Rare-Task Retentioncited by 10
- 2025Scaling with Collapse: Efficient and Predictable Training of LLM Familiescited by 9
- 2026Theory of Optimal Learning Rate Schedules and Scaling Laws for a Random Feature Modelcited by 7
- 2026Fast Catch-Up, Late Switching: Optimal Batch Size Scheduling via Functional Scaling Lawscited by 6
- 2025Efficient Hyperparameter Tuning via Trajectory Invariance Principlecited by 4
- 2026Decoupled DiLoCo for Resilient Distributed Pre-trainingcited by 4
- 2025Evaluating the Robustness of Chinchilla Compute-Optimal Scalingcited by 2
- 2025Predicting Training Re-evaluation Curves Enables Effective Data Curriculums for LLMscited by 2
- 2025Per-example gradients: a new frontier for understanding and improving optimizerscited by 2
- 2026Configuration-to-Performance Scaling Law with Neural Ansatzcited by 2
12 most cited of 20 All 20 (JSON)
- 2026Simulating Training Spikes of Deep Neural Networks by Layer-wise Random Feature Models9 Oct 2026
- 2026Understanding the Subspace Stabilization of the Hessian and Gradient Covariance Matrix25 Sep 2026
- 2026Towards joint scaling laws with optimal batch size schedules30 Jul 2026
- 2026Statistical Properties of Training & Generalization18 Jun 2026
- 2026Explaining Data Mixing Scaling Laws6 Jun 2026
- 2026Why Larger Models Learn More: Effects of Capacity, Interference, and Rare-Task Retention28 May 2026
- 2026When is Warmstarting Effective for Scaling Language Models?13 May 2026
- 2026Predicting Large Model Test Losses with a Noisy Quadratic System9 May 2026
- 2026Decoupled DiLoCo for Resilient Distributed Pre-training23 Apr 2026
- 2026What do near-optimal learning rate schedules look like?11 Mar 2026
- 2026Fast Catch-Up, Late Switching: Optimal Batch Size Scheduling via Functional Scaling Laws15 Feb 2026
- 2026Configuration-to-Performance Scaling Law with Neural Ansatz10 Feb 2026
12 newest of 20 All 20 (JSON)