Back to Search Start Over

Cross-to-merge training with class balance strategy for learning with noisy labels.

Authors :
Zhang, Qian
Zhu, Yi
Yang, Ming
Jin, Ge
Zhu, YingWen
Chen, Qiu
Source :
Expert Systems with Applications. Sep2024:Part C, Vol. 249, pN.PAG-N.PAG. 1p.
Publication Year :
2024

Abstract

• A pioneering framework called Cross-to-Merge Training (C2MT) is proposed for learning with noisy labels (LNL). • We are the first to perform parameter aggregation on LNL after cross-training. • We introduce a novel class-balancing strategy, the Median Balance Strategy (MBS), which proves useful for sample selection. • C2MT demonstrates robustness with respect to hyper-parameters and network architectures. • Comprehensive experimental results substantiate the advantages of C2MT. The collection of large-scale datasets inevitably introduces noisy labels, leading to a substantial degradation in the performance of deep neural networks (DNNs). Although sample selection is a mainstream method in the field of learning with noisy labels, which aims to mitigate the impact of noisy labels during model training, the testing performance of these methods exhibits significant fluctuations across different noise rates and types. In this paper, we propose Cross-to-Merge Training (C2MT), a novel framework that is insensitive to the prior information in sample selection progress, enhancing model robustness. In practical implementation, using cross-divided training data, two different networks are cross-trained with the co-teaching strategy for several local rounds, subsequently merged into a unified model by performing federated averages on the parameters of two models periodically. Additionally, we introduce a new class balance strategy, named Median Balance Strategy (MBS), during the cross-dividing process, which evenly divides the training data into a labeled subset and an unlabeled subset based on the estimated loss distribution characteristics. Extensive experimental results on both synthetic and real-world datasets demonstrate the effectiveness of C2MT. The Code will be available at: https://github.com/LanXiaoPang613/C2MT. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
09574174
Volume :
249
Database :
Academic Search Index
Journal :
Expert Systems with Applications
Publication Type :
Academic Journal
Accession number :
176785362
Full Text :
https://doi.org/10.1016/j.eswa.2024.123846