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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Captioning | MS COCO Karpathy (test) | CIDEr1.173 | 682 | |
| Visual Question Answering | VQA v2 (test-dev) | Overall Accuracy73.25 | 664 | |
| Visual Question Answering | VQA v2 (test-std) | Accuracy73.67 | 466 | |
| Image-to-Text Retrieval | Flickr30K 1K (test) | R@186.2 | 439 | |
| Text-to-Image Retrieval | Flickr30K 1K (test) | R@173.6 | 375 | |
| Visual Question Answering | VQA 2.0 (test-dev) | Accuracy73.25 | 337 | |
| Natural Language Visual Reasoning | NLVR2 (test-p) | Accuracy77.96 | 327 | |
| Natural Language Visual Reasoning | NLVR2 (dev) | Accuracy77.25 | 288 | |
| Image Retrieval | Flickr30k (test) | R@173.58 | 195 | |
| Visual Question Answering | VQA (test-dev) | Acc (All)73.25 | 147 |