arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2602.04346cs.LG
收藏

MirrorLA: 用于视觉线性注意力的特征图反射

MirrorLA: Reflecting Feature Map for Vision Linear Attention

Harbin Institute of Technology, Shenzhen(哈尔滨工业大学深圳学院);Pengcheng Laboratory(鹏城实验室);UQMM Lab, University of Queensland(UQMM实验室,昆士兰大学)

Weikang Meng, Liangyu Huo, Yadan Luo, Yaowei Wang, Yingjian Li, Zheng Zhang

AI总结:

MirrorLA通过几何框架和可学习反射实现线性注意力的非负特征图重定向,提升性能并保持表示忠实性。

摘要:

线性注意力显著降低了Transformer的计算复杂度,从二次降至线性,但其性能始终劣于基于softmax的注意力。我们发现这种退化的原因是核特征图上的非负性约束:标准投影如ReLU充当"被动截断"运算符,无差别地丢弃负域中的语义信息。我们提出了MirrorLA,一个几何框架,将被动截断替换为主动重定向。通过利用可学习的Householder反射,MirrorLA将特征几何旋转到非负正交域以最大化信息保留。我们的方法通过一致的多尺度设计恢复表示密度:首先通过块状等距性优化局部可区分性,利用方差感知调制稳定长上下文动态以多样化激活,最后通过跨头反射整合分散的子空间以诱导全局协方差混合。MirrorLA在标准基准上实现了最先进的性能,证明了严格线性效率可以在不牺牲表示忠实性的情况下实现。

英文摘要:

Linear attention significantly reduces the computational complexity of Transformers from quadratic to linear, yet it consistently lags behind softmax-based attention in performance. We identify the root cause of this degradation as the non-negativity constraint imposed on kernel feature maps: standard projections like ReLU act as "passive truncation" operators, indiscriminately discarding semantic information residing in the negative domain. We propose MirrorLA, a geometric framework that substitutes passive truncation with active reorientation. By leveraging learnable Householder reflections, MirrorLA rotates the feature geometry into the non-negative orthant to maximize information retention. Our approach restores representational density through a cohesive, multi-scale design: it first optimizes local discriminability via block-wise isometries, stabilizes long-context dynamics using variance-aware modulation to diversify activations, and finally, integrates dispersed subspaces via cross-head reflections to induce global covariance mixing. MirrorLA achieves state-of-the-art performance across standard benchmarks, demonstrating that strictly linear efficiency can be achieved without compromising representational fidelity.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑