arXiv:2409.19044cs.CLcs.AI2024-09NeurIPS被引 21

渐进式堆叠训练让模型更擅长推理,且速度提升40%。

On the Inductive Bias of Stacking Towards Improving Reasoning

论文配图:On the Inductive Bias of Stacking Towards Improving Reasoning
图 1 · 摘自论文原文
  • 提出MIDAS方法,分阶段逐步增加模型深度以加速训练。
  • 在1B~8B参数模型上,推理任务表现优于基线,尽管困惑度相近。
  • 发现堆叠训练具内在推理偏好,尤其适合阅读理解与数学题任务。

随着模型规模增大,渐进式堆叠训练(gradual stacking)因其高效性受到关注。该方法通过分阶段逐步增加模型深度,并利用前一阶段的小模型层初始化下一阶段。本文研究此类训练方式的归纳偏置,提出改进版MIDAS,可使语言模型训练速度提升最高达40%。意外发现:尽管在困惑度上与基线相当或略差,MIDAS在下游推理任务(如阅读理解、数学问题)上表现显著更优。通过构建推理原语(reasoning primitives)这一类合成任务,验证了堆叠预训练模型在无微调情况下也显著优于标准预训练。该现象在1B、2B和8B参数模型上均被证实。进一步分析表明,堆叠与循环模型存在关联,可能解释其推理偏好,获得充分实证支持。

原文摘要 · Abstract (English)

Given the increasing scale of model sizes, novel training strategies like gradual stacking [Gong et al., 2019, Reddi et al., 2023] have garnered interest. Stacking enables efficient training by gradually growing the depth of a model in stages and using layers from a smaller model in an earlier stage to initialize the next stage. Although efficient for training, the model biases induced by such growing approaches are largely unexplored. In this work, we examine this fundamental aspect of gradual stacking, going beyond its efficiency benefits. We propose a variant of gradual stacking called MIDAS that can speed up language model training by up to 40%. Furthermore we discover an intriguing phenomenon: MIDAS is not only training-efficient but surprisingly also has an inductive bias towards improving downstream tasks, especially tasks that require reasoning abilities like reading comprehension and math problems, despite having similar or slightly worse perplexity compared to baseline training. To further analyze this inductive bias, we construct reasoning primitives -- simple synthetic tasks that are building blocks for reasoning -- and find that a model pretrained with stacking is significantly better than standard pretraining on these primitives, with and without fine-tuning. This provides stronger and more robust evidence for this inductive bias towards reasoning. These findings of training efficiency and inductive bias towards reasoning are verified at 1B, 2B and 8B parameter language models. Finally, we conjecture the underlying reason for this inductive bias by exploring the connection of stacking to looped models and provide strong supporting empirical analysis.

模型训练推理能力归纳偏置堆叠训练

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