Deep neural networks to predict diabetic retinopathy

被引:182
作者
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 [J].
Abramoff, Michael David ;
Lou, Yiyue ;
Erginay, Ali ;
Clarida, Warren ;
Amelon, Ryan ;
Folk, James C. ;
Niemeijer, Meindert .
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 [J].
Antal, Balint ;
Hajdu, Andras .
KNOWLEDGE-BASED SYSTEMS, 2014, 60 :20-27
[6]   IntelliHealth: A medical decision support application using a novel weighted multi-layer classifier ensemble framework [J].
Bashir, Saba ;
Qamar, Usman ;
Khan, Farhan Hassan .
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 [J].
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. .
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