arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.04858cs.CVcs.AIcs.CLcs.IRcs.LG

Re-ViLM:用于零样本和少样本图像描述的检索增强视觉语言模型

Re-ViLM: Retrieval-Augmented Visual Language Model for Zero and Few-Shot Image Captioning

UIUC(伊利诺伊大学厄巴纳-香槟分校);NVIDIA(英伟达公司);UT Austin(德克萨斯大学奥斯汀分校)
Zhuolin Yang, Wei Ping, Zihan Liu, Vijay Korthikanti, Weili Nie, De-An Huang, Linxi Fan, Zhiding Yu, Shiyi Lan, Bo Li, Ming-Yu Liu, Yuke Zhu, Mohammad Shoeybi, … 展开作者

Zhuolin Yang, Wei Ping, Zihan Liu, Vijay Korthikanti, Weili Nie, De-An Huang, Linxi Fan, Zhiding Yu, Shiyi Lan, Bo Li, Ming-Yu Liu, Yuke Zhu, Mohammad Shoeybi, Bryan Catanzaro, Chaowei Xiao, Anima Anandkumar

AI总结:

本文提出基于Flamingo的检索增强视觉语言模型Re-ViLM,通过外部数据库显式存储知识以减少参数并适应新数据,在域外零样本和少样本图像到文本生成任务中,以少于基线4倍的参数显著提升性能。

摘要:

使用视觉编码器(例如Flamingo)增强预训练语言模型(LMs)在图像到文本生成中获得了最先进的结果。然而,这些模型将所有知识存储在其参数中,因此通常需要庞大的模型参数来对丰富的视觉概念和详尽的文本描述进行建模。此外,它们在整合新数据方面效率低下,需要计算成本高昂的微调过程。在这项工作中,我们引入了基于Flamingo构建的检索增强视觉语言模型Re-ViLM,支持从外部数据库检索相关知识,用于零样本和上下文少样本图像到文本生成。通过在外部数据库中显式存储特定知识,我们的方法减少了模型参数数量,并能在评估期间通过简单更新数据库轻松适应新数据。我们还构建了促进上下文少样本学习能力的交错的图像和文本数据。我们证明,Re-ViLM显著提升了图像到文本生成任务的性能,特别是在域外设置下的零样本和少样本生成中,与基线方法相比,参数量减少了4倍。

英文摘要:

Augmenting pretrained language models (LMs) with a vision encoder (e.g., Flamingo) has obtained the state-of-the-art results in image-to-text generation. However, these models store all the knowledge within their parameters, thus often requiring enormous model parameters to model the abundant visual concepts and very rich textual descriptions. Additionally, they are inefficient in incorporating new data, requiring a computational-expensive fine-tuning process. In this work, we introduce a Retrieval-augmented Visual Language Model, Re-ViLM, built upon the Flamingo, that supports retrieving the relevant knowledge from the external database for zero and in-context few-shot image-to-text generations. By storing certain knowledge explicitly in the external database, our approach reduces the number of model parameters and can easily accommodate new data during evaluation by simply updating the database. We also construct an interleaved image and text data that facilitates in-context few-shot learning capabilities. We demonstrate that Re-ViLM significantly boosts performance for image-to-text generation tasks, especially for zero-shot and few-shot generation in out-of-domain settings with 4 times less parameters compared with baseline methods.

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

↑