Response surface-driven hyperparameter optimization for XGBoost

Vasquez-Ramos J.; Ruiz-Sandoval M.G.; Oliva D.; Ramos-Soto O.; Ramos-Frutos J.; Sharawi M.; Pérez-Cisneros M. · Journal of Supercomputing (2025) · DOI: 10.1007/s11227-025-07600-4

Resumen

This work uses the response surface methodology (RSM) to determine the hyperparameters with which the XGBoost should work to improve its specificity in classifying patients with breast cancer. The algorithm was compared against two other machine learning algorithms, and its results were the worst compared to those of the other two algorithms. Therefore, the XGBoost hyperparameters were adjusted using the RSM to measure the effectiveness of the RSM in tuning the machine learning algorithms. A Box–Behnken experimental design was used to adjust the hyperparameters, and the response variable was specificity. The average specificity at the beginning was around 93% in the breast cancer problem. When applying the RSM to adjust the hyperparameters, there is an increase in specificity of more than 5%, reaching more than 98% in this variable. Furthermore, compared to the other algorithms, there are also improvements. Better results in specificity are obtained with RSM compared to hyperparameter tuning using Metaheuristics, Grid Search, and Bayesian Optimization in this study. © The Author(s), under exclusive licence to Springer Science+Business Media, LLC, part of Springer Nature 2025.

Instituciones: Universidad Nacional de Ingeniería