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

E2E-VLP: End-to-End Vision-Language Pre-training Enhanced by Visual Learning

About

Vision-language pre-training (VLP) on large-scale image-text pairs has achieved huge success for the cross-modal downstream tasks. The most existing pre-training methods mainly adopt a two-step training procedure, which firstly employs a pre-trained object detector to extract region-based visual features, then concatenates the image representation and text embedding as the input of Transformer to train. However, these methods face problems of using task-specific visual representation of the specific object detector for generic cross-modal understanding, and the computation inefficiency of two-stage pipeline. In this paper, we propose the first end-to-end vision-language pre-trained model for both V+L understanding and generation, namely E2E-VLP, where we build a unified Transformer framework to jointly learn visual representation, and semantic alignments between image and text. We incorporate the tasks of object detection and image captioning into pre-training with a unified Transformer encoder-decoder architecture for enhancing visual learning. An extensive set of experiments have been conducted on well-established vision-language downstream tasks to demonstrate the effectiveness of this novel VLP paradigm.

Haiyang Xu, Ming Yan, Chenliang Li, Bin Bi, Songfang Huang, Wenming Xiao, Fei Huang• 2021

Related benchmarks

TaskDatasetResultRank
Image CaptioningMS COCO Karpathy (test)
CIDEr1.173
682
Visual Question AnsweringVQA v2 (test-dev)
Overall Accuracy73.25
664
Visual Question AnsweringVQA v2 (test-std)
Accuracy73.67
466
Image-to-Text RetrievalFlickr30K 1K (test)
R@186.2
439
Text-to-Image RetrievalFlickr30K 1K (test)
R@173.6
375
Visual Question AnsweringVQA 2.0 (test-dev)
Accuracy73.25
337
Natural Language Visual ReasoningNLVR2 (test-p)
Accuracy77.96
327
Natural Language Visual ReasoningNLVR2 (dev)
Accuracy77.25
288
Image RetrievalFlickr30k (test)
R@173.58
195
Visual Question AnsweringVQA (test-dev)
Acc (All)73.25
147
Showing 10 of 18 rows

Other info

Follow for update