Diabetic Retinopathy Classification from Retinal Images using Machine Learning Approaches

Indronil Bhattacharjee,Al-Mahmud,Tareq Mahmud
2024-12-03
Abstract:Diabetic Retinopathy is one of the most familiar diseases and is a diabetes complication that affects eyes. Initially, diabetic retinopathy may cause no symptoms or only mild vision problems. Eventually, it can cause blindness. So early detection of symptoms could help to avoid blindness. In this paper, we present some experiments on some features of diabetic retinopathy, like properties of exudates, properties of blood vessels and properties of microaneurysm. Using the features, we can classify healthy, mild non-proliferative, moderate non-proliferative, severe non-proliferative and proliferative stages of DR. Support Vector Machine, Random Forest and Naive Bayes classifiers are used to classify the stages. Finally, Random Forest is found to be the best for higher accuracy, sensitivity and specificity of 76.5%, 77.2% and 93.3% respectively.
Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the early detection and classification of diabetic retinopathy (DR). Specifically, the author aims to automatically detect and classify different stages of diabetic retinopathy from fundus images through machine - learning methods, in order to help avoid blindness caused by the deterioration of the disease. ### Specific description of the problem 1. **Early detection**: Diabetic retinopathy may have no obvious symptoms or only show mild vision problems in the early stage, but if not treated in time, it may eventually lead to blindness. Therefore, early detection is crucial for preventing vision loss. 2. **Classification of different stages**: Diabetic retinopathy can be divided into non - proliferative diabetic retinopathy (NPDR) and proliferative diabetic retinopathy (PDR) according to its severity. NPDR is further divided into three stages: mild, moderate and severe. Accurately identifying these stages helps doctors develop more effective treatment plans. ### Solution To achieve the above goals, the author proposes a machine - learning - based system, which includes the following steps: 1. **Image pre - processing**: Pre - process the input color fundus images, including operations such as resizing, green - channel extraction, and contrast enhancement, to improve the image quality. 2. **Feature extraction**: Extract key features from the pre - processed images, mainly including the properties of exudates, blood vessels and microaneurysms. The specifically extracted features include: - Histogram and zero - order Hu moment of exudates - Histogram and zero - order Hu moment of blood vessels - Histogram and zero - order Hu moment of microaneurysms 3. **Classification model training and testing**: Use three classifiers (support vector machine SVM, random forest and naive Bayes) to classify the extracted features. The experimental results show that the random forest classifier performs best in this task, with an accuracy, sensitivity and specificity of 76.5%, 77.2% and 93.3% respectively. ### Conclusion Through the above methods, this study has successfully achieved the early detection and classification of diabetic retinopathy. Compared with existing methods, this system performs well in multi - class classification tasks and has high accuracy and specificity. Future work can consider introducing more classifiers (such as K - nearest neighbor classifier) and more features (such as hemorrhage), and using larger data sets to further improve the performance of the system. ### Formula representation When evaluating the performance of classifiers, the following formulas are used to calculate accuracy, sensitivity and specificity: \[ \text{Accuracy}=\frac{\text{True Positive}+\text{True Negative}}{\text{Total Number of Data}} \] \[ \text{Sensitivity}=\frac{\text{True Positive}}{\text{True Positive}+\text{False Negative}} \] \[ \text{Specificity}=\frac{\text{True Negative}}{\text{True Negative}+\text{False Positive}} \]