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

An Empirical Study of Training End-to-End Vision-and-Language Transformers

About

Vision-and-language (VL) pre-training has proven to be highly effective on various VL downstream tasks. While recent work has shown that fully transformer-based VL models can be more efficient than previous region-feature-based methods, their performance on downstream tasks often degrades significantly. In this paper, we present METER, a Multimodal End-to-end TransformER framework, through which we investigate how to design and pre-train a fully transformer-based VL model in an end-to-end manner. Specifically, we dissect the model designs along multiple dimensions: vision encoders (e.g., CLIP-ViT, Swin transformer), text encoders (e.g., RoBERTa, DeBERTa), multimodal fusion module (e.g., merged attention vs. co-attention), architectural design (e.g., encoder-only vs. encoder-decoder), and pre-training objectives (e.g., masked image modeling). We conduct comprehensive experiments and provide insights on how to train a performant VL transformer. METER achieves an accuracy of 77.64% on the VQAv2 test-std set using only 4M images for pre-training, surpassing the state-of-the-art region-feature-based model by 1.04%, and outperforming the previous best fully transformer-based model by 1.6%. Notably, when further scaled up, our best VQA model achieves an accuracy of 80.54%. Code and pre-trained models are released at https://github.com/zdou0830/METER.

Zi-Yi Dou, Yichong Xu, Zhe Gan, Jianfeng Wang, Shuohang Wang, Lijuan Wang, Chenguang Zhu, Pengchuan Zhang, Lu Yuan, Nanyun Peng, Zicheng Liu, Michael Zeng• 2021

Related benchmarks

TaskDatasetResultRank
Image CaptioningMS COCO Karpathy (test)
CIDEr128.2
682
Visual Question AnsweringVQA v2 (test-dev)
Overall Accuracy80.33
664
Visual Question AnsweringVQA v2 (test-std)
Accuracy80.54
466
Image-to-Text RetrievalFlickr30K 1K (test)
R@182.2
439
Text-to-Image RetrievalFlickr30k (test)
Recall@182.22
423
Image ClassificationDTD
Accuracy62.2
419
Image-to-Text RetrievalFlickr30k (test)
R@194.3
370
Visual Question AnsweringVQA 2.0 (test-dev)
Accuracy77.7
337
Image ClassificationCIFAR100
Accuracy70.3
331
Natural Language Visual ReasoningNLVR2 (test-p)
Accuracy83.47
327
Showing 10 of 63 rows

Other info

Code

Follow for update