只微调视觉令牌和关键注意力头,提升视觉语言模型效率
Selective LoRA for Visual Tokens and Attention Heads
- 仅对视觉令牌和部分注意力头的值路径进行低秩适配
- 参数量减少50%以上,图像主导任务性能持平或更优
- 适合需要高效微调视觉语言模型的研究者
低秩适配(LoRA)广泛用于参数高效微调,但其全令牌、全头的设计忽略了视觉语言模型输入的异质结构。我们提出面向视觉的PEFT方法Image-LoRA,将LoRA视为令牌级残差更新,并仅应用于视觉令牌。Image-LoRA进一步通过一次传递的秩-1视觉令牌探针估计,选择一个紧凑的注意力头子集,仅在这些头的值路径上进行适配。该令牌、头和值的三重选择设计显著减少可训练参数和仅适配器训练的浮点运算量,同时在无视觉令牌时保持冻结主干的纯文本前向传播不变。在控制文本:图像令牌比例的视觉定位基准上,Image-LoRA表现与标准LoRA相当或接近,尤其在图像令牌密集场景下展现出更优权衡。我们在TextVQA和VideoQA上验证其泛化能力,在GSM8K上确认纯文本保留性,并在ViLP上显示更强信息瓶颈可带来超越标准LoRA的增益。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。