HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

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

BenchmarkMethodologyMetrics
image-classification-on-clothing1mC2MT
Accuracy: 74.61%
learning-with-noisy-labels-on-animalC2MT
Accuracy: 85.9
ImageNet Pretrained: NO
Network: Vgg-19-BN
learning-with-noisy-labels-on-clothing1mC2MT
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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Cross-to-merge training with class balance strategy for learning with noisy labels | Papers | HyperAI