通过跨模态自蒸馏,让视觉语言模型更关注图像细节而非仅前景。
COSMOS: Cross-Modality Self-Distillation for Vision Language Pre-training

- 引入文本裁剪与交叉注意力,实现多模态自蒸馏。
- 在零样本任务中超越已有基线,部分性能媲美更大规模模型。
- 适合需要细粒度理解的视觉语言任务研究者。
基于对比损失训练的视觉语言模型(VLMs)在各类视觉与语言任务中取得显著进展。然而,对比损失的全局特性导致模型过度关注前景物体,忽视图像中其他重要信息,限制了其在下游任务中的表现。为此,我们提出COSMOS:跨模态自蒸馏的视觉语言预训练方法,将新颖的文本裁剪策略与交叉注意力模块融入自监督学习框架。通过构建图像与文本的全局和局部视图(即多模态增强),实现VLM中的自蒸馏。进一步引入交叉注意力模块,使COSMOS能通过跨模态自蒸馏损失学习全面的跨模态表示。COSMOS在多个零样本下游任务(包括检索、分类、语义分割)中持续优于先前强基线,并在视觉感知与上下文理解任务中超越基于CLIP且在更大数据集上训练的模型。代码已开源。
原文摘要 · Abstract (English)
Vision-Language Models (VLMs) trained with contrastive loss have achieved significant advancements in various vision and language tasks. However, the global nature of the contrastive loss makes VLMs focus predominantly on foreground objects, neglecting other crucial information in the image, which limits their effectiveness in downstream tasks. To address these challenges, we propose COSMOS: CrOSs-MOdality Self-distillation for vision-language pre-training that integrates a novel text-cropping strategy and cross-attention module into a self-supervised learning framework. We create global and local views of images and texts (i.e., multi-modal augmentations), which are essential for self-distillation in VLMs. We further introduce a cross-attention module, enabling COSMOS to learn comprehensive cross-modal representations optimized via a cross-modality self-distillation loss. COSMOS consistently outperforms previous strong baselines on various zero-shot downstream tasks, including retrieval, classification, and semantic segmentation. Additionally, it surpasses CLIP-based models trained on larger datasets in visual perception and contextual understanding tasks. Code is available at https://github.com/ExplainableML/cosmos.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。