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 ${\sim}56$ epochs (${\sim}4.6\times$ fewer), or ${\sim}67$ epochs (${\sim}3.8\times$ fewer) when matched to the baseline's ensemble size, and continues to improve beyond it. These gains reach cumulative ${\sim}12.9\times$ 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.


翻译:多周期训练正成为标准做法,因为算力增长快于高质量文本的供应。但单个模型的预训练在几个周期后就会饱和,远早于计算预算耗尽。我们认为,这需要从训练单个模型转向探索模型群体并聚合其预测的概念性转变。我们引入超周期预训练(q0),它将多周期预算转化为多样化的模型群体,其组合预测比单个精炼模型达到更低的验证损失。q0归结为三个核心原语。一个具有反相关学习率和权重衰减的循环调度,从多个并行轨迹中收集多样化模型。链式蒸馏使每个模型针对其前驱进行训练,从而使模型质量在群体中累积提升。一个在保留集上拟合的学习先验,为任意推理预算选择和加权成员。在1.8B参数模型上使用100M FineWeb令牌进行训练,q0仅用约56个周期(减少约4.6倍)或约67个周期(减少约3.8倍,当与基线集成大小匹配时)即可匹配强256周期集成基线,并继续超越。这些增益在Slowrun设置下达到累积约12.9倍的数据效率,并迁移到下游基准。关键在于,最优分配随预算变化,因此我们针对如何花费给定周期预算以最大化泛化给出了规范性配方,范围从单个周期到最大预算。

0
下载
关闭预览

相关内容

ACM/IEEE第23届模型驱动工程语言和系统国际会议,是模型驱动软件和系统工程的首要会议系列,由ACM-SIGSOFT和IEEE-TCSE支持组织。自1998年以来,模型涵盖了建模的各个方面,从语言和方法到工具和应用程序。模特的参加者来自不同的背景,包括研究人员、学者、工程师和工业专业人士。MODELS 2019是一个论坛,参与者可以围绕建模和模型驱动的软件和系统交流前沿研究成果和创新实践经验。今年的版本将为建模社区提供进一步推进建模基础的机会,并在网络物理系统、嵌入式系统、社会技术系统、云计算、大数据、机器学习、安全、开源等新兴领域提出建模的创新应用以及可持续性。 官网链接:http://www.modelsconference.org/
《多模态持续预训练实用指南》,52页pdf
专知会员服务
24+阅读 · 2024年9月3日
多模态预训练模型综述
专知会员服务
94+阅读 · 2023年11月20日
预训练语言模型的应用综述
专知会员服务
36+阅读 · 2023年1月23日
视频文本预训练简述
专知会员服务
22+阅读 · 2022年7月24日
视觉语言多模态预训练综述
专知会员服务
122+阅读 · 2022年7月11日
「知识增强预训练语言模型」最新研究综述
专知
18+阅读 · 2022年11月18日
强化学习开篇:Q-Learning原理详解
AINLP
37+阅读 · 2020年7月28日
一大批中文(BERT等)预训练模型等你认领!
PaperWeekly
15+阅读 · 2019年6月25日
自然语言处理中的语言模型预训练方法
PaperWeekly
14+阅读 · 2018年10月21日
国家自然科学基金
0+阅读 · 2017年12月31日
国家自然科学基金
6+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
12+阅读 · 2015年12月31日
国家自然科学基金
2+阅读 · 2014年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
VIP会员
最新内容
《无人地面战车(UGV)的崛起》报告
专知会员服务
7+阅读 · 7月16日
美陆军任务式指挥人工智能解决方案
专知会员服务
11+阅读 · 7月16日
相关基金
国家自然科学基金
0+阅读 · 2017年12月31日
国家自然科学基金
6+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
12+阅读 · 2015年12月31日
国家自然科学基金
2+阅读 · 2014年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
Top
微信扫码咨询专知VIP会员