改进视觉语言模型的推理对齐方式,显著提升组合性理解能力。
Revisiting Compositionality in Dual-Encoder Vision-Language Models: The Role of Inference
- 在推理时引入细粒度区域对齐机制,无需微调预训练模型。
- 在域内检索上媲美全量微调,在域外组合任务中大幅领先。
- 适合关注模型泛化能力与推理机制优化的研究者。
双编码器视觉语言模型(如CLIP)常因组合性任务表现不佳而被视为“词袋系统”。我们发现,这一局限可能并非源于表征能力不足,而是标准推理协议依赖全局余弦相似度所致。通过受控诊断实验,我们证明在推理阶段显式强制细粒度区域-片段对齐,可显著提升组合性能,且无需更新预训练编码器。随后,我们提出一个轻量级Transformer,直接从冻结的图像块和文本标记嵌入中学习此类对齐。相比全量微调及先前端到端组合训练方法,尽管这些方法在域内检索中表现良好,但其增益在分布外场景下无法稳定转移。相反,基于冻结表征的局部对齐策略在域内检索上达到全量微调水平,同时在控制型域外组合基准上实现显著提升。结果表明,全局嵌入匹配是双编码器VLM的关键瓶颈,对齐机制对鲁棒组合泛化至关重要。
原文摘要 · Abstract (English)
Dual-encoder Vision-Language Models (VLMs) such as CLIP are often characterized as bag-of-words systems due to their poor performance on compositional benchmarks. We argue that this limitation may stem less from deficient representations than from the standard inference protocol based on global cosine similarity. First, through controlled diagnostic experiments, we show that explicitly enforcing fine-grained region-segment alignment at inference dramatically improves compositional performance without updating pretrained encoders. We then introduce a lightweight transformer that learns such alignments directly from frozen patch and token embeddings. Comparing against full fine-tuning and prior end-to-end compositional training methods, we find that although these approaches improve in-domain retrieval, their gains do not consistently transfer under distribution shift. In contrast, learning localized alignment over frozen representations matches full fine-tuning on in-domain retrieval while yielding substantial improvements on controlled out-of-domain compositional benchmarks. These results identify global embedding matching as a key bottleneck in dual-encoder VLMs and highlight the importance of alignment mechanisms for robust compositional generalization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。