arXiv:2410.17779cs.CV2024-10IJCV被引 7

用嵌入式融合提升视觉语言模型效率,训练推理更快更省资源。

ADEM-VL: Adaptive and Embedded Fusion for Efficient Vision-Language Tuning

  • 不增参数,通过嵌入视觉特征到语言空间实现高效多模态融合。
  • 单次前向传播生成多尺度视觉特征,加速模型计算过程。
  • 动态丢弃无关视觉信息,只保留关键特征,适合资源受限场景。

近期多模态融合进展中,视觉-语言(VL)模型在图像描述、视觉问答等任务中表现优异。然而,构建此类模型需大量硬件资源,效率受限于两大因素:语言模型输入序列因加入视觉特征而变长,导致计算量增加;额外可学习参数增多,提升内存复杂度。为解决这一问题,我们提出 ADEM-VL,一种基于预训练大语言模型(LLM)的高效视觉-语言调优方法。该方法采用无参交叉注意力机制进行相似性度量,在多模态融合中仅将视觉特征嵌入语言空间,大幅减少可训练参数,显著加快训练与推理速度。为增强融合模块表征能力,我们设计了一种高效多尺度特征生成方案,仅需一次视觉编码器前向传播即可完成。此外,提出自适应融合机制,根据注意力得分动态剔除每段文本对应的低相关性视觉信息,确保融合聚焦于最相关特征。在视觉问答、图像描述和指令遵循等多个任务上实验表明,本框架性能优于现有方法,尤其在 ScienceQA 数据集上平均准确率提升 0.77%,同时降低训练与推理延迟,验证了其优越性。代码已公开于 https://github.com/Hao840/ADEM-VL。

原文摘要 · Abstract (English)

Recent advancements in multimodal fusion have witnessed the remarkable success of vision-language (VL) models, which excel in various multimodal applications such as image captioning and visual question answering. However, building VL models requires substantial hardware resources, where efficiency is restricted by two key factors: the extended input sequence of the language model with vision features demands more computational operations, and a large number of additional learnable parameters increase memory complexity. These challenges significantly restrict the broader applicability of such models. To bridge this gap, we propose ADEM-VL, an efficient vision-language method that tunes VL models based on pretrained large language models (LLMs) by adopting a parameter-free cross-attention mechanism for similarity measurements in multimodal fusion. This approach only requires embedding vision features into the language space, significantly reducing the number of trainable parameters and accelerating both training and inference speeds. To enhance representation learning in fusion module, we introduce an efficient multiscale feature generation scheme that requires only a single forward pass through the vision encoder. Moreover, we propose an adaptive fusion scheme that dynamically discards less relevant visual information for each text token based on its attention score. This ensures that the fusion process prioritizes the most pertinent visual features. With experiments on various tasks including visual question answering, image captioning, and instruction-following, we demonstrate that our framework outperforms existing approaches. Specifically, our method surpasses existing methods by an average accuracy of 0.77% on ScienceQA dataset, with reduced training and inference latency, demonstrating the superiority of our framework. The code is available at https://github.com/Hao840/ADEM-VL.

多模态融合模型效率视觉语言

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。