Alert button

A Label Management Mechanism for Retinal Fundus Image Classification of Diabetic Retinopathy

Jun 23, 2021
Mengdi Gao, Ximeng Feng, Mufeng Geng, Zhe Jiang, Lei Zhu, Xiangxi Meng, Chuanqing Zhou, Qiushi Ren, Yanye Lu

Figure 1 for A Label Management Mechanism for Retinal Fundus Image Classification of Diabetic Retinopathy
Figure 2 for A Label Management Mechanism for Retinal Fundus Image Classification of Diabetic Retinopathy
Figure 3 for A Label Management Mechanism for Retinal Fundus Image Classification of Diabetic Retinopathy
Figure 4 for A Label Management Mechanism for Retinal Fundus Image Classification of Diabetic Retinopathy

Share this with someone who'll enjoy it:

Diabetic retinopathy (DR) remains the most prevalent cause of vision impairment and irreversible blindness in the working-age adults. Due to the renaissance of deep learning (DL), DL-based DR diagnosis has become a promising tool for the early screening and severity grading of DR. However, training deep neural networks (DNNs) requires an enormous amount of carefully labeled data. Noisy label data may be introduced when labeling plenty of data, degrading the performance of models. In this work, we propose a novel label management mechanism (LMM) for the DNN to overcome overfitting on the noisy data. LMM utilizes maximum posteriori probability (MAP) in the Bayesian statistic and time-weighted technique to selectively correct the labels of unclean data, which gradually purify the training data and improve classification performance. Comprehensive experiments on both synthetic noise data (Messidor \& our collected DR dataset) and real-world noise data (ANIMAL-10N) demonstrated that LMM could boost performance of models and is superior to three state-of-the-art methods.

* 10 pages, 9 figures  
View paper onarxiv icon

Share this with someone who'll enjoy it: