SSP-RACL: Classification of Noisy Fundus Images with Self-Supervised Pretraining and Robust Adaptive Credal Loss
Fundus image classification is crucial in the computer aided diagnosis tasks, but label noise significantly impairs the performance of deep neural networks. To address this challenge, we propose a robust framework, Self-Supervised Pretraining with Robust Adaptive Credal Loss (SSP-RACL), for handling label noise in fundus image datasets. First, we use Masked Autoencoders (MAE) for pre-training to extract features, unaffected by label noise. Subsequently, RACL employ a superset learning framework, setting confidence thresholds and adaptive label relaxation parameter to construct possibility distributions and provide more reliable ground-truth estimates, thus effectively suppressing the memorization effect. Additionally, we introduce clinical knowledge-based asymmetric noise generation to simulate real-world noisy fundus image datasets. Experimental results demonstrate that our proposed method outperforms existing approaches in handling label noise, showing superior performance.
Authors
- Zekuan Yu (ORCID: https://orcid.org/0000-0003-3655-872X)
- Mengwen Ye
- Li You (ORCID: https://orcid.org/0000-0002-3919-4768)
- Yingzi Huangfu
Institutions
- Fudan University (CN)
Publication Details
- Published
- 2024-10-24
- DOI
- https://doi.org/10.1109/biocas61083.2024.10798243
- Primary Topic
- Retinal Imaging and Analysis
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Fudan University
- HORIZON EUROPE Health