ACD-U: Asymmetric co-teaching with machine unlearning for robust learning with noisy labels
Organizations: Kansai University 3-3-35 Yamate-cho, Suita, Osaka 564-8680, Japan
Abstract
Deep neural networks are prone to memorizing incorrect labels during training, which degrades their generalizability. Although recent methods have combined sample selection with semi-supervised learning (SSL) to exploit the memorization effect -- where networks learn from clean data before noisy data -- they cannot correct selection errors once a sample is misclassified. To overcome this, we propose asymmetric co-teaching with different architectures (ACD)-U, an asymmetric co-teaching framework that uses different model architectures and incorporates machine unlearning. ACD-U addresses this limitation through two core mechanisms. First, its asymmetric co-teaching pairs a contrastive language-image pretraining (CLIP)-pretrained vision Transformer with a convolutional neural network (CNN), leveraging their complementary learning behaviors: the pretrained model provides stable predictions, whereas the CNN adapts throughout training. This asymmetry, where the vision Transformer is trained only on clean samples and the CNN is trained through SSL, effectively mitigates confirmation bias. Second, selective unlearning enables post-hoc error correction by identifying incorrectly memorized samples through loss trajectory analysis and CLIP consistency checks, and then removing their influence via Kullback--Leibler divergence-based forgetting. This approach shifts the learning paradigm from passive error avoidance to active error correction. Experiments on synthetic and real-world noisy datasets, including CIFAR-10/100, CIFAR-N, WebVision, Clothing1M, and Red Mini-ImageNet, demonstrate state-of-the-art performance, particularly in high-noise regimes and under instance-dependent noise. The code is publicly available at https://github.com/meruemon/ACD-U.
Figures & tables
| Technical Gap | Problem in Existing Methods | Representative Methods | ACD-U Solution |
|---|---|---|---|
| Irreversible Error Embedding | When both networks misclassify a noisy sample as clean, error becomes permanent with no correction mechanism | Co-teaching [ 14 ] , DivideMix [ 16 ] , ProMix [ 20 ] , RankMatch [ 19 ] | Selective unlearning enables post-hoc removal of memorized errors through loss trajectory analysis |
| Uniform Training of Heterogeneous Models | Different architectures trained identically, ignoring their distinct memorization patterns | CLIPCleaner [ 23 ] (static validator), TURN [ 27 ] (frozen ViT extractor) | Asymmetric training: ViT uses clean samples only, CNN uses SSL on all data |
| Unknown Forgetting Targets | Standard unlearning assumes prior knowledge of problematic samples, incompatible with noisy-label discovery during training | SCRUB [ 29 ] , SAP [ 46 ] (pretrained only) | Dynamic identification via loss changes + CLIP consistency checks (Eqs. 2 - 5 ) |
| Independent Component Operation | Techniques applied sequentially without feedback between components | NoiseBox [ 22 ] (sequential), CLIPCleaner [ 23 ] (fixed CLIP) | Interdependent system: ViT guides CNN, CNN informs ViT, unlearning corrects both |
| Method | CIFAR-10/-100 | CIFAR-N | Red Mini-ImageNet | Clothing1M | WebVision |
|---|---|---|---|---|---|
| DivideMix [ 16 ] | ✓ | ✓ | ✓ | ✓ | ✓ |
| LongReMix [ 17 ] | ✓ | ✓ | ✓ | ✓ | |
| ProMix [ 20 ] | ✓ | ✓ | ✓ | ||
| RankMatch [ 19 ] | ✓ | ✓ | ✓ | ||
| SV-Learner [ 21 ] | ✓ | ✓ | ✓ | ||
| NoiseBox+SS-KNN [ 22 ] | ✓ | ✓ | ✓ | ✓ |
| Parameter | Dataset | ||||||
|---|---|---|---|---|---|---|---|
| CIFAR-10 | CIFAR-100 | CIFAR-10N | CIFAR-100N | Red Mini-ImageNet | WebVision | Clothing1M | |
| Architecture (net ) | PreAct ResNet18 | PreAct ResNet18 | PreAct ResNet18 | PreAct ResNet18 | PreAct ResNet18 | ResNet-50 | ResNet-50 |
| Architecture (net ) | ViT | ViT | ViT | ViT | ViT | ViT | ViT |
| Optimizer | SGD | SGD | SGD | SGD | SGD | SGD | SGD |
| Momentum | 0.9 | 0.9 | 0.9 | 0.9 | 0.9 | 0.9 | 0.9 |
| Weight Decay | 0.0005 | 0.0005 | 0.0005 | 0.0005 | 0.0005 | 0.001 | 0.001 |
| Dataset | CIFAR-10 | CIFAR-100 | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Noise type | Sym. | Asym. | Sym. | |||||||
| Method/Noise ratio | 20% | 50% | 80% | 90% | 40% | 20% | 50% | 80% | 90% | |
| DivideMix [ 16 ] | Best | 96.1 | 94.6 | 93.2 | 76.0 | 93.4 | 77.3 | 74.6 | 60.2 | 31.5 |
| Last | 95.7 | 94.4 | 92.9 | 75.4 | 92.1 | 76.9 | 74.2 | 59.6 | 31.0 | |
| LongReMix [ 17 ] | Best | 96.3 | 95.1 | 93.8 | 79.9 | 94.7 | 77.9 | 75.5 | 62.3 | 34.7 |
| Last | 96.0 | 94.8 | 93.3 | 79.1 | 94.3 | 77.5 | 74.9 | 61.7 | 30.7 | |
| Dataset | CIFAR-10N | CIFAR-100N | ||
|---|---|---|---|---|
| Method | aggre | random | worst | noisy |
| DivideMix [ 16 ] | 95.01 | 95.16 | 92.56 | 71.13 |
| ProMix [ 20 ] | 97.65 | 97.39 | 96.34 | 73.79 |
| LSL [ 24 ] | - | - | 94.57 | 74.46 |
| Semi-RML++ [ 25 ] | 96.84 | 96.67 | 94.87 | 73.68 |
| ACD-U | 96.46 | 96.28 | 94.64 | 75.98 |
| Method | WebVision | ImageNet | ||
|---|---|---|---|---|
| Top1 | Top5 | Top1 | Top5 | |
| DivideMix [ 16 ] | 77.3 | 91.6 | 75.2 | 90.8 |
| LongReMix [ 17 ] | 78.9 | 92.3 | _ | _ |
| RankMatch [ 19 ] | 79.9 | 93.6 | 77.4 | 94.3 |
| SV-Learner [ 21 ] | 78.9 | 93.6 | 76.9 | 93.9 |
| NoiseBox+SS-KNN [ 22 ] | 81.4 | 93.0 | 77.0 | 91.0 |
| Method | Top1 (%) |
|---|---|
| DivideMix [ 16 ] | 74.8 |
| LongReMix [ 17 ] | 74.4 |
| ProMix [ 20 ] | 74.9 |
| RankMatch [ 19 ] | 75.2 |
| SV-Learner [ 21 ] | 75.2 |
| NoiseBox+SS-KNN [ 22 ] | 74.7 |
| Method/ noise ratio | 20% | 40% | 60% | 80% |
|---|---|---|---|---|
| DivideMix [ 16 ] | 50.96 | 46.72 | 43.14 | 34.50 |
| LongReMix [ 17 ] | 56.03 | 50.69 | 46.81 | 38.24 |
| NoiseBox+SS-KNN [ 22 ] | 60.94 | 59.08 | 50.30 | 44.44 |
| LSL [ 24 ] | 54.68 | 49.80 | 45.46 | 36.78 |
| CLIPCleaner [ 23 ] | 61.44 | 58.42 | 53.18 | 43.82 |
| ACD-U | 61.52 | 60.66 | 57.00 | 48.94 |
| Method | CIFAR-100 (Sym.) | ||
|---|---|---|---|
| 50% | 80% | ||
| ACD-U | Best | 81.6 | 74.4 |
| Last | 81.2 | 73.3 | |
| ACD-U w/o unlearning | Best | 81.6 ( ) | 72.7 ( ) |
| Last | 81.0 ( ) | 72.3 ( ) | |
| ACD-U w/o ACD | Best | 79.7 ( ) | 74.4 ( ) |
| Method | HN | LN | CS |
|---|---|---|---|
| DivideMix | 6444 | 33167 | 10389 |
| ACD-U | 1127 | 38484 | 10389 |
| Method | CIFAR-100 (Sym.) | |
| 80% | ||
| ACD-U | Best | 74.4 |
| Last | 73.3 | |
| ACD-U w/o Low-loss | Best | 73.9 (-0.5) |
| Last | 73.5 (+0.2) | |
| ACD-U w/o Loss-drop | Best | 72.8 (-1.6) |
| Dataset | CIFAR-10 | CIFAR-100 | ||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Noise type | Sym. | Asym. | Sym. | |||||||
| Method/ Noise rate | 20% | 50% | 80% | 90% | 40% | 20% | 50% | 80% | 90% | |
| ACD-U | best | 97.2 | 96.8 | 95.6 | 93.9 | 95.7 | 83.3 | 81.6 | 74.4 | 66.5 |
| last | 97.0 | 96.6 | 95.4 | 93.3 | 94.4 | 83.0 | 81.2 | 73.3 | 63.4 | |
| ACD-U w/ ULdW | best | 97.3 | 96.7 | 95.6 | 94.0 | 95.9 | 83.3 | 81.5 | 74.0 | 61.6 |
| last | 97.0 | 96.6 | 95.3 | 93.5 | 94.9 | 82.9 | 81.0 | 73.7 | 60.9 | |
| Dataset | CIFAR-100 | ||||
|---|---|---|---|---|---|
| Noise type | Sym. | ||||
| Method/ Noise rate | 20% | 50% | 80% | 90% | |
| ACD-U | best | 83.3 | 81.6 | 74.4 | 66.5 |
| last | 83.0 | 81.2 | 73.3 | 63.4 | |
| ACD-U w/ stop-grad | best | 81.0 | 80.5 | 75.1 | 70.9 |
| last | 80.5 | 79.7 | 74.6 | 70.5 | |
Noisy-Label Learning
Noisy-label learning develops ways to train and evaluate models when some supervision is incorrect, ambiguous, or unreliable.
See all 93 papers