arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2301.05065cs.CVcs.AI
收藏

构建面向语言、视觉和视觉-语言理解任务的通用基础模型

Toward Building General Foundation Models for Language, Vision, and Vision-Language Understanding Tasks

ByteDance Research(字节跳动研究院);HKUST(香港科技大学)

Xinsong Zhang, Yan Zeng, Jipeng Zhang, Hang Li

AI总结:

本文提出通用基础模型X-FM,通过停止梯度与引导视觉编码器训练,在语言、视觉及视觉-语言任务上超越现有通用模型,媲美专用模型。

摘要:

基础模型或预训练模型已显著提升了各种语言、视觉和视觉-语言理解任务的性能。然而,现有的基础模型只能在一种类型的任务中表现最佳,即语言、视觉或视觉-语言任务。是否有可能构建一个在所有理解任务中表现最佳的基础模型(我们称之为通用基础模型)仍是一个开放问题。在本文中,我们提出了一种新的通用基础模型X-FM(X基础模型)。X-FM具有一个语言编码器、一个视觉编码器和一个融合编码器,以及一种新的训练方法。该训练方法包括两种从文本、图像和图像-文本对数据中学习X-FM的新技术。一种是在学习语言编码器时停止来自视觉-语言训练的梯度。另一种是利用视觉-语言训练来指导视觉编码器的学习。在基准数据集上的大量实验表明,X-FM可以显著优于现有的通用基础模型,并且比专门针对语言、视觉或视觉-语言理解而设计的现有基础模型表现更好或相当。代码和预训练模型已在https://github.com/zhangxinsong-nlp/XFM发布。

英文摘要:

Foundation models or pre-trained models have substantially improved the performance of various language, vision, and vision-language understanding tasks. However, existing foundation models can only perform the best in one type of tasks, namely language, vision, or vision-language. It is still an open question whether it is possible to construct a foundation model performing the best for all the understanding tasks, which we call a general foundation model. In this paper, we propose a new general foundation model, X-FM (the X-Foundation Model). X-FM has one language encoder, one vision encoder, and one fusion encoder, as well as a new training method. The training method includes two new techniques for learning X-FM from text, image, and image-text pair data. One is to stop gradients from the vision-language training when learning the language encoder. The other is to leverage the vision-language training to guide the learning of the vision encoder. Extensive experiments on benchmark datasets show that X-FM can significantly outperform existing general foundation models and perform better than or comparable to existing foundation models specifically for language, vision, or vision-language understanding. Code and pre-trained models are released at https://github.com/zhangxinsong-nlp/XFM.

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

↑