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

WenLan:通过大规模多模态预训练连接视觉和语言

WenLan: Bridging Vision and Language by Large-Scale Multi-Modal Pre-Training

Renmin University of China(中国人民大学);Institute of Computing Technology, Chinese Academy of Sciences(中国科学院计算技术研究所)
Yuqi Huo, Manli Zhang, Guangzhen Liu, Haoyu Lu, Yizhao Gao, Guoxing Yang, Jingyuan Wen, Heng Zhang, Baogui Xu, Weihao Zheng, Zongzheng Xi, Yueqian Yang, Anwen H… 展开作者

Yuqi Huo, Manli Zhang, Guangzhen Liu, Haoyu Lu, Yizhao Gao, Guoxing Yang, Jingyuan Wen, Heng Zhang, Baogui Xu, Weihao Zheng, Zongzheng Xi, Yueqian Yang, Anwen Hu, Jinming Zhao, Ruichen Li, Yida Zhao, Liang Zhang, Yuqing Song, Xin Hong, Wanqing Cui, Danyang Hou, Yingyan Li, Junyi Li, Peiyu Liu, Zheng Gong, Chuhao Jin, Yuchong Sun, Shizhe Chen, Zhiwu Lu, Zhicheng Dou, Qin Jin, Yanyan Lan, Wayne Xin Zhao, Ruihua Song, Ji-Rong Wen

摘要:

近年来,多模式预训练模型被广泛探索,以连接视觉和语言。然而,他们中的大多数明确的模型之间的跨模态的图像-文本对的相互作用,通过假设存在着很强的语义相关性之间的文本和图像模态。由于这种强假设在现实场景中通常是无效的,因此我们选择隐式地对大规模多模态预训练的跨模态相关性进行建模,这是我们团队领导的中国项目“WenLan”的重点。具体来说,在图像-文本对的弱相关性假设下,我们在跨模态对比学习框架内提出了一种称为BriVL的双塔预训练模型。与采用简单对比学习方法的OpenAI CLIP不同,我们通过将最新的方法MoCo应用于跨模态场景来设计更高级的算法。通过构建一个大型的基于字典的字典,我们的BriVL可以在有限的GPU资源中包含更多的负样本。我们进一步构建了一个名为RUC-CAS-WenLan的大型中文多源图像-文本数据集,用于预训练我们的BriVL模型。大量的实验表明,预训练的BriVL模型在各种下游任务上的性能优于UNITER和OpenAI CLIP。

英文摘要:

Multi-modal pre-training models have been intensively explored to bridge vision and language in recent years. However, most of them explicitly model the cross-modal interaction between image-text pairs, by assuming that there exists strong semantic correlation between the text and image modalities. Since this strong assumption is often invalid in real-world scenarios, we choose to implicitly model the cross-modal correlation for large-scale multi-modal pre-training, which is the focus of the Chinese project `WenLan' led by our team. Specifically, with the weak correlation assumption over image-text pairs, we propose a two-tower pre-training model called BriVL within the cross-modal contrastive learning framework. Unlike OpenAI CLIP that adopts a simple contrastive learning method, we devise a more advanced algorithm by adapting the latest method MoCo into the cross-modal scenario. By building a large queue-based dictionary, our BriVL can incorporate more negative samples in limited GPU resources. We further construct a large Chinese multi-source image-text dataset called RUC-CAS-WenLan for pre-training our BriVL model. Extensive experiments demonstrate that the pre-trained BriVL model outperforms both UNITER and OpenAI CLIP on various downstream tasks.

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

↑