Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

About

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.

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• 2021

Related benchmarks

TaskDatasetResultRank
Image ClassificationEuroSAT--
497
Image ClassificationDTD--
487
Image ClassificationSUN397--
425
Image ClassificationCIFAR100
Accuracy35.9
331
Image ClassificationImageNet
Top-1 Accuracy24.3
324
Image ClassificationCaltech-101
Top-1 Accuracy72
146
Image-to-Text RetrievalFlickr30K-CN
R@117.7
99
Text-to-Image RetrievalFlickr30K-CN
R@110.3
99
Image ClassificationFlowers
Top-1 Acc18.4
80
Image ClassificationCIFAR10
Top-1 Acc72.3
55
Showing 10 of 28 rows

Other info

Follow for update