混合注意力与循环结构的模型在训练中更高效,性能更强。
Olmo Hybrid: From Theory to Practice and Back
- 用门控ΔNet替代Transformer滑动窗口层,构建混合模型Olmo Hybrid。
- 7B参数下,混合模型在预训练和中期评估中均优于纯Transformer版。
- 理论证明混合模型可处理复杂任务,且表达力更强,适合大规模语言建模研究者。
近期工作展示了非Transformer语言模型(如线性循环神经网络和混合模型)的潜力,但尚无共识认为其架构优势足以承担扩展风险。为此,我们从多个维度提供证据:理论上,混合模型不仅继承Transformer与线性RNN的表达能力,还能表示两者之外的任务(如代码执行);实践上,我们训练了Olmo Hybrid(7B参数),将原Olmo 3 7B中的滑动窗口层替换为门控ΔNet层。结果显示,Olmo Hybrid在标准预训练和中期评估中均优于Olmo 3,表明混合模型在可控的大规模设置中具备优势。其训练效率显著高于纯Transformer,解释了更高性能。然而,为何在形式化问题上的更强表达力能带来更好的泛化表现仍不明确。为此,我们回归理论,论证更强表达力如何转化为更高的训练扩展效率,完成闭环。总体而言,混合模型是语言建模范式的重要延伸:不仅降低推理内存,更从根本上提升模型表达力与预训练扩展效率。
原文摘要 · Abstract (English)
Recent work has demonstrated the potential of non-transformer language models, especially linear recurrent neural networks (RNNs) and hybrid models that mix recurrence and attention. Yet there is no consensus on whether the potential benefits of these new architectures justify the risk and effort of scaling them up. To address this, we provide evidence for the advantages of hybrid models over pure transformers on several fronts. First, theoretically, we show that hybrid models do not merely inherit the expressivity of transformers and linear RNNs, but can express tasks beyond both, such as code execution. Putting this theory to practice, we train Olmo Hybrid, a 7B-parameter model largely comparable to Olmo 3 7B but with the sliding window layers replaced by Gated DeltaNet layers. We show that Olmo Hybrid outperforms Olmo 3 across standard pretraining and mid-training evaluations, demonstrating the benefit of hybrid models in a controlled, large-scale setting. We find that the hybrid model scales significantly more efficiently than the transformer, explaining its higher performance. However, its unclear why greater expressivity on specific formal problems should result in better scaling or superior performance on downstream tasks unrelated to those problems. To explain this apparent gap, we return to theory and argue why increased expressivity should translate to better scaling efficiency, completing the loop. Overall, our results suggest that hybrid models mixing attention and recurrent layers are a powerful extension to the language modeling paradigm: not merely to reduce memory during inference, but as a fundamental way to obtain more expressive models that scale better during pretraining.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。