Diabetes Prediction Model Based on SVM Optimized with RF Feature Selection and GWO
As the country with the heaviest diabetes burden, China faces significant impacts on its economic output and public health. Screening and predicting diabetes risk populations through machine learning holds practical significance. In this study, a random forest (RF) feature selection model was employed to filter a collected diabetes-related dataset, retaining eight high-importance features: glycated hemoglobin (HbA1c), fasting blood sugar (FBS), hypertension, systolic blood pressure (SBP), family history of diabetes, diet quality, diastolic blood pressure (DBP), and BMI. Among these, HbA1c and FBS exhibited significantly higher importance scores of 12.4429 and 10.7711, respectively. Subsequently, an SVM model optimized with the Grey Wolf Optimizer (GWO) was constructed based on the RF-filtered dataset and compared with PSO-SVM, BP neural network, and basic neural network models. The evaluation results were as follows: <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$A_{c c}=0.917, \text{AR}=0.907, \text{AP}=0.875, \mathrm{F} 1$</tex>-score <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$=0.891$</tex>, outperforming other models. This indicates that the proposed model demonstrates strong performance and potential in diabetes prediction.
Authors
- Junjie Xie (ORCID: https://orcid.org/0000-0002-1620-4281)
- Zhankui Jia (ORCID: https://orcid.org/0000-0001-7982-0279)
- Shihua Zhong (ORCID: https://orcid.org/0000-0002-1724-7873)
Institutions
- Shenyang Pharmaceutical University (CN)
Publication Details
- Published
- 2025-04-25
- DOI
- https://doi.org/10.1109/iccea65460.2025.11102305
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 0.00