Effective End-to-End Vision Language Pretraining with Semantic Visual Loss
This addresses the speed bottleneck for vision-language models in real-world applications, though it is incremental as it builds on existing end-to-end methods.
The paper tackles the problem of slow inference in vision-language pretraining by introducing auxiliary visual losses to train end-to-end models from raw pixels, achieving similar or better performance than region-feature models while running over 10 times faster and reducing pretraining GPU hours by 90%.
Current vision language pretraining models are dominated by methods using region visual features extracted from object detectors. Given their good performance, the extract-then-process pipeline significantly restricts the inference speed and therefore limits their real-world use cases. However, training vision language models from raw image pixels is difficult, as the raw image pixels give much less prior knowledge than region features. In this paper, we systematically study how to leverage auxiliary visual pretraining tasks to help training end-to-end vision language models. We introduce three types of visual losses that enable much faster convergence and better finetuning accuracy. Compared with region feature models, our end-to-end models could achieve similar or better performance on downstream tasks and run more than 10 times faster during inference. Compared with other end-to-end models, our proposed method could achieve similar or better performance when pretrained for only 10% of the pretraining GPU hours.