mlsanityData to 14 Oct 2026

Scaling with Collapse: Efficient and Predictable Training of LLM Families

arXiv 2509.25087 ↗First posted 29 Sep 2025cs.LG cs.AI cs.CLcited by 9

Not on the map. Why?

Abstract (arXiv)

Effective LLM training depends on predictable scaling of key quantities -- such as final loss and optimal hyperparameters -- with model and dataset size. Qiu et al. (2025) recently showed that this predictability can extend beyond scalars: whole training loss curves can *collapse* onto a universal trajectory after a simple normalization. What remains unclear is whether this phenomenon persists for LLM families trained under *practical scaling recipes*, where width, depth, learning rate, batch size, and weight decay are scaled jointly. We show that it does: loss curves collapse across scales precisely when optimization hyperparameters are set optimally for the given data budget, in accordance with recent empirical scaling laws. Collapse therefore emerges as a signature of compute-efficient training. We demonstrate two applications at scale: (1) deviation-from-collapse provides a sensitive, early diagnostic of training pathologies, and (2) predictability of collapsed curves enables early stopping in large-scale hyperparameter tuning. Finally, we train a competitive LLM family, *Celerity*, using these insights, establishing collapse as an effective tool for developing efficient LLMs.

Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Builds on 84 of its 110 references that are in the map. First posted on arXiv 29 Sep 2025.

Builds on

84 of 110 references.

References · 84

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2023GPT-4 Technical Reportcited by 18,396
  4. 2024The Llama 3 Herd of Modelscited by 14,715
  5. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
  6. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  7. 2025Qwen3 Technical Reportcited by 8,820
  8. 2020Scaling Laws for Neural Language Modelscited by 6,586
  9. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
  10. 2024Qwen2.5 Technical Reportcited by 4,942
  11. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
  12. 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,501
38 more
  1. 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
  2. 2025Gemma 3 Technical Reportcited by 1,824
  3. 2020GLU Variants Improve Transformercited by 1,804
  4. 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,631
  5. 2021Scaling Language Models: Methods, Analysis & Insights from Training Gophercited by 1,202
  6. 2016Optimization Methods for Large-Scale Machine Learningcited by 1,150
  7. 2021Train Short, Test Long: Attention with Linear Biases Enables Input Length Extrapolationcited by 1,103
  8. 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 905
  9. 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 723
  10. 2016Hyperband: A Novel Bandit-Based Approach to Hyperparameter Optimizationcited by 699
  11. 2016TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systemscited by 664
  12. 2024DeepSeek LLM: Scaling Open-Source Language Models with Longtermismcited by 642
  13. 2023The Falcon Series of Open Language Modelscited by 535
  14. 2020Scaling Laws for Autoregressive Generative Modelingcited by 522
  15. 2023Zephyr: Direct Distillation of LM Alignmentcited by 506
  16. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
  17. 2023Scaling Data-Constrained Language Modelscited by 403
  18. 2017Population Based Training of Neural Networkscited by 367
  19. 20242 OLMo 2 Furiouscited by 340
  20. 2025SmolLM2: When Smol Goes Big -- Data-Centric Training of a Small Language Modelcited by 335
  21. 2024OLMoE: Open Mixture-of-Experts Language Modelscited by 301
  22. 2018An Empirical Model of Large-Batch Trainingcited by 281
  23. 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 265
  24. 2021Hash Layers For Large Sparse Modelscited by 237
  25. 2024SOAP: Improving and Stabilizing Shampoo using Adamcited by 192
  26. 2020Feature Learning in Infinite-Width Neural Networkscited by 190
  27. 2023Small-scale proxies for large-scale Transformer training instabilitiescited by 183
  28. 2024Scaling Laws for Fine-Grained Mixture of Expertscited by 173
  29. 2024Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durationscited by 170
  30. 2024The Road Less Scheduledcited by 163
  31. 2022Revisiting Neural Scaling Laws in Language and Visioncited by 132
  32. 2019Which Algorithmic Choices Matter at Which Batch Sizes? Insights From a Noisy Quadratic Modelcited by 117
  33. 2022Broken Neural Scaling Lawscited by 109
  34. 2023Cerebras-GPT: Open Compute-Optimal Language Models Trained on the Cerebras Wafer-Scale Clustercited by 109
  35. 2023Tensor Programs VI: Feature Learning in Infinite-Depth Neural Networkscited by 97
  36. 2024Scaling Laws for Precisioncited by 95
  37. 2014Freeze-Thaw Bayesian Optimizationcited by 90
  38. 2024Resolving Discrepancies in Compute-Optimal Scaling of Language Modelscited by 88

50 most cited of 84 All 84 (JSON)

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 1; 2026: 8.

9 papers.