WENN-4: Weighted Ensemble for Enhanced Diabetic Retinopathy Detection
Diabetic Retinopathy (DR) is a serious eye condition that impacts individuals with diabetes, leading to retinal damage and potentially resulting in gradual vision loss. In this paper, we proposed a novel framework for diabetic retinopathy detection using deep learning techniques. The proposed framework employs an ensemble learning model named WENN-4 to classify fundus images into five classes and measure the severity of blindness of individuals. The proposed framework consists of following steps: customized preprocessing method that includes Gaussian Blurring and augmentation techniques, subsequently followed by inputting the processed images into diverse CNN architectures to extract discriminatory feature vectors for classification and early diagnoses and categorizes blindness severity levels, including no-DR, mild DR, severe, moderate, or Proliferative Diabetic Retinopathy (PDR). To enhance the overall performance of our proposed framework, we utilized an ensemble of models, WENN-4. This ensemble comprises ResNet-50, DenseNet-121, InceptionV3, and EfficientNetV2. This model achieves a remarkable accuracy of 93.7% accuracy on test data of APTOS 2019, surpassing existing methods. This ensemble of models lead to increased accuracy by leveraging complementary strengths, correcting individual misclassifications, and ultimately producing superior outcomes.
Authors
- Abhishek Vikram (ORCID: https://orcid.org/0009-0002-6075-2549)
- Amit M. Oza (ORCID: https://orcid.org/0000-0002-9510-8641)
- Santosh Kumar (ORCID: https://orcid.org/0000-0002-4130-6992)
- Abhinav Roy (ORCID: https://orcid.org/0009-0000-6854-1274)
- Bhavesh Gyanchandani
Institutions
- International Institute of Information Technology (IN)
- Indian Institute of Information Technology, Nagpur (IN)
Publication Details
- Published
- 2024-12-01
- DOI
- https://doi.org/10.1109/tencon61640.2024.10902990
- Primary Topic
- Retinal Imaging and Analysis
- Type
- article
- Field-Weighted Citation Impact
- 0.00