Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Expedited Training of Visual Conditioned Language Generation via Redundancy Reduction

About

In this paper, we introduce $\text{EVL}_{\text{Gen}}$, a streamlined framework designed for the pre-training of visually conditioned language generation models with high computational demands, utilizing frozen pre-trained large language models (LLMs). The conventional approach in vision-language pre-training (VLP) typically involves a two-stage optimization process: an initial resource-intensive phase dedicated to general-purpose vision-language representation learning, focused on extracting and consolidating relevant visual features. This is followed by a subsequent phase that emphasizes end-to-end alignment between visual and linguistic modalities. Our novel one-stage, single-loss framework bypasses the computationally demanding first training stage by gradually merging similar visual tokens during training, while avoiding model collapse caused by single-stage training of BLIP-2 type models. The gradual merging process effectively condenses visual information while preserving semantic richness, resulting in rapid convergence without compromising performance. Our experimental findings demonstrate that our approach accelerates the training of vision-language models by a factor of 5 without a noticeable impact on overall performance. Furthermore, we illustrate that our models significantly narrow the performance gap to current vision-language models using only 1/10 of the data. Finally, we showcase how our image-text models can seamlessly adapt to video-conditioned language generation tasks through novel soft attentive temporal token contextualizing modules. Code is available at \url{https://github.com/yiren-jian/EVLGen}.

Yiren Jian, Tingkai Liu, Yunzhe Tao, Chunhui Zhang, Soroush Vosoughi, Hongxia Yang• 2023

Related benchmarks

TaskDatasetResultRank
Image CaptioningMS COCO Karpathy (test)
CIDEr139.1
682
Visual Question AnsweringOK-VQA (test)
Accuracy27.2
296
Visual Question AnsweringOKVQA
Top-1 Accuracy30.6
283
Visual Question AnsweringGQA (test-dev)
Accuracy36.3
178
Video CaptioningMSR-VTT (test)
CIDEr74
121
Video CaptioningMSVD (test)
CIDEr158.2
111
Visual Question AnsweringVQA v2 (val)
Accuracy48.4
99
Visual Question AnsweringVQAv2 (test-dev)
Accuracy55.5
76
Image CaptioningFlickr30K zero-shot
CIDEr82
11
Image CaptioningMSCOCO (val)
CIDEr139.1
5
Showing 10 of 11 rows

Other info

Code

Follow for update