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

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Jie, Shibo, Tang, Yehui, Ding, Ning, Deng, Zhi-Hong, Han, Kai, Wang, Yunhe
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910440110424064
author Jie, Shibo
Tang, Yehui
Ding, Ning
Deng, Zhi-Hong
Han, Kai
Wang, Yunhe
author_facet Jie, Shibo
Tang, Yehui
Ding, Ning
Deng, Zhi-Hong
Han, Kai
Wang, Yunhe
contents 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
format Preprint
id arxiv_https___arxiv_org_abs_2405_05615
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Memory-Space Visual Prompting for Efficient Vision-Language Fine-Tuning
Jie, Shibo
Tang, Yehui
Ding, Ning
Deng, Zhi-Hong
Han, Kai
Wang, Yunhe
Computer Vision and Pattern Recognition
Computation and Language
Machine Learning
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
title Memory-Space Visual Prompting for Efficient Vision-Language Fine-Tuning
topic Computer Vision and Pattern Recognition
Computation and Language
Machine Learning
url https://arxiv.org/abs/2405.05615