arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2211.09623cs.CVcs.AIcs.CL

面向视觉-语言检索的跨模态适配器

Cross-Modal Adapter for Vision-Language Retrieval

Department of Automation, BNRist, Tsinghua University(自动化系,北京理工大学,清华大学);School of Computer Science and Technology, Beijing Institute of Technology(计算机科学与技术学院,北京理工大学)

Haojun Jiang, Jianke Zhang, Rui Huang, Chunjiang Ge, Zanlin Ni, Shiji Song, Gao Huang

AI总结:

针对视觉-语言检索中预训练模型全量微调过拟合风险高、成本大的问题,提出支持编码器级隐式跨模态交互的跨模态适配器,以少量参数实现高效迁移,在多个图文、视频文本检索数据集上性能优于现有适配器方法。

摘要:

视觉-语言检索是多模态学习中的重要研究方向,其目标是针对给定的文本查询检索出最相关的视觉候选对象。近年来,CLIP等预训练模型在检索任务中展现出巨大潜力。然而,随着预训练模型规模不断扩大,在下游检索数据集上对其进行全量微调存在很高的过拟合风险。此外,在实际应用中,为每个任务单独训练并存储一个大模型的成本十分高昂。为解决上述问题,我们提出了一种用于参数高效迁移学习的新型跨模态适配器(Cross-Modal Adapter)。受基于适配器的方法启发,我们仅通过少量参数化层调整预训练模型,但该方法存在两处显著差异:其一,我们的方法是面向多模态领域设计的;其二,它支持视觉编码器与语言编码器之间的编码器级隐式跨模态交互。该方法尽管结构简单,却具备三大显著优势:(1)大幅减少需微调的参数量;(2)节省训练时间;(3)可固定所有预训练参数,实现预训练模型在不同数据集间的共享。大量实验表明,在无需额外技巧的情况下,我们的方法在图文检索数据集(MSCOCO、Flickr30K)以及视频文本检索数据集(MSR-VTT、DiDeMo和ActivityNet)上的表现均优于现有基于适配器的方法。

英文摘要:

Vision-language retrieval is an important multi-modal learning topic, where the goal is to retrieve the most relevant visual candidate for a given text query. Recently, pre-trained models, e.g., CLIP, show great potential on retrieval tasks. However, as pre-trained models are scaling up, fully fine-tuning them on donwstream retrieval datasets has a high risk of overfitting. Moreover, in practice, it would be costly to train and store a large model for each task. To overcome the above issues, we present a novel Cross-Modal Adapter for parameter-efficient transfer learning. Inspired by adapter-based methods, we adjust the pre-trained model with a few parameterization layers. However, there are two notable differences. First, our method is designed for the multi-modal domain. Secondly, it allows encoder-level implicit cross-modal interactions between vision and language encoders. Although surprisingly simple, our approach has three notable benefits: (1) reduces the vast majority of fine-tuned parameters, (2) saves training time, and (3) allows all the pre-trained parameters to be fixed, enabling the pre-trained model to be shared across datasets. Extensive experiments demonstrate that, without bells and whistles, our approach outperforms adapter-based methods on image-text retrieval datasets (MSCOCO, Flickr30K) and video-text retrieval datasets (MSR-VTT, DiDeMo, and ActivityNet).

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑