Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Accelerating Pre-training of Multimodal LLMs via Chain-of-Sight

About

This paper introduces Chain-of-Sight, a vision-language bridge module that accelerates the pre-training of Multimodal Large Language Models (MLLMs). Our approach employs a sequence of visual resamplers that capture visual details at various spacial scales. This architecture not only leverages global and local visual contexts effectively, but also facilitates the flexible extension of visual tokens through a compound token scaling strategy, allowing up to a 16x increase in the token count post pre-training. Consequently, Chain-of-Sight requires significantly fewer visual tokens in the pre-training phase compared to the fine-tuning phase. This intentional reduction of visual tokens during pre-training notably accelerates the pre-training process, cutting down the wall-clock training time by ~73%. Empirical results on a series of vision-language benchmarks reveal that the pre-train acceleration through Chain-of-Sight is achieved without sacrificing performance, matching or surpassing the standard pipeline of utilizing all visual tokens throughout the entire training process. Further scaling up the number of visual tokens for pre-training leads to stronger performances, competitive to existing approaches in a series of benchmarks.

Ziyuan Huang, Kaixiang Ji, Biao Gong, Zhiwu Qing, Qinglong Zhang, Kecheng Zheng, Jian Wang, Jingdong Chen, Ming Yang• 2024

Related benchmarks

TaskDatasetResultRank
Visual Question AnsweringVizWiz
Accuracy50.7
1525
Object Hallucination EvaluationPOPE--
1455
Visual Question AnsweringVQA v2
Accuracy82.9
1362
Visual Question AnsweringGQA
Accuracy64
1249
Multimodal EvaluationMME--
658
Referring Expression ComprehensionRefCOCO+ (val)
Accuracy86.03
354
Referring Expression ComprehensionRefCOCO (val)
Accuracy90.72
344
Referring Expression ComprehensionRefCOCO (testA)
Accuracy0.9383
342
Referring Expression ComprehensionRefCOCOg (test)
Accuracy87.94
300
Referring Expression ComprehensionRefCOCOg (val)
Accuracy87.46
300
Showing 10 of 18 rows

Other info

Code

Follow for update