arXiv:2501.11063cs.CV2025-01被引 7

通过分组筛选正样本,让带噪声的数据也能高效参与度量学习。

Enhancing Sample Utilization in Noise-Robust Deep Metric Learning With Subgroup-Based Positive-Pair Selection

  • 基于概率和子群信息,为噪声样本找到可靠正样本对。
  • 在多个含噪数据集上超越当前最佳方法,提升显著。
  • 适合图像检索与人脸识别等需要高鲁棒性的任务使用。

真实世界数据中的噪声标签会严重影响深度学习模型性能。尽管已有大量研究关注分类任务中的噪声鲁棒性,但深度度量学习(DML)中的噪声标签问题仍较少被探索。现有方法多直接丢弃可疑噪声样本,造成训练数据浪费。为此,我们提出一种基于子群的正样本选择框架(SGPS),通过构建噪声样本的可靠正样本对来提升样本利用率。SGPS首先利用基于概率的清洁样本选择策略识别干净与噪声样本;为进一步利用噪声样本,借助子群生成模块发现其潜在相似样本,并通过正原型生成模块为每个噪声样本聚合出信息丰富的正原型。随后设计新对比损失函数,用于优化噪声样本及其选定的正样本对。该方法可无缝集成至现有成对式DML训练流程中,如图像检索与人脸识别。在多个合成与真实大规模含噪数据集上的实验表明,无需额外技巧,本方法即超越当前最先进的噪声标签DML方法。代码已开源。

原文摘要 · Abstract (English)

The existence of noisy labels in real-world data negatively impacts the performance of deep learning models. Although much research effort has been devoted to improving the robustness towards noisy labels in classification tasks, the problem of noisy labels in deep metric learning (DML) remains under-explored. Existing noisy label learning methods designed for DML mainly discard suspicious noisy samples, resulting in a waste of the training data. To address this issue, we propose a noise-robust DML framework with SubGroup-based Positive-pair Selection (SGPS), which constructs reliable positive pairs for noisy samples to enhance the sample utilization. Specifically, SGPS first effectively identifies clean and noisy samples by a probability-based clean sample selectionstrategy. To further utilize the remaining noisy samples, we discover their potential similar samples based on the subgroup information given by a subgroup generation module and then aggregate them into informative positive prototypes for each noisy sample via a positive prototype generation module. Afterward, a new contrastive loss is tailored for the noisy samples with their selected positive pairs. SGPS can be easily integrated into the training process of existing pair-wise DML tasks, like image retrieval and face recognition. Extensive experiments on multiple synthetic and real-world large-scale label noise datasets demonstrate the effectiveness of our proposed method. Without any bells and whistles, our SGPS framework outperforms the state-of-the-art noisy label DML methods. Code is available at \url{https://github.com/smuelpeng/SGPS-NoiseFreeDML}.

度量学习噪声鲁棒正样本选择

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。