NetraNet: A similarity learning method for diabetic retinopathy classification and clinical diagnosis
摘要
Diabetic retinopathy (DR) is the primary cause of visual impairments impacting the diabetic population worldwide. Early detection and classification of diabetic retinopathy (DR) are crucial for preventing vision impairment and blindness in diabetic patients.
ObjectiveWith the aim of improving the fine-grained DR classification, a similarity learning model, based on robust lesion-aware features and few-shot learning, is proposed in this work.
MethodA lightweight autoencoder is designed to enhance the detection of small lesions. It integrates channel and spatial attention layers, enabling the network to efficiently focus on lesion-relevant regions. N-shot meta-learning is applied for severity classification under limited data conditions. The proposed model is trained and assessed on six publicly available fundus image datasets. To evaluate adaptability to regional variations, the model is validated on a curated Indian fundus dataset comprising 100 well-labeled fundus images.
ResultsNetraNet is evaluated on all the key metrics that are necessary for medical diagnosis, obtaining the detection accuracy of 99.5%, 0.993 precision, and 0.994 recall. This high level of performance demonstrates the potential of NetraNet in DR diagnosis. For multi-class classification, the model achieves an accuracy of 95.9% with a Cohen’s kappa score of 0.95, indicating high performance and robustness. Our model attains a significant Kappa score of 0.94 on collected data, serving as a reliable tool for clinical DR screening.
ConclusionNetraNet exhibits strong performance across diverse datasets, even in limited data conditions, making it a favorable method for real-world clinical screening scenarios where well-labeled data is often limited. This work demonstrates the impact of strong feature extraction along with similarity learning models in medical image classification and offers a scalable solution for other fundus disease detection.