Improving Medical Image Classification with Label Noise Using Dual-uncertainty Estimation
Not on the map. Why?
Abstract (arXiv)
Deep neural networks are known to be data-driven and label noise can have a marked impact on model performance. Recent studies have shown great robustness to classic image recognition even under a high noisy rate. In medical applications, learning from datasets with label noise is more challenging since medical imaging datasets tend to have asymmetric (class-dependent) noise and suffer from high observer variability. In this paper, we systematically discuss and define the two common types of label noise in medical images - disagreement label noise from inconsistency expert opinions and single-target label noise from wrong diagnosis record. We then propose an uncertainty estimation-based framework to handle these two label noise amid the medical image classification task. We design a dual-uncertainty estimation approach to measure the disagreement label noise and single-target label noise via Direct Uncertainty Prediction and Monte-Carlo-Dropout. A boosting-based curriculum training procedure is later introduced for robust learning. We demonstrate the effectiveness of our method by conducting extensive experiments on three different diseases: skin lesions, prostate cancer, and retinal diseases. We also release a large re-engineered database that consists of annotations from more than ten ophthalmologists with an unbiased golden standard dataset for evaluation and benchmarking.
Key facts. Cited by 13 papers in the map (data through 7 Oct 2026). Builds on 30 of its 56 references that are in the map. First posted on arXiv 28 Feb 2021.
Builds on
30 of 56 references.
References · 30
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,335
- 2019CheXpert: A Large Chest Radiograph Dataset with Uncertainty Labels and Expert Comparisoncited by 1,410
- 2019MixMatch: A Holistic Approach to Semi-Supervised Learningcited by 1,390
- 2018Co-teaching: Robust Training of Deep Neural Networks with Extremely Noisy Labelscited by 1,053
- 2018Generalized Cross Entropy Loss for Training Deep Neural Networks with Noisy Labelscited by 994
- 2017ChestX-ray8: Hospital-scale Chest X-ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseasescited by 969
- 2016Universal adversarial perturbationscited by 774
- 2020DivideMix: Learning with Noisy Labels as Semi-supervised Learningcited by 691
- 2018Learning to Reweight Examples for Robust Deep Learningcited by 670
18 more
- 2017MentorNet: Learning Data-Driven Curriculum for Very Deep Neural Networks on Corrupted Labelscited by 652
- 2020Learning from Noisy Labels with Deep Neural Networks: A Surveycited by 580
- 2019Symmetric Cross Entropy for Robust Learning with Noisy Labelscited by 499
- 2017Regularizing Neural Networks by Penalizing Confident Output Distributionscited by 461
- 2017Robust Loss Functions under Label Noise for Deep Neural Networkscited by 441
- 2019Deep learning with noisy labels: exploring techniques and remedies in medical image analysiscited by 173
- 2020Dual T: Reducing Estimation Error for Transition Matrix in Label-noise Learningcited by 142
- 2018CurriculumNet: Weakly Supervised Learning from Large-Scale Web Imagescited by 126
- 2019Learning From Noisy Labels By Regularized Estimation Of Annotator Confusioncited by 126
- 2019Curriculum Loss: Robust Learning and Generalization against Label Corruptioncited by 87
- 2018Direct Uncertainty Prediction for Medical Second Opinionscited by 61
- 2019Asymmetric Co-Teaching for Unsupervised Cross Domain Person Re-Identificationcited by 36
- 2019Robust Learning at Noisy Labeled Medical Images: Applied to Skin Lesion Classificationcited by 33
- 2019IMAE for Noise-Robust Learning: Mean Absolute Error Does Not Treat Examples Equally and Gradient Magnitude's Variance Matterscited by 33
- 2019Quantifying and Leveraging Classification Uncertainty for Chest Radiograph Assessmentcited by 23
- 2019Interpreting chest X-rays via CNNs that exploit hierarchical disease dependencies and uncertainty labelscited by 8
- 2019A Survey on Deep Learning of Small Sample in Biomedical Image Analysiscited by 7
- 2017Transfer Learning with Label Noisecited by 4
Built on by
Citing papers per year: 2021: 5; 2022: 7; 2023: 0; 2024: 0; 2025: 1.
13 papers.
Sort:
- 2021Relational Subsets Knowledge Distillation for Long-tailed Retinal Diseases Recognitioncited by 27
- 2021Interpretability-Driven Sample Selection Using Self Supervised Learning For Disease Classification And Segmentationcited by 22
- 2021CT Image Synthesis Using Weakly Supervised Segmentation and Geometric Inter-Label Relations For COVID Image Analysiscited by 8
- 2022Unsupervised Domain Adaptive Fundus Image Segmentation with Category-level Regularizationcited by 6
- 2021Learning of Inter-Label Geometric Relationships Using Self-Supervised Learning: Application To Gleason Grade Segmentationcited by 5
- 2022Unsupervised Domain Adaptation Using Feature Disentanglement And GCNs For Medical Image Classificationcited by 5
- 2021Multimodal Generalized Zero Shot Learning for Gleason Grading using Self-Supervised Learningcited by 4
- 2022On the Ramifications of Human Label Uncertaintycited by 3
- 2022Interpretable Saliency Maps And Self-Supervised Learning For Generalized Zero Shot Medical Image Classificationcited by 2
- 2022Improved Super Resolution of MR Images Using CNNs and Vision Transformerscited by 1
- 2025Synthetic Generation and Latent Projection Denoising of Rim Lesions in Multiple Sclerosiscited by 1
- 2022Cyclic Generative Adversarial Networks With Congruent Image-Report Generation For Explainable Medical Image Analysiscited by 0
1 more
- 2022Establishment of Neural Networks Robust to Label Noisecited by 0
- 2025Synthetic Generation and Latent Projection Denoising of Rim Lesions in Multiple Sclerosis29 May 2025
- 2022Establishment of Neural Networks Robust to Label Noise28 Nov 2022
- 2022Cyclic Generative Adversarial Networks With Congruent Image-Report Generation For Explainable Medical Image Analysis16 Nov 2022
- 2022On the Ramifications of Human Label Uncertainty10 Nov 2022
- 2022Improved Super Resolution of MR Images Using CNNs and Vision Transformers24 Jul 2022
- 2022Unsupervised Domain Adaptive Fundus Image Segmentation with Category-level Regularization8 Jul 2022
- 2022Unsupervised Domain Adaptation Using Feature Disentanglement And GCNs For Medical Image Classification27 Jun 2022
- 2022Interpretable Saliency Maps And Self-Supervised Learning For Generalized Zero Shot Medical Image Classification4 Apr 2022
- 2021Multimodal Generalized Zero Shot Learning for Gleason Grading using Self-Supervised Learning15 Nov 2021
- 2021Learning of Inter-Label Geometric Relationships Using Self-Supervised Learning: Application To Gleason Grade Segmentation1 Oct 2021
- 2021CT Image Synthesis Using Weakly Supervised Segmentation and Geometric Inter-Label Relations For COVID Image Analysis15 Jun 2021
- 2021Relational Subsets Knowledge Distillation for Long-tailed Retinal Diseases Recognition22 Apr 2021
- 2021Relational Subsets Knowledge Distillation for Long-tailed Retinal Diseases Recognitioncited by 27
- 2021Interpretability-Driven Sample Selection Using Self Supervised Learning For Disease Classification And Segmentationcited by 22
- 2021CT Image Synthesis Using Weakly Supervised Segmentation and Geometric Inter-Label Relations For COVID Image Analysiscited by 8
- 2022Unsupervised Domain Adaptive Fundus Image Segmentation with Category-level Regularizationcited by 6
- 2021Learning of Inter-Label Geometric Relationships Using Self-Supervised Learning: Application To Gleason Grade Segmentationcited by 5
- 2022Unsupervised Domain Adaptation Using Feature Disentanglement And GCNs For Medical Image Classificationcited by 5
- 2021Multimodal Generalized Zero Shot Learning for Gleason Grading using Self-Supervised Learningcited by 4
- 2022On the Ramifications of Human Label Uncertaintycited by 3
- 2022Interpretable Saliency Maps And Self-Supervised Learning For Generalized Zero Shot Medical Image Classificationcited by 2
- 2022Improved Super Resolution of MR Images Using CNNs and Vision Transformerscited by 1
- 2025Synthetic Generation and Latent Projection Denoising of Rim Lesions in Multiple Sclerosiscited by 1
- 2022Cyclic Generative Adversarial Networks With Congruent Image-Report Generation For Explainable Medical Image Analysiscited by 0
12 most cited of 13 All 13 (JSON)
- 2025Synthetic Generation and Latent Projection Denoising of Rim Lesions in Multiple Sclerosis29 May 2025
- 2022Establishment of Neural Networks Robust to Label Noise28 Nov 2022
- 2022Cyclic Generative Adversarial Networks With Congruent Image-Report Generation For Explainable Medical Image Analysis16 Nov 2022
- 2022On the Ramifications of Human Label Uncertainty10 Nov 2022
- 2022Improved Super Resolution of MR Images Using CNNs and Vision Transformers24 Jul 2022
- 2022Unsupervised Domain Adaptive Fundus Image Segmentation with Category-level Regularization8 Jul 2022
- 2022Unsupervised Domain Adaptation Using Feature Disentanglement And GCNs For Medical Image Classification27 Jun 2022
- 2022Interpretable Saliency Maps And Self-Supervised Learning For Generalized Zero Shot Medical Image Classification4 Apr 2022
- 2021Multimodal Generalized Zero Shot Learning for Gleason Grading using Self-Supervised Learning15 Nov 2021
- 2021Learning of Inter-Label Geometric Relationships Using Self-Supervised Learning: Application To Gleason Grade Segmentation1 Oct 2021
- 2021CT Image Synthesis Using Weakly Supervised Segmentation and Geometric Inter-Label Relations For COVID Image Analysis15 Jun 2021
- 2021Relational Subsets Knowledge Distillation for Long-tailed Retinal Diseases Recognition22 Apr 2021
12 newest of 13 All 13 (JSON)