HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Toward Building General Foundation Models for Language, Vision, and Vision-Language Understanding Tasks

Xinsong Zhang Yan Zeng Jipeng Zhang Hang Li

Toward Building General Foundation Models for Language, Vision, and Vision-Language Understanding Tasks

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

zhangxinsong-nlp/XFM
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
cross-modal-retrieval-on-coco-2014XFM (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-bXFM (base)
Accuracy (%): 79.8
visual-grounding-on-refcoco-testaXFM (base)
Accuracy (%): 90.4
visual-grounding-on-refcoco-valXFM (base)
Accuracy (%): 86.1
visual-question-answering-on-vqa-v2-test-devXFM (base)
Accuracy: 80.4
visual-reasoning-on-nlvr2-devXFM (base)
Accuracy: 87.6
visual-reasoning-on-nlvr2-testXFM (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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp