解决图文人物检索中的标签噪声与随机掩码丢失关键信息问题
AMNS: Attention-Weighted Selective Mask and Noise Label Suppression for Text-to-Image Person Retrieval
- 用双向相似度匹配损失缓解标签噪声影响
- 提出注意力加权选择性掩码,保留文本相关语义特征
- 适合处理标注不准确或图像质量差的场景
现有文本到图像的人物检索方法通常假设图像-文本对完全对齐,但实际中因图像质量差和标注错误存在噪声对应(NC)问题。此外,随机掩码增强可能无意中丢弃关键语义内容,导致图像与文本描述产生错误匹配。为此,本文提出一种噪声标签抑制方法以缓解NC,并设计注意力加权选择性掩码(AWM)策略应对随机掩码带来的问题。具体而言,双向相似度分布匹配(BSDM)损失使模型能有效学习正样本对,同时避免过度依赖它们,从而降低对噪声标签的过拟合风险;结合权重调整焦点(WAF)损失,提升模型处理困难样本的能力。此外,AWM通过图像编码器的EMA版本处理原始图像,有选择地保留与文本具有强语义关联的视觉令牌,实现更优特征提取。大量实验证明,该方法在解决噪声问题和提升检索性能方面效果显著。
原文摘要 · Abstract (English)
Most existing text-to-image person retrieval methods usually assume that the training image-text pairs are perfectly aligned; however, the noisy correspondence(NC) issue (i.e., incorrect or unreliable alignment) exists due to poor image quality and labeling errors. Additionally, random masking augmentation may inadvertently discard critical semantic content, introducing noisy matches between images and text descriptions. To address the above two challenges, we propose a noise label suppression method to mitigate NC and an Attention-Weighted Selective Mask (AWM) strategy to resolve the issues caused by random masking. Specifically, the Bidirectional Similarity Distribution Matching (BSDM) loss enables the model to effectively learn from positive pairs while preventing it from over-relying on them, thereby mitigating the risk of overfitting to noisy labels. In conjunction with this, Weight Adjustment Focal (WAF) loss improves the model's ability to handle hard samples. Furthermore, AWM processes raw images through an EMA version of the image encoder, selectively retaining tokens with strong semantic connections to the text, enabling better feature extraction. Extensive experiments demonstrate the effectiveness of our approach in addressing noise-related issues and improving retrieval performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。