Stacked Ensemble-Based Type-2 Diabetes Prediction Using Machine Learning Techniques

Annals of Emerging Technologies in Computing

article

Rahim, Abdur¹  Hossain, Alfaz¹  Hossain, Najmul¹  Shin, Jungpil²  Yun, Keun Soo³ 

[1]Pabna University of Science and Technology

[2]The University of Aizu

[3]Ulsan College

关键词: Base and Meta Model;    Diabetes Type 2;    Machine Learning Techniques;    Stacked Ensemble;   

DOI  :  10.33166/AETiC.2023.01.003

学科分类：电子与电气工程

来源: International Association for Educators and Researchers (IAER)

【 摘 要 】

Diabetes is a long-term disease caused by the human body's inability to make enough insulin or to use it properly. This is one of the curses of the present world. Although it is not very severe in the initial stage, over time, it takes a deadly shape and gradually affects a variety of human organs, such as the heart, kidney, liver, eyes, and brain, leading to death. Many researchers focus on the machine and in-depth learning strategies to efficiently predict diabetes based on numerous risk variables such as insulin, BMI, and glucose in this healthcare issue. We proposed a robust approach based on the stacked ensemble method for predicting diabetes using several machine learning (ML) methods. The stacked ensemble comprises two models: the base model and the meta-model. Base models use a variety of models of ML, such as Support Vector Machine (SVM), K Nearest Neighbor (KNN), Naïve Bayes (NB), and Random Forest (RF), which make different assumptions about predictions, and meta-models make final predictions using Logistic Regression from predictive outputs from base models. To assess the efficiency of the proposed model, we have considered the PIMA Indian Diabetes Dataset (PIMA-IDD). We used linear and stratified sampling to ensure dataset consistency and K-fold cross-validation to prevent model overfitting. Experiments revealed that the proposed stacked ensemble model outperformed the model specified in the base classifier as well as the comprehensive methods, with an accuracy of 94.17%.

【 授权许可】

CC BY