无需额外标注,让CLIP自动对齐图像区域与文本词组,提升细粒度理解能力。
Fine-grained CLIP fine-tuning with self-annotated region alignment

- 运行时自动生成区域-词组对齐,仅用图文对训练
- 在细粒度任务上显著提升性能,同时保持图像级任务表现
- 适合需要细粒度视觉理解但无标注数据的场景
对比语言-图像预训练(CLIP)在细粒度密集特征表示方面存在局限,因其预训练聚焦于整体图像与文本描述的匹配。为避免从零开始训练视觉语言模型带来的巨大数据与计算开销,已有研究尝试通过微调方案增强CLIP的细粒度能力。然而现有方法通常依赖额外的区域标注,限制了语义多样性并增加数据处理成本,且常牺牲CLIP原有的全局视觉表征能力。为此,我们提出SFF-CLIP(自标注细粒度微调),仅使用图像-文本对即可提升CLIP在细粒度表示上的能力,同时保持其原始全局视觉-语义一致性。具体而言,设计了一种运行时区域-短语对齐机制,从输入句子中提取概念词组,并利用文本特定热图将它们与提取的基于区域的特征进行对齐。大量实验表明,SFF-CLIP在细粒度密集特征表示上实现显著性能提升,同时维持原CLIP在图像级任务上的表现。代码将在后续发布。
原文摘要 · Abstract (English)
Contrastive Language-Image Pre-training (CLIP) has been shown to have limitations in its fine-grained dense feature representation, due to its pre-training focusing on matching the whole image to a text description. Considering the large data and computational burden in pre-training a vision-language model from scratch, a series of works aim to enhance the fine-grained ability of CLIP through a fine-tuning scheme. However, existing works suffer from a variety of limitations: additional region annotations are usually required, which limits the semantic diversity due to the predefined categories and leads to a large effort to process the training data; and they usually sacrifice CLIP's original ability for global visual representation. To bypass these limitations, we propose SFF-CLIP (Self-annotated Fine-grained Fine-tuning for CLIP), which only uses image-text pairs as input to boost the fine-grained representation ability in the CLIP fine-tuning, while maintaining the global visual-semantic consistency. Concretely, a run-time region-phrase alignment scheme is designed, which obtains concept phrases from the input sentence, and aligns them with corresponding extracted region-based features using text-specific heat maps. Extensive experiments demonstrate that SFF-CLIP leads to significant performance improvements on fine-grained dense feature representation, as well as maintaining the performance of the original CLIP on image-level tasks. Code will be released later.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。