Command Palette
Search for a command to run...
Cross-to-merge training with class balance strategy for learning with noisy labels
{Qiu Chen YingWen Zhu Ge Jin Ming Yang Yi Zhu Qian Zhang}
Abstract
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 crossdivided 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://g ithub.com/LanXiaoPang613/C2MT.
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| image-classification-on-clothing1m | C2MT | Accuracy: 74.61% |
| learning-with-noisy-labels-on-animal | C2MT | Accuracy: 85.9 ImageNet Pretrained: NO Network: Vgg-19-BN |
| learning-with-noisy-labels-on-clothing1m | C2MT | Test Accuracy: 74.61 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.