Efficient diabetic retinopathy diagnosis through U-Net–KNN integration in retinal fundus images

Abstract

Diabetic retinopathy (DR) is a retinal disorder that may lead to blindness in people all over the world. The major cause of DR is diabetes for a longer period and early detection is the only solution to prevent the vision. This paper focuses on the classes of Normal eye (No DR), Mild NPDR (Non-Proliferative Diabetic Retinopathy), Moderate NPDR, Severe NPDR, and PDR. On retinal fundus images, an effective method for identifying diabetic retinopathy (DR) is proposed by combining the U-Net architecture with the K-nearest neighbours (KNN) algorithm. The U-Net architecture is used for segmenting exudates in retinal pictures, and the KNN algorithm is used for final classification. The combination of U-Net and KNN enables accurate feature extraction and efficient classification, effectively overcoming the computational challenges common to deep learning models. The experiments are carried out utilizing a publicly available dataset of retinal fundus images from Kaggle to assess the effectiveness of our suggested strategy. The proposed architecture provides precise output when compared to other models GoogleNet, ResNet18, and VGG16. The proposed model provides a training accuracy of 82.96% and detection of PDR with high accuracy in the short period which prevents loss of vision in early stage.