An ensemble learning approach for diabetes prediction using boosting techniques

被引:18
作者
Ganie, Shahid Mohammad [1 ]
Pramanik, Pijush Kanti Dutta [2 ]
Malik, Majid Bashir [3 ]
Mallik, Saurav [4 ]
Qin, Hong [5 ]
机构
[1] Woxsen Univ, AI Res Ctr, Sch Business, Hyderabad, India
[2] Galgotias Univ, Sch Comp Applicat & Technol, Greater Noida, India
[3] Baba Ghulam Shah Badshah Univ, Dept Comp Sci, Rajauri, India
[4] Harvard Univ, Sch Publ Hlth, Dept Environm Hlth, Boston, MA 02138 USA
[5] Univ Tennessee Chattanooga, Coll Engn & Comp Sci, Chattanooga, TN 37403 USA
关键词
diabetes prediction; ensemble learning; XGBoost; CatBoost; LightGBM; AdaBoost; gradient boost;
D O I
10.3389/fgene.2023.1252159
中图分类号
Q3 [遗传学];
学科分类号
071007 ; 090102 ;
摘要
Introduction: Diabetes is considered one of the leading healthcare concerns affecting millions worldwide. Taking appropriate action at the earliest stages of the disease depends on early diabetes prediction and identification. To support healthcare providers for better diagnosis and prognosis of diseases, machine learning has been explored in the healthcare industry in recent years.Methods: To predict diabetes, this research has conducted experiments on five boosting algorithms on the Pima diabetes dataset. The dataset was obtained from the University of California, Irvine (UCI) machine learning repository, which contains several important clinical features. Exploratory data analysis was used to identify the characteristics of the dataset. Moreover, upsampling, normalisation, feature selection, and hyperparameter tuning were employed for predictive analytics.Results: The results were analysed using various statistical/machine learning metrics and k-fold cross-validation techniques. Gradient boosting achieved the greatest accuracy rate of 92.85% among all the classifiers. Precision, recall, f1-score, and receiver operating characteristic (ROC) curves were used to further validate the model.Discussion: The suggested model outperformed the current studies in terms of prediction accuracy, demonstrating its applicability to other diseases with similar predicate indications.
引用
收藏
页数:15
相关论文
共 26 条
[1]  
Aziz Norshakirah, 2020, 2020 International Conference on Computational Intelligence (ICCI), P11, DOI 10.1109/ICCI51257.2020.9247843
[2]  
Beschi Raja J., 2019, Int. J. Eng. Adv. Technol., V9, P3181, DOI [10.35940/ijeat.A9898.109119, DOI 10.35940/IJEAT.A9898.109119, 10.35940/ijeat.a9898.109119]
[3]  
Diabetes Federation International and IDF, 2019, IDF diabetes Atlas 2019
[4]  
Dutta D, 2018, 2018 IEEE 9TH ANNUAL INFORMATION TECHNOLOGY, ELECTRONICS AND MOBILE COMMUNICATION CONFERENCE (IEMCON), P924, DOI 10.1109/IEMCON.2018.8614871
[5]  
Ganie Shahid Mohammad, 2022, International Journal of Medical Engineering and Informatics, P473, DOI 10.1504/IJMEI.2022.126519
[6]  
Ganie S.M., 2021, Advances in Intelligent Systems andComputing, P487
[7]  
Ganie S.M., 2023, Comput. Syst. Sci. Eng., V46, P3993, DOI [10.32604/csse.2023.035244, DOI 10.32604/CSSE.2023.035244]
[8]   Performance analysis and prediction of type 2 diabetes mellitus based on lifestyle data using machine learning approaches [J].
Ganie, Shahid Mohammad ;
Malik, Majid Bashir ;
Arif, Tasleem .
JOURNAL OF DIABETES AND METABOLIC DISORDERS, 2022, 21 (01) :339-352
[9]  
Ganie SM., 2022, Healthcare Analytics, V2, DOI [10.1016/j.health.2022.100092, DOI 10.1016/J.HEALTH.2022.100092]
[10]   CatBoost for big data: an interdisciplinary review [J].
Hancock, John T. ;
Khoshgoftaar, Taghi M. .
JOURNAL OF BIG DATA, 2020, 7 (01)