English

MirrorLA: Reflecting Feature Map for Vision Linear Attention

Machine Learning 2026-02-05 v1

Abstract

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.

Keywords

Cite

@article{arxiv.2602.04346,
  title  = {MirrorLA: Reflecting Feature Map for Vision Linear Attention},
  author = {Weikang Meng and Liangyu Huo and Yadan Luo and Yaowei Wang and Yingjian Li and Zheng Zhang},
  journal= {arXiv preprint arXiv:2602.04346},
  year   = {2026}
}
R2 v1 2026-07-01T09:35:36.471Z