arXiv:2606.20936cs.CLcs.AI2026-06

对比纯Transformer与混合模型在词元级的性能差异,发现混合模型更擅长处理语义状态依赖任务。

Comparing Transformers and Hybrid Models at the Token Level

  • 在相同输入下,对齐的混合模型与Transformer在词元级别直接比较
  • 混合模型在开放类词汇和指代追踪任务中损失更低,闭合类词和括号匹配任务中优势小
  • 揭示了循环层提升语义状态跟踪、注意力机制利于模式复制的分工机制

本文利用Olmo 3和Olmo Hybrid的开源权重,对匹配的Transformer与混合模型在相同前缀下的目标词元进行损失对比,按自然词元标签、复制特征、分隔符结构及受控合成探针分层分析。混合模型在多数标签族中损失更低,尤其在开放类内容词上优势显著,而在许多封闭类功能词上优势较小。在散文、代码和标记语言中,混合模型对起始分隔符的损失降低明显,对对应结束分隔符则几乎无优势,且在重复n-gram上基本无差别。合成探针显示:混合模型在代词记忆和实体追踪任务中占优,而Transformer在括号匹配这类需选择闭合分隔符的任务中表现更好。结果表明,混合模型中的循环层提升了文档语义状态依赖的预测能力,而注意力机制更擅长基于n-gram复制或句法括号匹配的任务。最后通过概念验证的筛选评估,展示了词元级分解如何增强对混合架构预训练诊断的精细度。

原文摘要 · Abstract (English)

Hybrid language models that mix attention and recurrent layers have shown promise: theoretically, recurrent layers ameliorate the limitations of pure transformers on state tracking, and empirically, hybrids can outperform pure transformers in loss and downstream evaluations \citep{waleffe2024empirical,merrill2026olmohybrid}. Yet it remains unclear which data or capabilities drive these gains, and to what degree they reflect the theoretical advantages motivating hybrid models. We address this question using the open weights from Olmo 3 \citep{olmo2025olmo3} and Olmo Hybrid \citep{merrill2026olmohybrid}: we compare the loss of a matched transformer and hybrid at the same target tokens under the same prefixes, stratifying the results by natural token tags, copy features, delimiter structure, and controlled synthetic probes. The hybrid has lower loss on most tag families, but the gains are not uniform: they are largest for open-class content words and smaller for many closed-class function words. Across prose, code, and markup, the hybrid's loss advantage is larger on opening delimiters than on the corresponding closing delimiters, and nearly vanishes on repeated $n$-grams. Synthetic probes show the same split: the hybrid is favored on pronoun-memory and entity-tracking tasks, whereas the transformer is favored on bracket-matching tasks that require choosing closing delimiters. These patterns suggest that the recurrent layers in hybrids improve predictions that leverage the semantic state of a document, whereas attention helps on tokens predictable by $n$-gram copying or syntactic bracket matching. We conclude with proof-of-concept filtered evaluations showing how token-level decompositions can sharpen pretraining diagnostics for hybrid architectures.

混合模型词元级分析语言模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。