Diabetes Risk Prediction Model Based on Improved XGBoost with Genetic Algorithm
Diabetes is a chronic and persistent health issue. Detecting it in the prediabetes phase in a timely manner and reducing the potential complications is the best option. Therefore, it is very necessary to predict whether a patient has diabetes. This article takes the dataset from the Kaggle platform as the research object, conducts sampling data preprocessing on the diabetes dataset; selects the XGBoost model, and introduces the genetic algorithm to optimize its hyperparameters, forming an improved GA-XGBoost model based on the genetic algorithm; analyzes the SHAP values of each feature, and uses the features with greater impact as typical characteristics for whether the patient is ill; through comparative experiments, the improved model shows a significant enhancement in predictive effect compared to the traditional model.
Authors
- Wenguang Li
- Yan Peng (ORCID: https://orcid.org/0009-0002-1646-6780)
- Shi Xiong (ORCID: https://orcid.org/0009-0009-2211-8521)
Institutions
- Zigong First People's Hospital (CN)
- Sichuan University of Science and Engineering (CN)
Publication Details
- Published
- 2024-12-06
- DOI
- https://doi.org/10.1145/3727993.3728059
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 0.00