千万亿参数的Transformer及其训练方法
Quadrillion-parameter transformers and how to train them

原始链接: https://qlabs.sh/research/parameters

Q Labs认为,AI与人类在样本效率上的差距可以通过在有限数据上训练更大的模型来缩小。他们的数据受限缩放定律从数据集大小、模型大小、轮次和集成规模预测损失,在Chinchilla式定律基础上加入了过拟合项和集成增益。在固定数据集上,计算最优模型大小随计算量持续增长,因此模型应更大并训练更多轮次,权重衰减也应相应缩放。他们还使用通过循环学习率进行的快照集成和深度集成;集成可减少过拟合,并在大预算下成为计算最优。外推到5T tokens时,规定参数量约为1e29 FLOPs下Chinchilla的7,000倍,达到千万亿规模网络。整体方案:先用快照集成扩展模型大小和轮次,再添加深度集成成员,在先前方案损失恶化的场景改善损失。他们总结道,这在有限数据上将网络规模提升四个数量级,但进一步进展仍需要新算法和硬件。

Hacker News 上一篇题为“Quadrillion-parameter transformers and how to train them”(来自 qlabs.sh)的帖子获得了 17 点和 3 条评论。一位评论者(joefourier)质疑将人脑(约 100T 突触参数、约 200M tokens、1:500,000 的比例)与 AI 模型作比较,认为人类具有来自进化而非随机初始化的强归纳偏置——将进化表述为“预训练”,人类学习表述为“微调”。另一位评论者(spwa4)质疑进化的“tokens”会存储在哪里,指出 DNA 缺乏容量。原帖作者(frozenseven)链接到 arxiv.org/abs/2610.37842 的论文。
相关文章

原文

Q Labs Research

Akshay Vegesna, Ethan Baron, Anirudh Satheesh, Samip DahalNovember 2026

TL;DR

$$L = \underbrace{E\vphantom{\left(\frac{N}{U^{\gamma}}\right)^{\kappa}}}_{\text{irreducible}} + \underbrace{\frac{A}{N^{\alpha}}\vphantom{\left(\frac{N}{U^{\gamma}}\right)^{\kappa}}}_{\text{small model}} + \underbrace{\frac{B}{(UR)^{\beta}}\vphantom{\left(\frac{N}{U^{\gamma}}\right)^{\kappa}}}_{\text{tokens seen}} + \underbrace{P\,(R-1)^{\delta}\left(\frac{N}{U^{\gamma}}\right)^{\kappa}}_{\text{overfitting}} - \underbrace{\frac{F}{N^{\nu}\,U^{\epsilon}\,R^{\rho}}\left(1-\frac{1}{M}\right)^{\mu}}_{\text{ensembling}}$$

1.Introduction

The 1,000,000x sample efficiency gap between AI and humans is one of the central mysteries of the field, with no good explanation. One clue is model size: Chinchilla’s prescription for 30T tokens is a ~1.5T-parameter model, a tokens per parameter of 20:1 (Hoffmann et al., 2022). Humans, by contrast, have ~100T parameters (counting each synapse as a weight) and learn from just ~200M tokens, a tokens per parameter of 1:500,000. The tokens per parameter gap between AI and humans is thus 10,000,000x, and an important step forward for the sample efficiency gap may be to close it. On 30T tokens, we should be training quintillion-parameter neural networks. Our current networks are way too small, by seven orders of magnitude.

Our new paper (Baron et al., 2026) makes progress towards this vision. Extrapolating its scaling law to a dataset of 5T tokens, the compute-optimal parameter count reaches ~7,000x the standard Chinchilla prescription at 1029 FLOPs. Given enough compute, we give prescriptions for training quadrillion-parameter networks.

Prescribed total parameters against training compute for a 5T-token dataset
Figure 1. Extrapolated compute-optimal parameter count on 5T tokens, summed over every model in the ensemble. Dots are the law's prescription at each budget, and the line is a power-law fit through them. Chinchilla stops at 250B parameters, while our law keeps growing with compute.

2.Parameters are all you need

Our new paper (Baron et al., 2026) finds that when the dataset is held fixed, the compute-optimal model size keeps growing with the compute budget. In this regime, we also want to increase the number of epochs and the number of ensemble members with the compute budget. By contrast, Chinchilla prescribes a fixed model size with a fixed dataset: 20 tokens per parameter. Extrapolating our paper’s prescriptions to 5T tokens, we find that the ensemble holds over 300 parameters per token at 1029 FLOPs, nearly 7,000x Chinchilla’s parameters-per-token ratio, and the count keeps growing with compute. We should be increasing parameter counts by orders of magnitude to achieve optimal performance in the multi-epoch regime.

Compute is growing much faster than the supply of high quality language training data (Villalobos et al., 2024), so language models will be increasingly trained in a multi-epoch data-constrained regime. To study this regime directly, we train transformers on fixed datasets, across a range of data sizes, model sizes, and epoch count, and fit how loss depends on each. Each run also uses snapshot ensembling, averaging the predictions of checkpoints saved during training (Huang et al., 2017; Mandal et al., 2026), which lowers loss at no extra training compute. While snapshot ensembling is a bit unusual, we find that ensembling is an important technique to integrate in a first-class way within the data-constrained regime.

Concretely, we fit a single scaling law that predicts the loss $L$ from four quantities: the number of unique tokens in the dataset $U$, the model size $N$, the number of epochs $R$, and the number of ensemble members $M$. The law extends the existing data-constrained scaling laws (Muennighoff et al., 2023; Lovelace et al., 2026) with one new term for ensembling:

$$\begin{aligned} L(U, N, R, M) ={}& \underbrace{E\vphantom{\left(\frac{N}{U^{\gamma}}\right)^{\kappa}}}_{\text{irreducible}} + \underbrace{\frac{A}{N^{\alpha}}\vphantom{\left(\frac{N}{U^{\gamma}}\right)^{\kappa}}}_{\text{small model}} + \underbrace{\frac{B}{(UR)^{\beta}}\vphantom{\left(\frac{N}{U^{\gamma}}\right)^{\kappa}}}_{\text{tokens seen}} + \underbrace{P\,(R-1)^{\delta}\left(\frac{N}{U^{\gamma}}\right)^{\kappa}}_{\text{overfitting}} \\[10pt] &- \underbrace{\frac{F}{N^{\nu}\,U^{\epsilon}\,R^{\rho}}\left(1-\frac{1}{M}\right)^{\mu}}_{\text{ensembling}} \end{aligned}$$

The first three terms come from Chinchilla: we have an irreducible cost, a term for the size of the model, and a term for the size of the dataset. The fourth term is an overfitting penalty following Lovelace et al. (2026), and the fifth is the gain from ensembling. It is zero for a single model and grows with each added model towards a ceiling. The constants and exponents are fit to our runs.

3.Larger models for more epochs

Prior data-constrained scaling laws hold weight decay fixed across scales (Lovelace et al., 2026). These pre-existing laws predict loss increasing with compute, and don't consider that we can use larger models with more regularization as part of the scaling law. Existing transfer rules don’t fix this problem. MuP covers learning rate, but not weight decay (Yang et al., 2022). The weight decay heuristics contradict each other (Wang and Aitchison, 2025; Bergsma et al., 2025; Qiu et al., 2025), and none account for epochs.

So we fit power laws for how these hyperparameters should scale in terms of unique data, model size, and epochs. We find that optimal weight decay grows steeply with epochs and with model size. With larger compute budgets and fixed datasets, we should train larger models for more epochs. This follows the finding from Kim et al. (2026), but goes further by giving concrete prescriptions for the training parameters.

Fitting and extrapolating the hyperparameter power laws
Figure 2. How the hyperparameter prescription is built. Left: at each grid cell, a quadratic fit over effective learning rate and weight decay locates the optimum. Middle: power laws through those optima, here weight decay against epochs for three model sizes on 50M tokens. Right: at 16x the fitted compute, our law lands near the optimum while fixed-hyperparameter rules miss.

Furthermore the law extrapolates well. We find that it is near-optimal at 16x of the fitted compute, and within 0.01 nats of the optimal loss on every axis we test. We find that each prior heuristic would miss on at least one axis.

Extrapolated to 5T tokens, the parameter count our data-constrained scaling law prescribes is ~7,000x larger than Chinchilla would predict at 1029 FLOPs, and the models train for several epochs each.

4.(Fast) Ensembling as a scaling axis

While our models are much bigger than Chinchilla would prescribe, we would like to scale parameter counts further in order to improve sample efficiency. We explore ensembling through the snapshot ensembles of q0 (Mandal et al., 2026), and through deep ensembles (Lakshminarayanan et al., 2017).

With snapshot ensembling, we save checkpoints from different points in a single training run and average their predictions (Huang et al., 2017; Mandal et al., 2026). Since the checkpoints come from one run, this costs no extra training compute. To make the checkpoints diverse, we train with a cyclic learning rate in the second half of training, which repeatedly raises and lowers the learning rate in one-epoch cycles. At the lowest learning rate of each cycle, we observe that the model converges to good solutions. We find that cycling the learning rate is better than holding the learning rate at its peak, and averaging the snapshots’ predictions lowers loss further. We hypothesize that this procedure works because of mode connectivity (Garipov et al., 2018): the good solutions are joined by low-loss paths, so short cycles can reach different strong models without much training.

We also include deep ensembling in our scaling laws. Deep ensembling trains independent models with different seeds and averages their predictions (Lakshminarayanan et al., 2017), so each member costs a full training run. The snapshots from each run are pooled into one ensemble. Unlike adding parameters, which increases overfitting, adding ensemble members reduces it, so we give ensembling its own term in the data-constrained scaling law. The term is zero for a single model, and its gain grows toward a ceiling as members are added. This law fits and extrapolates better than the alternatives, including on runs from Kim et al. (2026).

We find that deep ensembling becomes compute-optimal when the compute budget is large relative to the dataset. Past this point, the optimal number of members grows roughly linearly with compute. So the prescription with limited data is to grow model size and epoch count first with scaled weight decay, including snapshot ensembling in the multi-epoch training. Then we allocate the extra compute to deep ensemble members. In this way, loss continues to improve with more compute, whereas prior recipes eventually get worse.

Validation loss against training compute for the prior recipe, hyperparameter transfer, ensembling, and both
Figure 3. Validation loss against compute on 100M tokens. The prior recipe gets worse with compute. Hyperparameter transfer and ensembling each help, and together loss keeps falling. Dashed curves are each law's predicted optimum; the dotted line marks the top of the fit range.

5.Conclusion

In order to close the sample efficiency gap between AI and humans, we are motivated to train larger networks than today’s models. From comparison with the brain, we may want to increase the size of current models by seven orders of magnitude. In our latest paper (Baron et al., 2026), we provide prescriptions that increase the size of the network by four orders of magnitude on limited datasets, and give prescriptions showing how to train them. We think we can improve data efficiency by going further, but it will require new algorithms, hardware and more. We’ll have more to share on this soon.

References

  1. Ethan Baron, Shmuel Berman, Anirudh Satheesh, Akshay Vegesna, Samip Dahal, Andrew Gordon Wilson. Improving Data-Constrained Scaling with Hyperparameter Transfer and Ensembling. arXiv:2610.37842, 2026.
  2. Shane Bergsma, Nolan Dey, Gurpreet Gosal, Gavia Gray, Daria Soboleva, Joel Hestness. Power Lines: Scaling Laws for Weight Decay and Batch Size in LLM Pre-training. arXiv:2505.13738, 2025.
  3. Timur Garipov, Pavel Izmailov, Dmitrii Podoprikhin, Dmitry Vetrov, Andrew Gordon Wilson. Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNs. NeurIPS, 2018.
  4. Jordan Hoffmann, Sebastian Borgeaud, Arthur Mensch, Elena Buchatskaya, Trevor Cai, Eliza Rutherford, Diego de Las Casas, Lisa Anne Hendricks, Johannes Welbl, Aidan Clark, Tom Hennigan, Eric Noland, Katie Millican, George van den Driessche, Bogdan Damoc, Aurelia Guy, Simon Osindero, Karen Simonyan, Erich Elsen, Jack W. Rae, Oriol Vinyals, Laurent Sifre. Training Compute-Optimal Large Language Models. arXiv:2203.15556, 2022.
  5. Gao Huang, Yixuan Li, Geoff Pleiss, Zhuang Liu, John E. Hopcroft, Kilian Q. Weinberger. Snapshot Ensembles: Train 1, Get M for Free. ICLR, 2017.
  6. Konwoo Kim, Suhas Kotha, Percy Liang, Tatsunori Hashimoto. Pre-training under Infinite Compute. ICLR, 2026.
  7. Balaji Lakshminarayanan, Alexander Pritzel, Charles Blundell. Simple and Scalable Predictive Uncertainty Estimation using Deep Ensembles. NeurIPS, 2017.
  8. Justin Lovelace, Christian Belardi, Srivatsa Kundurthy, Shriya Sudhakar, Kilian Q. Weinberger. Prescriptive Scaling Laws for Data Constrained Training. arXiv:2605.01640, 2026.
  9. Bishwas Mandal, Shmuel Berman, Akshay Vegesna, Samip Dahal. q0: Primitives for Hyper-Epoch Pretraining. arXiv:2606.03938, 2026.
  10. Niklas Muennighoff, Alexander M. Rush, Boaz Barak, Teven Le Scao, Aleksandra Piktus, Nouamane Tazi, Sampo Pyysalo, Thomas Wolf, Colin Raffel. Scaling Data-Constrained Language Models. NeurIPS, 2023.
  11. Shikai Qiu, Zixi Chen, Hoang Phan, Qi Lei, Andrew Gordon Wilson. Hyperparameter Transfer Enables Consistent Gains of Matrix-Preconditioned Optimizers Across Scales. arXiv:2512.05620, 2025.
  12. Pablo Villalobos, Anson Ho, Jaime Sevilla, Tamay Besiroglu, Lennart Heim, Marius Hobbhahn. Will We Run Out of Data? Limits of LLM Scaling Based on Human-Generated Data. ICML, 2024.
  13. Xi Wang, Laurence Aitchison. How to Set AdamW's Weight Decay as You Scale Model and Dataset Size. ICML, 2025.
  14. Greg Yang, Edward J. Hu, Igor Babuschkin, Szymon Sidor, Xiaodong Liu, David Farhi, Nick Ryder, Jakub Pachocki, Weizhu Chen, Jianfeng Gao. Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfer. NeurIPS, 2021.

← Back to Q

联系我们 contact @ memedata.com