Assessing and Learning Alignment of Unimodal Vision and Language Models
Journal:
arXiv
Published Date:
Dec 5, 2024
Abstract
How well are unimodal vision and language models aligned? Although prior work
have approached answering this question, their assessment methods do not
directly translate to how these models are used in practical vision-language
tasks. In this paper, we propose a direct assessment method, inspired by linear
probing, to assess vision-language alignment. We identify that the degree of
alignment of the SSL vision models depends on their SSL training objective, and
we find that the clustering quality of SSL representations has a stronger
impact on alignment performance than their linear separability. Next, we
introduce Swift Alignment of Image and Language (SAIL), a efficient transfer
learning framework that aligns pretrained unimodal vision and language models
for downstream vision-language tasks. Since SAIL leverages the strengths of
pretrained unimodal models, it requires significantly fewer (6%) paired
image-text data for the multimodal alignment compared to models like CLIP which
are trained from scratch. SAIL training only requires a single A100 GPU, 5
hours of training and can accommodate a batch size up to 32,768. SAIL achieves
73.4% zero-shot accuracy on ImageNet (vs. CLIP's 72.7%) and excels in zero-shot
retrieval, complex reasoning, and semantic segmentation. Additionally, SAIL
improves the language-compatibility of vision encoders that in turn enhance the
performance of multimodal large language models. The entire codebase and model
weights are open-source: https://lezhang7.github.io/sail.github.io/