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

Shibo Jie, Yehui Tang, Ning Ding, Zhi-Hong Deng, Kai Han +1 more
5/9/2024
cs.CVcs.CLcs.LG

Official code: JieShibo/MemVP (49 stars, last commit 2024-05-12).

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

View on arXivView PDF

Code Implementations(1)

JieShibo/MemVPOfficial100%

[ICML 2024] Memory-Space Visual Prompting for Efficient Vision-Language Fine-Tuning

495Shell, PythonMay 9, 20242 years ago

Similar Papers

Cite this paper

@article{jie2024memoryspace,
  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},
  year   = {2024},
  eprint = {2405.05615},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2405.05615v1},
  journal = {ICML 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2405.05615.svg)](https://codeofpaper.com/paper/2405.05615)

Discussion