Skip to main navigation Skip to search Skip to main content

Response surface-driven hyperparameter optimization for XGBoost

  • Jair Vasquez-Ramos
  • , María Guadalupe Ruiz-Sandoval
  • , Diego Oliva
  • , Oscar Ramos-Soto
  • , Jorge Ramos-Frutos
  • , Marwa Sharawi
  • , Marco Pérez-Cisneros

Research output: Contribution to journalArticlepeer-review

5 Scopus citations

Abstract

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.

Original languageEnglish
Article number10
Pages (from-to)1112
Number of pages1
JournalJournal of Supercomputing
Volume81
Issue number10
DOIs
StatePublished - 7 Jul 2025

UN SDGs

This output contributes to the following UN Sustainable Development Goals (SDGs)

  1. SDG 3 - Good Health and Well-being
    SDG 3 Good Health and Well-being

Keywords

  • Machine learning
  • Parameters
  • Response surface methodology
  • XGBoost

Fingerprint

Dive into the research topics of 'Response surface-driven hyperparameter optimization for XGBoost'. Together they form a unique fingerprint.

Cite this