English

Selective LoRA for Visual Tokens and Attention Heads

Computer Vision and Pattern Recognition 2026-05-12 v2 Artificial Intelligence

Abstract

Low-rank adaptation (LoRA) is widely used for parameter-efficient fine-tuning, but its standard all-token, all-head design ignores the heterogeneous structure of vision language model (VLM) inputs. We introduce \emph{Image-LoRA}, a vision-oriented PEFT recipe that views LoRA as a token-level residual update and applies this update only to visual tokens. Image-LoRA further restricts adaptation to the value path of a compact subset of attention heads, selected using a one-pass influence estimate from a rank-1 visual-token-only probe. This token-, head-, and value-selective design reduces trainable parameters and adapter-only training FLOPs while leaving the pure-text forward pass of the frozen backbone unchanged when no visual tokens are present. Across visual localization benchmarks with controlled text:image token ratios, Image-LoRA matches or closely approaches standard LoRA, while showing especially favorable trade-offs in image-token-heavy regimes. We further validate its generality on TextVQA and VideoQA, verify pure-text preservation on GSM8K, and show on ViLP that a stronger information bottleneck can yield gains over standard LoRA.

Keywords

Cite

@article{arxiv.2512.19219,
  title  = {Selective LoRA for Visual Tokens and Attention Heads},
  author = {Tiange Luo and Lajanugen Logeswaran and Jaekyeom Kim and Justin Johnson and Honglak Lee},
  journal= {arXiv preprint arXiv:2512.19219},
  year   = {2026}
}
R2 v1 2026-07-01T08:36:34.794Z