让视觉语言模型在细粒度检索中表现更好,同时不丢失通用能力。
Infusing fine-grained visual knowledge to Vision-Language Models
- 通过持续学习思路设计正则化策略,平衡领域适应与知识保留
- 细粒度检索性能显著提升,且无需文本数据即可保持图文对齐
- 强调验证集设计与超参调优,提升方法可复现性与泛化能力
大规模对比预训练使视觉语言模型(VLM)能生成适用于多种视觉与多模态任务的表征。然而,这些预训练表征在细粒度开放集视觉检索任务中仍表现不佳,当前最优结果需使用标注的特定领域样本微调视觉编码器。直接微调常导致灾难性遗忘,严重削弱模型的通用视觉与跨模态能力。本文提出一种专门设计的微调方法,在细粒度领域适应与保留预训练模型的广泛多模态知识之间实现最优平衡。受持续学习研究启发,系统分析了标准正则化技术,并提出高效有效的组合策略。此外,还关注常被忽视但关键的验证集设计与超参数调优问题,以确保跨数据集和预训练模型的可复现性与鲁棒泛化。我们在细粒度与粗粒度图像-图像、图像-文本检索基准上进行了广泛评估,所提方法持续取得优异结果,尤其在微调过程中不使用任何文本数据或原始文本编码器的情况下,仍能保持良好的视觉-文本对齐。代码与模型检查点:https://github.com/nikosips/infusing
原文摘要 · Abstract (English)
Large-scale contrastive pre-training produces powerful Vision-and-Language Models (VLMs) capable of generating representations (embeddings) effective for a wide variety of visual and multimodal tasks. However, these pretrained embeddings remain suboptimal for fine-grained open-set visual retrieval, where state-of-the-art results require fine-tuning the vision encoder using annotated domain-specific samples. Naively performing such fine-tuning typically leads to catastrophic forgetting, severely diminishing the model's general-purpose visual and cross-modal capabilities. In this work, we propose a fine-tuning method explicitly designed to achieve optimal balance between fine-grained domain adaptation and retention of the pretrained VLM's broad multimodal knowledge. Drawing inspiration from continual learning literature, we systematically analyze standard regularization techniques aimed at knowledge retention and propose an efficient and effective combination strategy. Additionally, we address the commonly overlooked yet critical aspects of validation set design and hyperparameter tuning to ensure reproducibility and robust generalization across datasets and pretrained models. We extensively evaluate our method on both fine-grained and coarse-grained image-image and image-text retrieval benchmarks. Our approach consistently achieves strong results, notably retaining the visual-text alignment without utilizing any text data or the original text encoder during fine-tuning. Code and model checkpoints: https://github.com/nikosips/infusing .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。