用边缘检测和相似性匹配,让CLIP训练更省力且不丢关键信息。
Seeing What Matters: Empowering CLIP with Patch Generation-to-Selection
- 先选候选区域,再用边缘检测保留主体部分
- 通过相似性计算与优化选择,提升特征对齐效果
- 适合追求高效训练且重视语义保真的研究者
CLIP模型通过大规模图像-文本对预训练,在跨模态对齐上表现优异,具备强大的零样本分类与检索能力。然而其训练仍需大量计算资源,对数据处理与内存要求高。近期掩码策略通过选择性移除图像块提升效率,但常损失关键语义信息,影响视觉与语言的对齐效果。本文提出一种简洁有效的方法——Patch Generation-to-Selection(CLIP-PGS),在保持关键语义内容的同时提升训练效率。该方法采用渐进式掩码机制:首先预选少量候选掩码块;接着利用Sobel边缘检测生成边缘掩码,优先保留主要物体区域;最后基于候选块与其邻域块的相似性得分,结合最优传输归一化优化选择过程,确保相似性矩阵平衡。实验表明,CLIP-PGS在零样本分类与检索任务中达到新最优性能,并在鲁棒性评估与语言组合性基准测试中表现优异。
原文摘要 · Abstract (English)
The CLIP model has demonstrated significant advancements in aligning visual and language modalities through large-scale pre-training on image-text pairs, enabling strong zero-shot classification and retrieval capabilities on various domains. However, CLIP's training remains computationally intensive, with high demands on both data processing and memory. To address these challenges, recent masking strategies have emerged, focusing on the selective removal of image patches to improve training efficiency. Although effective, these methods often compromise key semantic information, resulting in suboptimal alignment between visual features and text descriptions. In this work, we present a concise yet effective approach called Patch Generation-to-Selection to enhance CLIP's training efficiency while preserving critical semantic content. Our method introduces a gradual masking process in which a small set of candidate patches is first pre-selected as potential mask regions. Then, we apply Sobel edge detection across the entire image to generate an edge mask that prioritizes the retention of the primary object areas. Finally, similarity scores between the candidate mask patches and their neighboring patches are computed, with optimal transport normalization refining the selection process to ensure a balanced similarity matrix. Our approach, CLIP-PGS, sets new state-of-the-art results in zero-shot classification and retrieval tasks, achieving superior performance in robustness evaluation and language compositionality benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。