arXiv:2602.05393cs.CLcs.LG2026-02被引 1

用小模型的后期知识加速大模型早期训练,提升速度与效果。

Late-to-Early Training: LET LLMs Learn Earlier, So Faster and Better

  • 用预训练小模型的深层表示指导大模型浅层学习
  • 1.4B模型训练提速1.6倍,下游任务准确率提升5%
  • 适合资源有限但想快速训练高效大模型的研究者

随着大语言模型通过扩大规模取得显著成功,预训练变得愈发关键却计算成本高昂。尽管已有大量耗费巨大算力训练的预训练模型,一个根本性现实问题仍未被充分探索:能否利用现有小型预训练模型加速更大模型的训练?本文提出晚到早训练(LET)范式,使大模型在早期训练阶段就能显式学习后期知识和早期层信息。核心思路是使用预训练模型(即后期训练阶段)的深层表示来指导目标模型早期层的训练。我们识别出两个关键机制:晚到早步学习和晚到早层学习。这些机制显著加速训练收敛,并稳健提升语言建模能力和下游任务表现,实现更快训练与更优性能。在1.4B和7B参数模型上的广泛实验表明其高效性与有效性。值得注意的是,在Pile数据集上训练1.4B模型时,该方法相比标准训练实现最高1.6倍加速,下游任务准确率提升近5%,即使使用的预训练模型参数量仅为目标模型的十分之一。

原文摘要 · Abstract (English)

As Large Language Models (LLMs) achieve remarkable empirical success through scaling model and data size, pretraining has become increasingly critical yet computationally prohibitive, hindering rapid development. Despite the availability of numerous pretrained LLMs developed at significant computational expense, a fundamental real-world question remains underexplored: \textit{Can we leverage existing small pretrained models to accelerate the training of larger models?} In this paper, we propose a Late-to-Early Training (LET) paradigm that enables LLMs to explicitly learn later knowledge in earlier steps and earlier layers. The core idea is to guide the early layers of an LLM during early training using representations from the late layers of a pretrained (i.e. late training phase) model. We identify two key mechanisms that drive LET's effectiveness: late-to-early-step learning and late-to-early-layer learning. These mechanisms significantly accelerate training convergence while robustly enhancing both language modeling capabilities and downstream task performance, enabling faster training with superior performance. Extensive experiments on 1.4B and 7B parameter models demonstrate LET's efficiency and effectiveness. Notably, when training a 1.4B LLM on the Pile dataset, our method achieves up to 1.6$\times$ speedup with nearly 5\% improvement in downstream task accuracy compared to standard training, even when using a pretrained model with 10$\times$ fewer parameters than the target model.

大模型训练知识迁移加速训练小模型赋能

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。