An Improved Ensemble-Based Machine Learning Model with Feature Optimization for Early Diabetes Prediction
Fuente:
arXiv
Gespeichert in:
| Hauptverfasser: | , , , , , |
|---|---|
| Format: | Preprint |
| Veröffentlicht: |
2025
|
| Schlagworte: | |
| Online-Zugang: | |
| Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
| _version_ | 1866918239569707008 |
|---|---|
| author | Islam, Md. Najmul Rimon, Md. Miner Hossain Shamim, Shah Sadek-E-Akbor Fahad, Zarif Mohaimen Mony, Md. Jehadul Islam Chowdhury, Md. Jalal Uddin |
| author_facet | Islam, Md. Najmul Rimon, Md. Miner Hossain Shamim, Shah Sadek-E-Akbor Fahad, Zarif Mohaimen Mony, Md. Jehadul Islam Chowdhury, Md. Jalal Uddin |
| contents | Diabetes is a serious worldwide health issue, and successful intervention depends on early detection. However, overlapping risk factors and data asymmetry make prediction difficult. To use extensive health survey data to create a machine learning framework for diabetes classification that is both accurate and comprehensible, to produce results that will aid in clinical decision-making. Using the BRFSS dataset, we assessed a number of supervised learning techniques. SMOTE and Tomek Links were used to correct class imbalance. To improve prediction performance, both individual models and ensemble techniques such as stacking were investigated. The 2015 BRFSS dataset, which includes roughly 253,680 records with 22 numerical features, is used in this study. Strong ROC-AUC performance of approximately 0.96 was attained by the individual models Random Forest, XGBoost, CatBoost, and LightGBM.The stacking ensemble with XGBoost and KNN yielded the best overall results with 94.82\% accuracy, ROC-AUC of 0.989, and PR-AUC of 0.991, indicating a favourable balance between recall and precision. In our study, we proposed and developed a React Native-based application with a Python Flask backend to support early diabetes prediction, providing users with an accessible and efficient health monitoring tool. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2512_02023 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | An Improved Ensemble-Based Machine Learning Model with Feature Optimization for Early Diabetes Prediction Islam, Md. Najmul Rimon, Md. Miner Hossain Shamim, Shah Sadek-E-Akbor Fahad, Zarif Mohaimen Mony, Md. Jehadul Islam Chowdhury, Md. Jalal Uddin Machine Learning Diabetes is a serious worldwide health issue, and successful intervention depends on early detection. However, overlapping risk factors and data asymmetry make prediction difficult. To use extensive health survey data to create a machine learning framework for diabetes classification that is both accurate and comprehensible, to produce results that will aid in clinical decision-making. Using the BRFSS dataset, we assessed a number of supervised learning techniques. SMOTE and Tomek Links were used to correct class imbalance. To improve prediction performance, both individual models and ensemble techniques such as stacking were investigated. The 2015 BRFSS dataset, which includes roughly 253,680 records with 22 numerical features, is used in this study. Strong ROC-AUC performance of approximately 0.96 was attained by the individual models Random Forest, XGBoost, CatBoost, and LightGBM.The stacking ensemble with XGBoost and KNN yielded the best overall results with 94.82\% accuracy, ROC-AUC of 0.989, and PR-AUC of 0.991, indicating a favourable balance between recall and precision. In our study, we proposed and developed a React Native-based application with a Python Flask backend to support early diabetes prediction, providing users with an accessible and efficient health monitoring tool. |
| title | An Improved Ensemble-Based Machine Learning Model with Feature Optimization for Early Diabetes Prediction |
| topic | Machine Learning |
| url | https://arxiv.org/abs/2512.02023 |