arXiv:2603.07461cs.CLcs.AI2026-03被引 3

提出双流Transformer,让模型可解释性与性能可控平衡。

The Dual-Stream Transformer: Channelized Architecture for Interpretable Language Modeling

  • 将残差流拆为注意力更新的令牌流和前馈网络更新的上下文流。
  • 独立头混合使验证损失上升8%,推荐策略仅增2.5%开销。
  • 支持推理时放大注意力,适合需要透明决策的场景。

标准Transformer将所有计算混在单一残差流中,难以追踪各组件功能。本文提出双流Transformer,将残差流分解为两个功能独立的分支:由注意力更新的令牌流和由前馈网络更新的上下文流。注意力头间的信息流动通过多级混合策略控制,从完全独立(最大可解释性)到密集连接(标准Transformer行为)。该设计实现了可解释性与性能间的可调权衡。在2900万参数的语言建模任务上测试,完全独立头混合使验证损失相对基线增加8%;推荐的Kronecker混合策略仅增加2.5%开销。所有配置在注意力放大(推理时逻辑值放大至16倍)下仍保持功能性生成,性能退化范围为16%至27%。这种鲁棒性表明模型学习的是离散算法,不依赖软概率混合。该架构为可解释语言模型提供了结构性基础。

原文摘要 · Abstract (English)

Standard transformers entangle all computation in a single residual stream, obscuring which components perform which functions. We introduce the Dual-Stream Transformer, which decomposes the residual stream into two functionally distinct components: a token stream updated by attention and a context stream updated by feed-forward networks. Information flow between attention heads is controlled through a hierarchy of mixing strategies, from fully independent (maximum interpretability) to dense (standard transformer behavior). This design exposes a tunable tradeoff between interpretability and performance. We measure this tradeoff on language modeling tasks at 29M parameters. Fully independent head mixing increases validation loss by 8\% relative to dense baselines. The recommended Kronecker mixing strategy, which permits scalar communication between heads while preserving within-head structure, costs only 2.5\%. All configurations maintain functional generation under attention amplification (scaling logits by factors up to 16 at inference time), with degradation ranging from 16\% to 27\%. This robustness suggests the architectures learn discrete algorithms that operate independently of soft probabilistic mixing. The architecture provides a foundation for interpretable language models where internal structure is exposed by design. \footnote{This work was partially supported by DARPA Contract HR001125C0302.}

可解释性Transformer双流结构

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