用大模型提升CLIP长文本理解能力,同时不破坏原有对齐
ProCLIP: Progressive Vision-Language Alignment via LLM-based Embedder
- 分阶段对齐:先用CLIP知识蒸馏建立初始对齐,再逐步优化
- 解决大模型与CLIP视觉-语言空间不匹配问题,避免破坏预训练知识
- 适合需要长文本、多语言理解的视觉-语言任务研究者
原始CLIP文本编码器最大输入长度为77个标记,限制了其处理长文本和细粒度语义理解的能力,且不支持多语言输入。现有方法尝试用基于大语言模型(LLM)的嵌入器替代原编码器以增强长文本与多语言理解能力。然而,由于大模型与CLIP的视觉-语言空间独立预训练,缺乏对齐先验,直接使用对比学习进行对齐会破坏CLIP图像编码器原有的内在对齐关系,导致预训练知识未被充分利用。为此,我们提出ProCLIP,一种基于课程学习的渐进式视觉-语言对齐框架,将CLIP图像编码器与基于大模型的嵌入器有效对齐。具体而言,ProCLIP首先将CLIP文本编码器的知识蒸馏至大模型嵌入器中,利用其丰富的预训练知识并建立初始对齐;随后通过图像-文本对比微调进一步对齐,采用自蒸馏正则化防止过拟合。在表示继承与对比微调过程中,引入实例语义对齐损失与嵌入结构对齐损失,实现更有效的对齐。代码已公开于https://github.com/VisionXLab/ProCLIP。
原文摘要 · Abstract (English)
The original CLIP text encoder is limited by a maximum input length of 77 tokens, which hampers its ability to effectively process long texts and perform fine-grained semantic understanding. In addition, the CLIP text encoder lacks support for multilingual inputs. All these limitations significantly restrict its applicability across a broader range of tasks. Recent studies have attempted to replace the CLIP text encoder with an LLM-based embedder to enhance its ability in processing long texts, multilingual understanding, and fine-grained semantic comprehension. However, because the representation spaces of LLMs and the vision-language space of CLIP are pretrained independently without alignment priors, direct alignment using contrastive learning can disrupt the intrinsic vision-language alignment in the CLIP image encoder, leading to an underutilization of the knowledge acquired during pre-training. To address this challenge, we propose ProCLIP, a curriculum learning-based progressive vision-language alignment framework to effectively align the CLIP image encoder with an LLM-based embedder. Specifically, ProCLIP first distills knowledge from CLIP's text encoder into the LLM-based embedder to leverage CLIP's rich pretrained knowledge while establishing initial alignment between the LLM embedder and CLIP image encoder. Subsequently, ProCLIP further aligns the CLIP image encoder with the LLM-based embedder through image-text contrastive tuning, employing self-distillation regularization to avoid overfitting. To achieve a more effective alignment, instance semantic alignment loss and embedding structure alignment loss are employed during representation inheritance and contrastive tuning. The Code is available at https://github.com/VisionXLab/ProCLIP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。