用程序生成数据训练模型,能催生可拆解的推理结构。
Transformers Pretrained on Procedural Data Contain Modular Structures for Algorithmic Reasoning
- 用不同规则生成的合成数据,让模型产生互补的推理结构
- 注意力层和MLP块分别承载不同规则带来的可迁移结构
- 多规则组合可协同增强多种推理能力,适合研究模型可解释性
大规模语义丰富数据预训练是语言模型发展的关键。令人意外的是,近期研究表明,仅通过简单无语义的算法生成的合成数据,也能带来与自然语言预训练相当的收益。但这些简单合成数据究竟赋予模型何种能力、这些能力在模型架构中如何分布、又如何体现在权重中仍不明确。本文识别出几种有益的程序化数据形式,并发现其能提升小型Transformer模型的特定算法推理技能。核心发现是:不同程序规则会诱发模型中不同但互补的归纳结构。通过大量消融实验与部分迁移测试,我们发现这些结构分布在模型不同位置——注意力层常携带最可迁移的信息,而某些预训练规则则将有用结构赋予MLP模块。最有趣的是,多个规则诱导的结构可组合使用,共同强化多种能力。这表明语言模型的知识获取与推理能力可能被解耦,为提升模型鲁棒性与数据效率提供了新路径。
原文摘要 · Abstract (English)
Pretraining on large, semantically rich datasets is key for developing language models. Surprisingly, recent studies have shown that even synthetic data, generated procedurally through simple semantic-free algorithms, can yield some of the same benefits as natural language pretraining. It is unclear what specific capabilities such simple synthetic data instils in a model, where these capabilities reside in the architecture, and how they manifest within its weights. In this short paper, we identify several beneficial forms of procedural data, together with specific algorithmic reasoning skills that improve in small transformers. Our core finding is that different procedural rules instil distinct but complementary inductive structures in the model. With extensive ablations and partial-transfer experiments, we discover that these structures reside in different parts of the model. Attention layers often carry the most transferable information, but some pretraining rules impart useful structure to MLP blocks instead. Most interestingly, the structures induced by multiple rules can be composed to jointly reinforce multiple capabilities. These results suggest an exciting possibility of disentangling the acquisition of knowledge from reasoning in language models, with the goal of improving their robustness and data efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。