提出可提升模型推理能力的轻量级结构,让弱模型也能媲美强模型。
Physics of Language Models: Part 4.1, Architecture Design and the Magic of Canon Layers
- 设计轻量层促进相邻词元间信息横向流动,适配各类序列架构。
- 使弱模型推理深度提升2倍,线性注意力性能媲美SOTA模型。
- 通过可控合成任务验证,适合研究架构演进与未来模型预测。
理解语言模型的架构差异极具挑战性,尤其在学术规模预训练(如13亿参数、1000亿词元)下,结果常被噪声和随机性掩盖。为此,我们引入受控的合成预训练任务,以隔离并评估核心模型能力。在此框架中,我们发现CANON LAYERS:一种名为“卡农”的轻量级架构组件,能促进相邻词元间的水平信息流。该层通过对邻近词元表示进行加权求和,并无缝集成至Transformer、线性注意力、状态空间模型等序列架构中。我们呈现了12项关键结果,包括使推理深度提升2倍、增强推理广度与知识操控能力。它能使无位置编码(NoPE)模型达到旋转位置编码(RoPE)水平,使线性注意力媲美Mamba2/GDN等前沿线性模型——在合成任务与真实学术规模预训练中均得到验证。这一合成实验平台提供了经济、严谨的方法,用于揭示通常在学术尺度下被遮蔽的核心能力。借助无限高质量数据,甚至可预测未来架构在训练流程优化后的表现,如更优数据筛选或基于强化学习的后训练,从而解锁更深的推理与层级化推断。
原文摘要 · Abstract (English)
Understanding architectural differences in language models is challenging, especially at academic-scale pretraining (e.g., 1.3B parameters, 100B tokens), where results are often dominated by noise and randomness. To overcome this, we introduce controlled synthetic pretraining tasks that isolate and evaluate core model capabilities. Within this framework, we discover CANON LAYERS: lightweight architectural components -- named after the musical term "canon" -- that promote horizontal information flow across neighboring tokens. Canon layers compute weighted sums of nearby token representations and integrate seamlessly into Transformers, linear attention, state-space models, or any sequence architecture. We present 12 key results. This includes how Canon layers enhance reasoning depth (e.g., by $2\times$), reasoning breadth, knowledge manipulation, etc. They lift weak architectures like NoPE to match RoPE, and linear attention to rival SOTA linear models like Mamba2/GDN -- validated both through synthetic tasks and real-world academic-scale pretraining. This synthetic playground offers an economical, principled path to isolate core model capabilities often obscured at academic scales. Equipped with infinite high-quality data, it may even PREDICT how future architectures will behave as training pipelines improve -- e.g., through better data curation or RL-based post-training -- unlocking deeper reasoning and hierarchical inference.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。