arXiv:2608.01075cs.AI2026-08

分离注意力匹配与内容检索的深度路由,提升模型性能

Role-Decoupled Attention Residuals: Separating Matching and Content Retrieval Across Depth

  • 让查询键与值独立选择不同深度的残差信息
  • 120M和343M模型下困惑度分别降低2.97%和2.43%
  • 仅增加每层一个向量参数,无额外计算开销

深度路由残差架构允许Transformer层从早期表示中检索信息,而非仅继承前一层状态。现有块注意力残差使用单一依赖内容的深度混合来构建查询、键和值的输入,这将两个功能不同的决策——注意力匹配位置与内容检索——耦合在一起。本文提出角色解耦注意力残差(RD-AttnRes),在查询和键共享同一深度路径的同时,为值学习独立的深度路径。该设计在保持原有结构的基础上,每层仅增加一个模型宽度向量,不引入额外的令牌间注意力操作。在FineWeb-Edu数据集上,使用冻结配对预训练协议,对120M和343M参数模型进行2.0B-token训练,共5组种子验证。所有10组对比中,验证负对数似然均下降,平均降幅分别为0.0301和0.0247,对应困惑度降低2.97%和2.43%。早期预算控制实验表明,改进并非由额外参数、重复路由执行或固定值路径造成。路由诊断显示查询-键与值的深度分布持续分化。结果表明,在所评估训练条件下,注意力匹配与内容检索应从残差层次中独立读取。

原文摘要 · Abstract (English)

Depth-routing residual architectures allow Transformer layers to retrieve earlier representations instead of inheriting only the immediately preceding state. Existing Block Attention Residuals, however, use a single content-dependent depth mixture to construct the inputs to queries, keys, and values. This design couples two functionally different decisions: queries and keys determine where attention matches, whereas values determine what content is retrieved. We therefore ask whether matching and content retrieval should be forced to read from the same depth. We introduce Role-Decoupled Attention Residuals (RD-AttnRes), a minimal extension that shares one depth route between queries and keys while learning an independent value route over the same residual sources. Tying the two routing queries exactly recovers the parent architecture, while decoupling them adds only one model-width vector per layer and introduces no additional token-to-token attention operation. We evaluate RD-AttnRes using a frozen, paired pretraining protocol on FineWeb-Edu with five matched seeds for both 120M- and 343M-parameter models and a 2.0B-token training budget. RD-AttnRes improves validation negative log-likelihood in all 10 matched comparisons. The mean reductions are 0.0301 and 0.0247, corresponding to perplexity reductions of 2.97 percent and 2.43 percent at 120M and 343M parameters, respectively. Early-budget controls indicate that neither the additional parameter count, duplicated routing execution, nor a fixed value route reproduces the improvement. Routing diagnostics further reveal persistent divergence between the query-key and value depth distributions. These results suggest that, within the evaluated training regime, attention matching and content retrieval benefit from distinct reads over the residual hierarchy.

注意力机制深度路由模型优化

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