Enhancing semi‐supervised contrastive learning through saliency map for diabetic retinopathy grading

Jiacheng Zhang,Rong Jin,Wenqiang Liu
DOI: https://doi.org/10.1049/cvi2.12308
IF: 1.484
2024-08-29
IET Computer Vision
Abstract:The challenges of the complexity of diabetic retinopathy (DR) lesion characteristics are addressed by proposing the enhancing semi‐supervised contrastive learning through saliency estimation map (SCL‐SEM) framework for DR grading. By focusing on inter‐class and intra‐class variations in DR grading, the proposed method significantly improves the accuracy of automated DR grading systems. Experiments on EyePACS and Messidor datasets confirm our approach's superiority, outperforming state‐of‐the‐art methods. Diabetic retinopathy (DR) is a severe ophthalmic condition that can lead to blindness if not diagnosed and provided timely treatment. Hence, the development of efficient automated DR grading systems is crucial for early screening and treatment. Although progress has been made in DR detection using deep learning techniques, these methods still face challenges in handling the complexity of DR lesion characteristics and the nuances in grading criteria. Moreover, the performance of these algorithms is hampered by the scarcity of large‐scale, high‐quality annotated data. An innovative semi‐supervised fundus image DR grading framework is proposed, employing a saliency estimation map to bolster the model's perception of fundus structures, thereby improving the differentiation between lesions and healthy regions. By integrating semi‐supervised and contrastive learning, the model's ability to recognise inter‐class and intra‐class variations in DR grading is enhanced, allowing for precise discrimination of various lesion features. Experiments conducted on publicly available DR grading datasets, such as EyePACS and Messidor, have validated the effectiveness of our proposed method. Specifically, our approach outperforms the state of the art on the kappa metric by 0.8% on the full EyePACS dataset and by 3.2% on a 10% subset of EyePACS, demonstrating its superiority over previous methodologies. The authors' code is publicly available at https://github.com/500ZhangJC/SCL‐SEM‐framework‐for‐DR‐Grading.
computer science, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?