Health Risk Classification Using XGBoost with Bayesian Hyperparameter Optimization

Health Risk Classification Using XGBoost with Bayesian Hyperparameter Optimization

health risk classification; hyperparameters; optimization; XGBoost

Health risk classification is important. However, health risk classification is challenging to address using conventional analytical techniques. The XGBoost algorithm offers many advantages over the traditional methods for risk classification. Hyperparameter Optimization (HO) of XGBoost is critical for maximizing the performance of the XGBoost algorithm. The manual selection of hyperparameters requires a large amount of time and computational resources. Automatic HO is needed to avoidthis problem. Several studies have shown that Bayesian Optimization (BO) works better than Grid Search (GS) or Random Search (RS).Based on these problems, this study proposes health risk classification using XGBoost with Bayesian Hyperparameters Optimization. The goal of this study is to reduce the time required to select the best XGBoost hyperparameters and improve the accuracy and generalization of XGBoost performance in health risk classification. The variables used were patient demographics and medical information, including age, blood pressure, cholesterol, and lifestyle variables. The experimental results show that the proposed approach outperforms other well-known ML techniques and the XGBoost method without HO. The average accuracy, precision, recall and f1-score produced by the proposed method are 0.926, 0.920, 0.928, and 0.923, respectively. However, improvements are needed to obtain a faster and more accurate method in the future.

Syaiful Anam1*, Imam N. Purwanto2, Dwi M. Mahanani3, Feby I. Yusuf4, Hady Rasikhun5

https://jurnal.iaii.or.id/index.php/RESTI/article/view/6307/1065

Department of Mathematics, Brawijaya University, Malang, Indonesia

June 13, 2025

FAJAR BAGUS W

PDF

ENGLISH

TEXT