arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2503.17080cs.CV

Seeing What Matters: 赋予CLIP Patch Generation-to-Selection能力

Seeing What Matters: Empowering CLIP with Patch Generation-to-Selection

Nanjing University of Science and Technology(南京理工大学);Zhejiang Sci-Tech University(浙江理工大学);Beijing Institute of Technology(北京理工大学)

Gensheng Pei, Tao Chen, Yujia Wang, Xinhao Cai, Xiangbo Shu, Tianfei Zhou, Yazhou Yao

AI总结:

针对CLIP训练计算量大、现有掩码策略易损失语义的问题,提出Patch Generation-to-Selection渐进式掩码方法,在保留语义的同时提升训练效率,在零样本分类等任务上取得SOTA结果。

摘要:

CLIP模型通过在图文对上进行大规模预训练,在视觉与语言模态对齐方面取得了显著进展,在多个领域展现出强大的零样本分类和检索能力。然而,CLIP的训练计算量仍然很大,对数据处理和内存都有很高要求。为应对这些挑战,近期出现了掩码策略,重点是选择性移除图像块以提升训练效率。这类方法虽有效,但往往会损失关键语义信息,导致视觉特征与文本描述的对齐效果欠佳。本研究提出了一种简洁有效的方法——Patch Generation-to-Selection(补丁生成-选择),在保留关键语义内容的同时提升CLIP的训练效率。该方法引入渐进式掩码流程:首先预选出少量候选图像块作为潜在掩码区域;接着对整幅图像应用Sobel边缘检测生成边缘掩码,优先保留主要目标区域;最后计算候选掩码块与其邻接块的相似度得分,通过最优传输归一化优化选择过程,确保相似度矩阵均衡。我们的方法CLIP-PGS在零样本分类和检索任务上刷新了当前最优(SOTA)结果,在鲁棒性评估和语言组合性基准测试中也表现出更优性能。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑