arXiv 论文揭示混合架构中的竞争学习现象并提出分阶段预训练方法
arXiv 论文 2610.26830 揭示混合架构语言模型中的竞争学习现象:不同 mixer 层竞争学习同一函数,胜者会抑制其他层的学习信号。层级归因显示现代循环层比 softmax attention 更快学会检索函数,导致 attention 多数情况下学不到检索;在训练中途停止循环层的权重更新可帮助 attention 学会检索。据此提出的分阶段预训练方法在最高 1B 参数模型上显著改善语言建模损失、常识推理和多种上下文长度下的检索任务。
推荐理由:论文指出现代循环层与 softmax attention 在训练中竞争学习检索函数并互相抑制,提出的分阶段训练方法在 1B 参数规模上验证有效。