Classification of Diabetic Retinopathy through Deep Feature Extraction and Classic Machine Learning Approach

Radifa Hilya Paradisa, Devvi Sarwinda, Alhadi Bustamam, Terry Argyadiva

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

Diabetic Retinopathy (DR) is a complication of diabetes, the leading cause of vision loss in working-age adults. An ophthalmologist can carry out the diagnosis of DR by examining color fundus images. However, the fundus image analysis process takes a long time. Automatic detection of DR is achallenging task. One of the deep learning approaches, Convolutional Neural Networks (CNN), is efficient in image classification tasks. In this research, a CNN architecture is used, namely ResNet-50, as feature extraction and classification. The ResNet-50 feature output at the feature extraction stage is also used as input for machine learning classifiers such as Support Vector Machine (SVM), Random Forest (RF), k-Nearest Neighbor (k-NN), and Extreme Gradient Boosting (XGBoost). The model works by using fundus images from the DIARETDBI dataset. Data augmentation and preprocessing are proposed in this study to facilitate the model in recognizing images. The performance of each classifier is evaluated based on accuracy, sensitivity, and specificity. The SVM classifier achieved 99% for accuracy and sensitivity in the 80:20 dataset composition. The k-NN classifier obtains the highest specificity for the same dataset's design by 100%.

Original languageEnglish
Title of host publication2020 3rd International Conference on Information and Communications Technology, ICOIACT 2020
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages377-381
Number of pages5
ISBN (Electronic)9781728173566
DOIs
Publication statusPublished - 24 Nov 2020
Event3rd International Conference on Information and Communications Technology, ICOIACT 2020 - Yogyakarta, Indonesia
Duration: 24 Nov 202025 Nov 2020

Publication series

Name2020 3rd International Conference on Information and Communications Technology, ICOIACT 2020

Conference

Conference3rd International Conference on Information and Communications Technology, ICOIACT 2020
CountryIndonesia
CityYogyakarta
Period24/11/2025/11/20

Keywords

  • Convolutional Neural Network
  • Diabetic Retinopathy
  • Feature Extraction
  • Machine Learning Classifier
  • ResNet-50

Fingerprint Dive into the research topics of 'Classification of Diabetic Retinopathy through Deep Feature Extraction and Classic Machine Learning Approach'. Together they form a unique fingerprint.

Cite this