探究模型架构如何影响语言模型基础能力,提出避免能力下降的关键设计原则。
How Does Sequence Modeling Architecture Influence Base Capabilities of Pre-trained Language Models? Exploring Key Architecture Design Principles to Avoid Base Capabilities Degradation

- 提出限域预训练+分布外测试新范式,更准确揭示架构差异。
- 发现带状态的架构会显著降低模型基础能力,而Transformer保持稳定。
- 提炼出全序列任意选择能力是关键设计原则,可指导未来模型设计。
以Transformer为代表的预训练语言模型展现出强大的基础能力,其核心的自注意力机制已成为序列建模的经典架构。不同于以往聚焦提升注意力效率的研究,本文关注序列建模架构对基础能力的影响。我们指出,现有架构设计工作中普遍采用的混合域预训练设置无法充分揭示不同架构间的基础能力差异。为此,我们提出一种限域预训练结合分布外测试的新设置,成功在早期阶段揭示了架构间的显著能力差异。接着,分析有状态序列建模架构发现,其基础能力相比Transformer出现显著退化。通过一系列组件分析,我们总结出关键设计原则:序列建模架构需具备全序列任意元素选择能力,方可避免基础能力退化。最后,我们以极简的Top-1元素选择架构实证验证该原则,并推广至更实用的Top-1块选择架构。实验结果证明该原则的有效性,为未来架构改进与创新提供了重要参考。
原文摘要 · Abstract (English)
Pre-trained language models represented by the Transformer have been proven to possess strong base capabilities, and the representative self-attention mechanism in the Transformer has become a classic in sequence modeling architectures. Different from the work of proposing sequence modeling architecture to improve the efficiency of attention mechanism, this work focuses on the impact of sequence modeling architectures on base capabilities. Specifically, our concern is: How exactly do sequence modeling architectures affect the base capabilities of pre-trained language models? In this work, we first point out that the mixed domain pre-training setting commonly adopted in existing architecture design works fails to adequately reveal the differences in base capabilities among various architectures. To address this, we propose a limited domain pre-training setting with out-of-distribution testing, which successfully uncovers significant differences in base capabilities among architectures at an early stage. Next, we analyze the base capabilities of stateful sequence modeling architectures, and find that they exhibit significant degradation in base capabilities compared to the Transformer. Then, through a series of architecture component analysis, we summarize a key architecture design principle: A sequence modeling architecture need possess full-sequence arbitrary selection capability to avoid degradation in base capabilities. Finally, we empirically validate this principle using an extremely simple Top-1 element selection architecture and further generalize it to a more practical Top-1 chunk selection architecture. Experimental results demonstrate our proposed sequence modeling architecture design principle and suggest that our work can serve as a valuable reference for future architecture improvements and novel designs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。