Deep neural networks to predict diabetic retinopathy

被引:173
作者
Gadekallu, Thippa Reddy [1 ]
Khare, Neelu [1 ]
Bhattacharya, Sweta [1 ]
Singh, Saurabh [2 ]
Maddikunta, Praveen Kumar Reddy [1 ]
Srivastava, Gautam [3 ,4 ,5 ]
机构
[1] Vellore Inst Technol, Vellore, Tamil Nadu, India
[2] Dongguk Univ, Coll Engn, Dept Ind & Syst Engn, Seoul 04620, South Korea
[3] Brandon Univ, Dept Math & Comp Sci, Brandon, MB R7A 6A9, Canada
[4] China Med Univ, Res Ctr Interneural Comp, Taichung 40402, Taiwan
[5] Asia Univ, Coll Informat & Elect Engn, Taichung, Taiwan
关键词
Diabetic retinopathy; Deep neural networks; Dimensionality reduction; Principal component analysis; Grey Wolf Optimizer (GWO); Deep learning; PREVALENCE; VALIDATION; ALGORITHM;
D O I
10.1007/s12652-020-01963-7
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Diabetic retinopathy is a prominent cause of blindness among elderly people and has become a global medical problem over the last few decades. There are several scientific and medical approaches to screen and detect this disease, but most of the detection is done using retinal fungal imaging. The present study uses principal component analysis based deep neural network model using Grey Wolf Optimization (GWO) algorithm to classify the extracted features of diabetic retinopathy dataset. The use of GWO enables to choose optimal parameters for training the DNN model. The steps involved in this paper include standardization of the diabetic retinopathy dataset using a standardscaler normalization method, followed by dimensionality reduction using PCA, then choosing of optimal hyper parameters by GWO and finally training of the dataset using a DNN model. The proposed model is evaluated based on the performance measures namely accuracy, recall, sensitivity and specificity. The model is further compared with the traditional machine learning algorithms-support vector machine (SVM), Naive Bayes Classifier, Decision Tree and XGBoost. The results show that the proposed model offers better performance compared to the aforementioned algorithms.
引用
收藏
页码:5407 / 5420
页数:14
相关论文
共 43 条
  • [1] Improved Automated Detection of Diabetic Retinopathy on a Publicly Available Dataset Through Integration of Deep Learning
    Abramoff, Michael David
    Lou, Yiyue
    Erginay, Ali
    Clarida, Warren
    Amelon, Ryan
    Folk, James C.
    Niemeijer, Meindert
    [J]. INVESTIGATIVE OPHTHALMOLOGY & VISUAL SCIENCE, 2016, 57 (13) : 5200 - 5206
  • [2] [Anonymous], 2015, P 3 INT C IND APPL E
  • [3] [Anonymous], 2019, World Bank Open Data
  • [4] [Anonymous], 2019, SCIENCE, DOI DOI 10.1186/S40494-019-0278-6
  • [5] An ensemble-based system for automatic screening of diabetic retinopathy
    Antal, Balint
    Hajdu, Andras
    [J]. KNOWLEDGE-BASED SYSTEMS, 2014, 60 : 20 - 27
  • [6] IntelliHealth: A medical decision support application using a novel weighted multi-layer classifier ensemble framework
    Bashir, Saba
    Qamar, Usman
    Khan, Farhan Hassan
    [J]. JOURNAL OF BIOMEDICAL INFORMATICS, 2016, 59 : 185 - 200
  • [7] Artificial intelligence using deep learning to screen for referable and vision-threatening diabetic retinopathy in Africa: a clinical validation study
    Bellemo, Valentina
    Lim, Zhan W.
    Lim, Gilbert
    Nguyen, Quang D.
    Xie, Yuchen
    Yip, Michelle Y. T.
    Hamzah, Haslina
    Ho, Jinyi
    Lee, Xin Q.
    Hsu, Wynne
    Lee, Mong L.
    Musonda, Lillian
    Chandran, Manju
    Chipalo-Mutati, Grace
    Muma, Mulenga
    Tan, Gavin S. W.
    Sivaprasad, Sobha
    Menon, Geeta
    Wong, Tien Y.
    Ting, Daniel S. W.
    [J]. LANCET DIGITAL HEALTH, 2019, 1 (01): : E35 - E44
  • [8] Chen T., 2016, KDD16 P 22 ACM, P785, DOI [DOI 10.1145/2939672.2939785, 10.1145/2939672.2939785]
  • [9] Cristianini N, 2008, ENCY ALGORITHMS, P928
  • [10] Cristianini N., 2000, Support Vector Machines