Improving Medical Image Classification with Label Noise Using Dual-uncertainty Estimation
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.
Code (0)
등록된 구현이 없습니다.
Tasks
BenchmarkingGeneral Classificationimage-classificationImage ClassificationImage Classification with Label NoiseMedical Image ClassificationSimilar Papers 제목 키워드 기반
Co-Correcting: Noise-tolerant Medical Image Classification via mutual Label Correction
With the development of deep learning, medical image classification has been significantly improved. However, deep learning requires massive data with labels. While labeling the samples by human experts is expensive and …
ClassificationDeep Learningimage-classificationImage Classification+2Bayesian Statistics Guided Label Refurbishment Mechanism: Mitigating Label Noise in Medical Image Classification
Purpose: Deep neural networks (DNNs) have been widely applied in medical image classification, benefiting from its powerful mapping capability among medical images. However, these existing deep learning-based methods dep…
Classificationimage-classificationImage ClassificationMedical Image ClassificationInvestigating the Robustness of Vision Transformers against Label Noise in Medical Image Classification
Label noise in medical image classification datasets significantly hampers the training of supervised deep learning methods, undermining their generalizability. The test performance of a model tends to decrease as the la…
Classificationimage-classificationImage ClassificationMedical Image ClassificationUnleashing the Potential of Open-set Noisy Samples Against Label Noise for Medical Image Classification
Addressing mixed closed-set and open-set label noise in medical image classification remains a largely unexplored challenge. Unlike natural image classification, which often separates and processes closed-set and open-se…
Classificationimage-classificationImage ClassificationMedical Image ClassificationLabel-noise-tolerant medical image classification via self-attention and self-supervised learning
Deep neural networks (DNNs) have been widely applied in medical image classification and achieve remarkable classification performance. These achievements heavily depend on large-scale accurately annotated training data.…
Contrastive Learningimage-classificationImage ClassificationMedical Image Classification+1