OmniVL:One Foundation Model for Image-Language and Video-Language Tasks
About
This paper presents OmniVL, a new foundation model to support both image-language and video-language tasks using one universal architecture. It adopts a unified transformer-based visual encoder for both image and video inputs, and thus can perform joint image-language and video-language pretraining. We demonstrate, for the first time, such a paradigm benefits both image and video tasks, as opposed to the conventional one-directional transfer (e.g., use image-language to help video-language). To this end, we propose a decoupled joint pretraining of image-language and video-language to effectively decompose the vision-language modeling into spatial and temporal dimensions and obtain performance boost on both image and video tasks. Moreover, we introduce a novel unified vision-language contrastive (UniVLC) loss to leverage image-text, video-text, image-label (e.g., image classification), video-label (e.g., video action recognition) data together, so that both supervised and noisily supervised pretraining data are utilized as much as possible. Without incurring extra task-specific adaptors, OmniVL can simultaneously support visual only tasks (e.g., image classification, video action recognition), cross-modal alignment tasks (e.g., image/video-text retrieval), and multi-modal understanding and generation tasks (e.g., image/video question answering, captioning). We evaluate OmniVL on a wide range of downstream tasks and achieve state-of-the-art or competitive results with similar model size and data scale.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Captioning | MS COCO Karpathy (test) | CIDEr133.9 | 682 | |
| Visual Question Answering | VQA v2 (test-dev) | Overall Accuracy78.33 | 664 | |
| Video Question Answering | MSRVTT-QA | Accuracy44.1 | 481 | |
| Visual Question Answering | VQA v2 (test-std) | Accuracy78.4 | 466 | |
| Image-to-Text Retrieval | Flickr30K 1K (test) | R@197.3 | 439 | |
| Image Classification | DTD | Accuracy76.5 | 419 | |
| Text-to-Video Retrieval | DiDeMo (test) | R@152.4 | 376 | |
| Text-to-Image Retrieval | Flickr30K 1K (test) | R@187.9 | 375 | |
| Video Question Answering | MSRVTT-QA (test) | Accuracy44.1 | 371 | |
| Text-to-Video Retrieval | DiDeMo | R@10.524 | 360 |