用群体模型替代单模型,让预训练更高效。
q0: Primitives for Hyper-Epoch Pretraining

- 通过循环调度收集多个多样化模型,提升多样性。
- 链式蒸馏使模型质量逐代提升,整体表现超越单模型。
- 自适应选择策略适合不同计算预算,适用性强。
多轮次训练已成为常态,因算力增长快于高质量文本供给。但单个模型在数轮内即达到饱和,远未耗尽算力预算。我们提出从训练单一模型转向探索模型群体并聚合其预测的范式转变。引入超轮次预训练(q0),将多轮预算转化为多样模型群体,其联合预测性能优于单个精调模型。q0基于三个核心机制:具有反相关学习率与权重衰减的循环调度,从少数并行轨迹中获取多样化模型;链式蒸馏,使每轮模型基于前一轮优化,实现质量累积;基于保留集训练的可学习先验,按推理预算筛选并加权模型成员。在18亿参数模型上,使用1亿条FineWeb数据训练,q0仅需约56轮(较256轮基准少4.6倍)或约67轮(少3.8倍),即可匹配基准性能,并持续超越。在慢速运行(Slowrun)设置下,数据效率提升达12.9倍,并迁移至下游任务。关键的是,最优资源配置随预算变化,我们提供了从单轮到大规模预算的配置指南,以最大化泛化能力。
原文摘要 · Abstract (English)
Multi-epoch training is becoming the standard now that compute is growing faster than the supply of high-quality text. But pretraining a single model saturates within a few passes, long before the compute budget is exhausted. We argue this calls for a conceptual shift from training a single model toward exploring a population of models and aggregating their predictions. We introduce hyper-epoch pretraining (q0), which turns a multi-epoch budget into a population of diverse models whose combined predictions reach a lower validation loss than a single refined model. q0 reduces to three core primitives. A cyclic schedule with anti-correlated learning rate and weight decay collects diverse models from a few parallel trajectories. Chain distillation trains each model against its predecessor so that model quality compounds across the population. A learned prior, fit on a held out set, selects and weights members for any inference budget. On a 1.8B-parameter model trained on 100M FineWeb tokens, q0 matches a strong 256-epoch ensemble baseline using only ~56 epochs (~4.6x fewer), or ~67 epochs (~3.8x fewer) when matched to the baseline's ensemble size, and continues to improve beyond it. These gains reach cumulative ~12.9x data efficiency under the Slowrun setting and transfer to downstream benchmarks. Crucially, the optimal allocation shifts with the budget, so we give prescriptive recipes for how to spend a given epoch budget to maximize generalization, from a single epoch up to the largest budgets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。