Robust classification with noisy labels using Venn-Abers predictors

被引:0
作者
Lemghari, Ichraq [1 ,2 ]
Le Hegarat-Mascle, Sylvie [1 ]
Aldea, Emanuel [1 ]
Vandoni, Jennifer [2 ]
机构
[1] Paris Saclay Univ, SATIE, Gif Sur Yvette, France
[2] Safran Tech, Digital Sci & Technol Dept, Magny Les Hameaux, France
关键词
supervised learning; classification; noisy labels; set-valued decisions; partial ignorance modeling; Venn-Abers predictors;
D O I
10.1117/1.JEI.33.3.031210
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
. The advent of deep learning methods has led to impressive advances in computer vision tasks over the past decades, largely due to their ability to extract non-linear features that are well adapted to the task at hand. For supervised approaches, data labeling is essential to achieve a high level of performance; however, this task can be so fastidious or even troublesome in difficult contexts (e.g., specific defect detection, unconventional data annotations, etc.) that experts can sometimes erroneously provide the wrong ground truth label. Considering classification problems, this paper addresses the issue of handling noisy labels in datasets. Specifically, we first detect the noisy samples of a dataset using set-valued labels and then improve their classification using Venn-Abers predictors. The obtained results reach more than 0.99 and 0.90 accuracy for noisified versions of two widely used image classification datasets, digit MNIST and CIFAR-10 respectively with a 40% two-class pair-flip noise ratio and 0.87 accuracy for CIFAR-10 with 10-class uniform 40% noise ratio.
引用
收藏
页数:19
相关论文
共 50 条
[11]  
Ghosh A, 2017, AAAI CONF ARTIF INTE, P1919
[12]  
Goldberger Jacob, 2016, INT C LEARN REPR
[13]  
Guettari N, 2016, IEEE IMAGE PROC, P2742, DOI 10.1109/ICIP.2016.7532858
[14]   Co-teaching: Robust Training of Deep Neural Networks with Extremely Noisy Labels [J].
Han, Bo ;
Yao, Quanming ;
Yu, Xingrui ;
Niu, Gang ;
Xu, Miao ;
Hu, Weihua ;
Tsang, Ivor W. ;
Sugiyama, Masashi .
ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 31 (NIPS 2018), 2018, 31
[15]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[16]  
Hendrycks D, 2018, ADV NEUR IN, V31
[17]  
Hurwicz L., 1952, 355 COWL COMM
[18]  
Goodfellow IJ, 2015, Arxiv, DOI arXiv:1412.6572
[19]   Deep Bilevel Learning [J].
Jenni, Simon ;
Favaro, Paolo .
COMPUTER VISION - ECCV 2018, PT X, 2018, 11214 :632-648
[20]  
Johansson U, 2019, Data Min, P28