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

By Zi-Yi Dou, Yichong Xu, Zhe Gan, Jianfeng Wang, Shuohang Wang, Lijuan Wang, Chenguang Zhu, Nanyun Peng, Zicheng Liu, Michael Zeng. 2021

462 citations recorded by the source corpus.

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, De-BERTa), 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. Meterachieves 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.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Multimodal Learning, Model Architectures

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Jianfeng Gao
  2. Zhe Gan
  3. Mohit Bansal
  4. Lijuan Wang
  5. Furu Wei
  6. Zicheng Liu
  7. Chunyuan Li
  8. Jianwei Yang

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Dacheng Tao
  3. Jianfeng Gao
  4. Xiaogang Wang
  5. Trevor Darrell
  6. Cordelia Schmid
  7. Ming-Hsuan Yang
  8. Jitendra Malik

In Multimodal Learning

  1. Zhe Gan
  2. Lijuan Wang
  3. Jianfeng Gao
  4. Zicheng Liu
  5. Linjie Li
  6. Jianfeng Wang
  7. Xiaowei Hu
  8. Jianwei Yang

In Model Architectures

  1. Jianfeng Gao
  2. Jianwei Yang
  3. Lu Yuan
  4. Chunyuan Li
  5. Bin Xiao
  6. Xiyang Dai
  7. Pengchuan Zhang
  8. Yu Qiao

Citations in the corpus