sis 22(4): 5

Research Article

Convolutional Neural Network for Multi-class Classification of Diabetic Eye Disease

Download769 downloads
  • @ARTICLE{10.4108/eai.16-12-2021.172436,
        author={Rubina Sarki and Khandakar Ahmed and Hua Wang and Yanchun Zhang and Kate Wang},
        title={Convolutional Neural Network for Multi-class Classification of Diabetic Eye Disease},
        journal={EAI Endorsed Transactions on Scalable Information Systems},
        volume={9},
        number={4},
        publisher={EAI},
        journal_a={SIS},
        year={2021},
        month={12},
        keywords={Diabetic Eye Disease, Deep Learning, Multi-class Classification, Image Processing},
        doi={10.4108/eai.16-12-2021.172436}
    }
    
  • Rubina Sarki
    Khandakar Ahmed
    Hua Wang
    Yanchun Zhang
    Kate Wang
    Year: 2021
    Convolutional Neural Network for Multi-class Classification of Diabetic Eye Disease
    SIS
    EAI
    DOI: 10.4108/eai.16-12-2021.172436
Rubina Sarki1,*, Khandakar Ahmed1, Hua Wang1, Yanchun Zhang1, Kate Wang2
  • 1: Victoria University
  • 2: RMIT University
*Contact email: rubina.sarki@live.vu.edu.au

Abstract

Prompt examination increases the chances of effective treatment of Diabetic Eye Disease (DED) and reduces the likelihood of permanent deterioration of vision. A key tool commonly used for the initial diagnosis of patients with DED or other eye disorders is the screening of retinal fundus images. Manual detection with these images is, however, labour intensive and time consuming. As deep learning (DL) has recently been demonstrated to provide impressive benefits to clinical practice, researchers have attempted to use DL method to detect retinal eye diseases from retinal fundus photographs. DL techniques in machine learning (ML) have achieved state-of-the-art performance in the binary classification of healthy and diseased retinal fundus images while the classification of multi-class retinal eye diseases remains an open challenge. Multi- class DED is therefore considered in this study seeking to develop an automated classification framework for DED. Detecting multiple DEDs from retinal fundus images is an important research topic with practical consequences. Our proposed model was tested on various retinal fundus images gathered from the publicly available dataset and annotated by an ophthalmologist. This experiment was conducted employing a new convolutional neural network (CNN) model. Our proposed model for multi-class classification achieved a maximum accuracy of 81.33%, sensitivity of 100%, and specificity of 100%.