Centrum: Model-based Database Auto-tuning with Minimal Distributional Assumptions
Yuanhao Lai, Pengfei Zheng, Chenpeng Ji, Yan Li, Songhan Zhang, Rutao Zhang, Zhengang Wang, Yunfei Du
Abstract
Gaussian Process (GP)-based Bayesian optimization (BO), i.e., GP-BO, emerges as a prevailing model-based framework for DBMS (Database Management System) auto-tuning. However, recent work shows GP-BO-based DBMS auto-tuners are significantly outperformed by auto-tuners based on SMAC, which features random forest surrogate models; such results motivate us to rethink and investigate the limitations of GP-BO in auto-tuner design. We find that the fundamental assumptions of GP-BO are widely violated when modeling and optimizing DBMS performance, while tree-ensemble-BOs (e.g., SMAC) can avoid the assumption pitfalls and deliver improved tuning efficiency and effectiveness. Moreover, we argue that existing tree-ensemble-BOs restrict further advancement in DBMS auto-tuning. First, existing tree-ensemble-BOs can only achieve distribution-free point estimates, but still impose unrealistic distributional assumptions on uncertainty (interval) estimates, which can compromise surrogate modeling and distort the acquisition function. Second, recent advances in (ensemble) gradient boosting, which can further enhance surrogate modeling against vanilla GP and random forest counterparts, have rarely been applied in optimizing DBMS auto-tuners. To address these issues, we propose a novel model-based DBMS auto-tuner, Centrum . Centrum achieves and improves distribution-free point and interval estimation in surrogate modeling with a two-phase learning procedure of stochastic gradient boosting ensembles (SGBE). Moreover, Centrum adopts a generalized SGBE-estimated locally-adaptive conformal prediction to facilitate a distribution-free interval (uncertainty) estimation and acquisition function. To our knowledge, Centrum is the first auto-tuner that realizes distribution-freeness to stress and enhance BO's practicality in DBMS auto-tuning, and the first to seamlessly fuse gradient boosting ensembles and conformal inference in BO. Extensive physical and simulation experiments on two DBMSs and three workloads show that Centrum outperforms 21 state-of-the-art (SOTA) DBMS auto-tuners based on BO with GP, random forest, gradient boosting, OOB (Out-Of-Bag) conformal ensemble and other surrogates, as well as that based on reinforcement learning and genetic algorithms.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on18
- Revisiting Deep Learning Models for Tabular DataYury Gorishniy, Ivan Rubachev, Valentin Khrulkov, Artem BabenkoNeurIPS 2021 · 1,847 citations
- NGBoost: Natural Gradient Boosting for Probabilistic PredictionTony Duan, Anand Avati, Daisy Yi Ding, Khanh K. Thai et al.ICML 2020 · 433 citations
- Unexpected Improvements to Expected Improvement for Bayesian OptimizationSebastian Ament, Samuel Daulton, David Eriksson, Maximilian Balandat et al.NeurIPS 2023 · 280 citations
- Conformal prediction interval for dynamic time-seriesChen Xu, Yao XieICML 2021 · 174 citations
- Learning Search Space Partition for Black-box Optimization using Monte Carlo Tree SearchLinnan Wang, Rodrigo Fonseca, Yuandong TianNeurIPS 2020 · 163 citations
Related papers
- CoffeeBoost: Gradient Boosting Native Conformal Inference for Bayesian OptimizationYuanhao Lai, Pengfei Zheng, Chenpeng Ji, Cheng Qiu et al.AAAI 2025 · 1 citation
- TopTune: Tailored Optimization for Categorical and Continuous Knobs Towards Accelerated and Improved Database Performance TuningRukai Wei, Yu Liu, Yufeng Hou, Heng Cui et al.ICDE 2025 · 1 citation
- Conformal Prediction as Bayesian QuadratureJake C. Snell, Thomas L. GriffithsICML 2025
- Boosted Conformal Prediction IntervalsRan Xie, Rina Barber, Emmanuel J. CandèsNeurIPS 2024 · 34 citations
- DOT: Dynamic Knob Selection and Online Sampling for Automated Database TuningYifan Wang, Debabrota Basu, Pierre Bourhis, Romain Rouvoy et al.VLDB 2026
