计算机系统应用2024,Vol.33Issue(2) :54-61.DOI:10.15888/j.cnki.csa.009396

基于样本选择的标签含噪图像分类

Label Noisy Image Classification Based on Sample Selection

闻铮 曹国
计算机系统应用2024,Vol.33Issue(2) :54-61.DOI:10.15888/j.cnki.csa.009396

基于样本选择的标签含噪图像分类

Label Noisy Image Classification Based on Sample Selection

闻铮 1曹国1
扫码查看

作者信息

  • 1. 南京理工大学计算机科学与工程学院,南京 210094
  • 折叠

摘要

标签噪声广泛存在、无法避免且影响深度网络模型的性能.利用神经网络的"记忆效应",基于小损失原则的样本选择方法能简单有效地处理标签噪声.本文基于特征空间中样本距离越近越相似的原则,结合样本的高低置信度假设,提出了新的样本选择原则以及二阶段加权样本选择重标签方法(WSSR-2s).(1)在训练前期阶段,对于高置信度样本,在特征空间中对其票权进行加权,更好地引导训练;(2)在训练中后期阶段,对于低置信度样本,将其票权转移给其最相似的特征样本,以更正确地训练.在合成噪声数据集CIFAR-10、CIFAR-100 以及真实噪声数据集ANIMAL-10N、WebVision的实验结果表明,本文提出的方法取得更高的精度,能够更好地处理标签噪声问题.

Abstract

Label noise is widely present and unavoidable,and it affects the performance of deep network models.Sample selection methods based on the principle of small loss can easily and effectively handle label noise by the"memory effect"of neural networks.This study proposes a new sample selection principle and a two-stage weighted sample selection and relabeling method(WSSR-2s)based on the principle that a closer sample distance in the feature space results in more similarity,combined with the assumption of high and low confidence of the samples.In the early training stage,for high-confidence samples,their voting rights are weighted in the feature space to better guide training.In the middle and later stages of training,for low-confidence samples,their voting rights are transferred to their most similar feature samples for more accurate training.The experimental results on synthetic noise datasets CIFAR-10 and CIFAR-100,as well as real noise datasets ANIMAL-10N and WebVision,show that the proposed method achieves higher accuracy and can better handle label noise problems.

关键词

标签噪声/样本选择/置信度假说/样本票权/样本加权

Key words

label noise/sample selection/confidence hypothesis/sample voting right/sample weighting

引用本文复制引用

基金项目

国家自然科学基金(62201282)

江苏省自然科学基金(BK20231456)

出版年

2024
计算机系统应用
中国科学院软件研究所

计算机系统应用

CSTPCD
影响因子:0.449
ISSN:1003-3254
参考文献量3
段落导航相关论文