arXiv:2509.05282cs.CL2025-09中稿 · COLM被引 2

剖析线性注意力衰减机制设计空间,揭示关键配置影响性能

Elucidating the Design Space of Decay in Linear Attention

  • 系统分析衰减参数化、共享、粒度与相对位置编码兼容性
  • 发现参数配置范围狭窄,错误设置会严重降低性能
  • 标量衰减常不如向量衰减,但特定策略下可反超

本文对线性复杂度序列模型中的衰减机制进行了全面研究。通过在四个关键维度上系统划分衰减机制的设计空间——参数化策略(计算衰减的方法)、参数共享(是否使用额外参数计算衰减)、衰减粒度(标量与向量衰减的对比)以及与相对位置编码方法(如旋转位置编码RoPE)的兼容性——我们在多种语言建模任务上进行了广泛实验。结果表明:首先,衰减的参数化策略需精细设计,有效配置通常局限于特定参数区间;其次,参数共享不能随意使用,否则可能导致衰减值过大或过小,显著影响性能;第三,在相同参数化策略下,标量衰减普遍表现逊于向量衰减,但在某些替代策略下,标量衰减可能意外优于向量衰减;最后,分析显示,主流的相对位置编码方法如RoPE,通常无法为多数线性注意力机制带来实质性收益。

原文摘要 · Abstract (English)

This paper presents a comprehensive investigation into the decay mechanisms inherent in linear complexity sequence models. We systematically delineate the design space of decay mechanisms across four pivotal dimensions: parameterization strategy, which refers to the computational methodology for decay; parameter sharing, which involves the utilization of supplementary parameters for decay computation; decay granularity, comparing scalar versus vector-based decay; and compatibility with relative positional encoding methods, such as Rotary Position Embedding (RoPE). Through an extensive series of experiments conducted on diverse language modeling tasks, we uncovered several critical insights. Firstly, the design of the parameterization strategy for decay requires meticulous consideration. Our findings indicate that effective configurations are typically confined to a specific range of parameters. Secondly, parameter sharing cannot be used arbitrarily, as it may cause decay values to be too large or too small, thereby significantly impacting performance. Thirdly, under identical parameterization strategies, scalar decay generally underperforms compared to its vector-based counterpart. However, in certain scenarios with alternative parameterization strategies, scalar decay may unexpectedly surpass vector decay in efficacy. Lastly, our analysis reveals that RoPE, a commonly employed relative positional encoding method, typically fails to provide tangible benefits to the majority of linear attention mechanisms.

注意力机制线性注意力位置编码参数设计

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