A Study on the Segmentation and Classification of Diabetic Retinopathy Images Using the K-Means Clustering Method

Abstract

Diabetic Retinopathy (DR) is a retinal disease caused by diabetes, representing one of the most prevalent causes of vision loss affecting millions of people worldwide. Swift detection and treatment of this condition are crucial for preventing the disease. Various deep learning and machine learning algorithms have been employed for disease detection and classification, often overlooking the data preprocessing stage. In the data preprocessing phase of this study, segmentation of important lesions, such as hard exudates, was conducted using the K-Means clustering method. The identified lesions were highlighted on the original images. The resulting dataset was then classified using pre-trained architectures, namely EfficientNetV2-M, ResNet50, MobileNet, and DenseNet121. After training on the APTOS dataset, the EfficientNetV2-M model achieved an accuracy of 95.16%. The classification results indicated the contribution of the lesion highlighting process during data preprocessing to the overall classification accuracy. © 2024 IEEE.