CCT: A Cyclic Co-Teaching Approach to Train Deep Neural Networks With Noisy Labels

被引:0
作者
Kamassury, Jorge K. S. [1 ,2 ]
Pickler, Henrique [1 ]
Cordeiro, Filipe R. [3 ]
Silva, Danilo [1 ]
机构
[1] Fed Univ Santa Catarina UFSC, Dept Elect & Elect Engn, BR-88040900 Florianopolis, SC, Brazil
[2] SENAI Inst Innovat Embedded Syst ISI SE, BR-88054700 Florianopolis, SC, Brazil
[3] Fed Rural Univ Pernambuco UFRPE, Dept Comp, Visual Comp Lab, BR-52171900 Recife, PE, Brazil
关键词
Training; Noise measurement; Noise; Optimization; Overfitting; Computational modeling; Artificial neural networks; Annotations; Robustness; Noise robustness; Co-teaching; deep neural networks; cyclic training; cyclic sample retention rate; learning with noisy labels;
D O I
10.1109/ACCESS.2025.3548510
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The detrimental impact of noisy labels on the generalization performance of deep neural networks has sparked research interest in learning with noisy labels (LNL). Among the various methods proposed to mitigate this effect, the Co-Teaching method, characterized by co-training with the small-loss criterion, is one of the most established approaches and is widely employed as a key component in recent LNL methods. Although Co-Teaching can mitigate the overfitting effect, it still remains, especially in scenarios with high rates of label noise in datasets. Strategies from the LNL literature to address this typically include the use of disagreement techniques and alternative loss functions. In this paper, we propose the Cyclic Co-Teaching (CCT) method, which employs cyclic variations in the learning rate and sample retention rate at the mini-batch level, along with a checkpoint mechanism that ensures that training in subsequent cycles always resumes from the best models obtained so far. For optimizing the method, we developed a framework that incorporates a pre-training phase to obtain an optimized vanilla model used to initialize CCT model weights, and a transparent univariate optimization strategy for hyperparameters that does not necessarily require a clean validation set. Experimental results on synthetic and real-world datasets, under different types and levels of noise and employing various neural network architectures, demonstrate that CCT outperforms several state-of-the-art LNL methods in most evaluated scenarios.
引用
收藏
页码:43843 / 43860
页数:18
相关论文
共 53 条
[1]  
Berthelot D, 2019, ADV NEUR IN, V32
[2]  
Bossard L, 2014, LECT NOTES COMPUT SC, V8694, P446, DOI 10.1007/978-3-319-10599-4_29
[3]  
Chang HS, 2017, ADV NEUR IN, V30
[4]   LongReMix: Robust learning with high confidence samples in a noisy label environment [J].
Cordeiro, Filipe R. ;
Sachdeva, Ragav ;
Belagiannis, Vasileios ;
Reid, Ian ;
Carneiro, Gustavo .
PATTERN RECOGNITION, 2023, 133
[5]   AutoAugment: Learning Augmentation Strategies from Data [J].
Cubuk, Ekin D. ;
Zoph, Barret ;
Mane, Dandelion ;
Vasudevan, Vijay ;
Le, Quoc V. .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :113-123
[6]  
Dauphin Y. N., 2017, P INT C LEARN REPR I
[7]   Classification in the Presence of Label Noise: a Survey [J].
Frenay, Benoit ;
Verleysen, Michel .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2014, 25 (05) :845-869
[8]  
Goodfellow I, 2016, ADAPT COMPUT MACH LE, P1
[9]   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
[10]  
Han Bo., 2020, arXiv