English

Memory-Space Visual Prompting for Efficient Vision-Language Fine-Tuning

Computer Vision and Pattern Recognition 2024-05-10 v1 Computation and Language Machine Learning

Abstract

Current solutions for efficiently constructing large vision-language (VL) models follow a two-step paradigm: projecting the output of pre-trained vision encoders to the input space of pre-trained language models as visual prompts; and then transferring the models to downstream VL tasks via end-to-end parameter-efficient fine-tuning (PEFT). However, this paradigm still exhibits inefficiency since it significantly increases the input length of the language models. In this paper, in contrast to integrating visual prompts into inputs, we regard visual prompts as additional knowledge that facilitates language models in addressing tasks associated with visual information. Motivated by the finding that Feed-Forward Network (FFN) of language models acts as "key-value memory", we introduce a novel approach termed memory-space visual prompting (MemVP), wherein visual prompts are concatenated with the weights of FFN for visual knowledge injection. Experimental results across various VL tasks and language models reveal that MemVP significantly reduces the training time and inference latency of the finetuned VL models and surpasses the performance of previous PEFT methods. Code: https://github.com/JieShibo/MemVP

Keywords

Cite

@article{arxiv.2405.05615,
  title  = {Memory-Space Visual Prompting for Efficient Vision-Language Fine-Tuning},
  author = {Shibo Jie and Yehui Tang and Ning Ding and Zhi-Hong Deng and Kai Han and Yunhe Wang},
  journal= {arXiv preprint arXiv:2405.05615},
  year   = {2024}
}

Comments

Accepted to ICML2024

R2 v1 2026-06-28T16:21:48.597Z