用小模型辅助大模型预训练,提速且提质量
A Little Help Goes a Long Way: Efficient LLM Training by Leveraging Small LMs
- 用小模型生成软标签并筛选关键训练样本
- 2.8B大模型训练时间减少,性能提升
- 适合资源有限但想高效训练大模型的研究者
大型语言模型(LLM)预训练成本高昂,通常需在大规模语料上优化自监督目标(如下一词预测)。本文提出一种新范式:利用小型语言模型(SLM)提升大模型预训练效率与质量。具体通过SLM提供软标签作为额外监督信号,并筛选出具有信息量和难度的训练样本。该方法使大模型有效迁移小模型的预测分布,同时聚焦数据分布中的关键区域。实验证明,使用1.5B参数的小模型在Pile数据集上辅助2.8B参数的大模型预训练,显著缩短训练时间并提升整体性能。理论方面,构建统计框架分析小模型监督对大模型训练的增益机制,揭示其虽监督质量较低但仍可提升大模型表现,且需平衡软标签带来的偏差与方差。实验结果验证了该理论框架的有效性。
原文摘要 · Abstract (English)
A primary challenge in large language model (LLM) development is their onerous pre-training cost. Typically, such pre-training involves optimizing a self-supervised objective (such as next-token prediction) over a large corpus. This paper explores a promising paradigm to improve LLM pre-training efficiency and quality by suitably leveraging a small language model (SLM). In particular, this paradigm relies on an SLM to both (1) provide soft labels as additional training supervision, and (2) select a small subset of valuable ("informative" and "hard") training examples. Put together, this enables an effective transfer of the SLM's predictive distribution to the LLM, while prioritizing specific regions of the training data distribution. Empirically, this leads to reduced LLM training time compared to standard training, while improving the overall quality. Theoretically, we develop a statistical framework to systematically study the utility of SLMs in enabling efficient training of high-quality LLMs. In particular, our framework characterizes how the SLM's seemingly low-quality supervision can enhance the training of a much more capable LLM. Furthermore, it also highlights the need for an adaptive utilization of such supervision, by striking a balance between the bias and variance introduced by the SLM-provided soft labels. We corroborate our theoretical framework by improving the pre-training of an LLM with 2.8B parameters by utilizing a smaller LM with 1.5B parameters on the Pile dataset.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。