Diabetic Retinopathy Prediction by Ensemble Learning Based on Biochemical and Physical Data

被引:14
|
作者
Shen, Zun [1 ]
Wu, Qingfeng [1 ]
Wang, Zhi [2 ]
Chen, Guoyi [1 ]
Lin, Bin [1 ]
机构
[1] Xiamen Univ, Sch Informat, Xiamen 361005, Peoples R China
[2] Tsinghua Univ, Dept Microelect & Nanoelect, Beijing 100876, Peoples R China
关键词
XGBoost feature selection; stacking ensemble learning; model fusion; diabetic retinopathy prediction; SELECTION;
D O I
10.3390/s21113663
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
(1) Background: Diabetic retinopathy, one of the most serious complications of diabetes, is the primary cause of blindness in developed countries. Therefore, the prediction of diabetic retinopathy has a positive impact on its early detection and treatment. The prediction of diabetic retinopathy based on high-dimensional and small-sample-structured datasets (such as biochemical data and physical data) was the problem to be solved in this study. (2) Methods: This study proposed the XGB-Stacking model with the foundation of XGBoost and stacking. First, a wrapped feature selection algorithm, XGBIBS (Improved Backward Search Based on XGBoost), was used to reduce data feature redundancy and improve the effect of a single ensemble learning classifier. Second, in view of the slight limitation of a single classifier, a stacking model fusion method, Sel-Stacking (Select-Stacking), which keeps Label-Proba as the input matrix of meta-classifier and determines the optimal combination of learners by a global search, was used in the XGB-Stacking model. (3) Results: XGBIBS greatly improved the prediction accuracy and the feature reduction rate of a single classifier. Compared to a single classifier, the accuracy of the Sel-Stacking model was improved to varying degrees. Experiments proved that the prediction model of XGB-Stacking based on the XGBIBS algorithm and the Sel-Stacking method made effective predictions on diabetes retinopathy. (4) Conclusion: The XGB-Stacking prediction model of diabetic retinopathy based on biochemical and physical data had outstanding performance. This is highly significant to improve the screening efficiency of diabetes retinopathy and reduce the cost of diagnosis.
引用
收藏
页数:19
相关论文
共 50 条
  • [1] A Machine Learning Ensemble Classifier for Early Prediction of Diabetic Retinopathy
    Somasundaram S K
    Alli P
    [J]. Journal of Medical Systems, 2017, 41
  • [2] A Machine Learning Ensemble Classifier for Early Prediction of Diabetic Retinopathy
    Somasundaram, S. K.
    Alli, P.
    [J]. JOURNAL OF MEDICAL SYSTEMS, 2017, 41 (12)
  • [3] Detection and classification of diabetic retinopathy based on ensemble learning
    Ankur Biswas
    Rita Banik
    [J]. Advances in Computational Intelligence, 2024, 4 (3):
  • [4] Ensemble Architecture for Prediction of Grading of Diabetic Retinopathy
    Jain, Shruti
    Saxena, Sanket
    Sinha, Shivam
    [J]. CYBERNETICS AND SYSTEMS, 2022,
  • [5] Learning ensemble classifiers for diabetic retinopathy assessment
    Saleh, Emran
    Blaszczynski, Jerzy
    Moreno, Antonio
    Valls, Aida
    Romero-Aroca, Pedro
    de la Riya-Fernandez, Sofia
    Slowinsk, Roman
    [J]. ARTIFICIAL INTELLIGENCE IN MEDICINE, 2018, 85 : 50 - 63
  • [6] Diabetic retinopathy prediction based on deep learning and deformable registration
    Mohammed Oulhadj
    Jamal Riffi
    Khodriss Chaimae
    Adnane Mohamed Mahraz
    Bennis Ahmed
    Ali Yahyaouy
    Chraibi Fouad
    Abdellaoui Meriem
    Benatiya Andaloussi Idriss
    Hamid Tairi
    [J]. Multimedia Tools and Applications, 2022, 81 : 28709 - 28727
  • [7] Deep Learning for Diabetic Retinopathy Prediction
    Rodriguez-Leon, Ciro
    Arevalo, William
    Banos, Oresti
    Villalonga, Claudia
    [J]. ADVANCES IN COMPUTATIONAL INTELLIGENCE, IWANN 2021, PT I, 2021, 12861 : 537 - 546
  • [8] Diabetic retinopathy prediction based on deep learning and deformable registration
    Oulhadj, Mohammed
    Riffi, Jamal
    Chaimae, Khodriss
    Mahraz, Adnane Mohamed
    Ahmed, Bennis
    Yahyaouy, Ali
    Fouad, Chraibi
    Meriem, Abdellaoui
    Idriss, Benatiya Andaloussi
    Tairi, Hamid
    [J]. MULTIMEDIA TOOLS AND APPLICATIONS, 2022, 81 (20) : 28709 - 28727
  • [9] Data Homogeneity Effect in Deep Learning-Based Prediction of Type 1 Diabetic Retinopathy
    Lo, Jui-En
    Kang, Eugene Yu-Chuan
    Chen, Yun-Nung
    Hsieh, Yi-Ting
    Wang, Nan-Kai
    Chen, Ta-Ching
    Chen, Kuan-Jen
    Wu, Wei-Chi
    Hwang, Yih-Shiou
    Lo, Fu-Sung
    Lai, Chi-Chun
    [J]. JOURNAL OF DIABETES RESEARCH, 2021, 2021
  • [10] Diabetic Retinopathy Diagnosis with Ensemble Deep-Learning
    Kumar, Snehil
    [J]. ICVISP 2019: PROCEEDINGS OF THE 3RD INTERNATIONAL CONFERENCE ON VISION, IMAGE AND SIGNAL PROCESSING, 2019,