Command Palette
Search for a command to run...
Toward Building General Foundation Models for Language, Vision, and Vision-Language Understanding Tasks
Xinsong Zhang Yan Zeng Jipeng Zhang Hang Li

Abstract
Foundation models or pre-trained models have substantially improved the performance of various language, vision, and vision-language understanding tasks. However, existing foundation models can only perform the best in one type of tasks, namely language, vision, or vision-language. It is still an open question whether it is possible to construct a foundation model performing the best for all the understanding tasks, which we call a general foundation model. In this paper, we propose a new general foundation model, X-FM (the X-Foundation Model). X-FM has one language encoder, one vision encoder, and one fusion encoder, as well as a new training method. The training method includes two new techniques for learning X-FM from text, image, and image-text pair data. One is to stop gradients from the vision-language training when learning the language encoder. The other is to leverage the vision-language training to guide the learning of the vision encoder. Extensive experiments on benchmark datasets show that X-FM can significantly outperform existing general foundation models and perform better than or comparable to existing foundation models specifically for language, vision, or vision-language understanding. Code and pre-trained models are released at https://github.com/zhangxinsong-nlp/XFM.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| cross-modal-retrieval-on-coco-2014 | XFM (base) | Image-to-text R@1: 84.2 Image-to-text R@10: 98.4 Image-to-text R@5: 96.4 Text-to-image R@1: 67.0 Text-to-image R@10: 92.4 Text-to-image R@5: 87.2 |
| visual-grounding-on-refcoco-test-b | XFM (base) | Accuracy (%): 79.8 |
| visual-grounding-on-refcoco-testa | XFM (base) | Accuracy (%): 90.4 |
| visual-grounding-on-refcoco-val | XFM (base) | Accuracy (%): 86.1 |
| visual-question-answering-on-vqa-v2-test-dev | XFM (base) | Accuracy: 80.4 |
| visual-reasoning-on-nlvr2-dev | XFM (base) | Accuracy: 87.6 |
| visual-reasoning-on-nlvr2-test | XFM (base) | Accuracy: 88.4 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.