cs.LGJan 18, 2026

Tree-Structured Synergy of Large Language Models and Bayesian Optimization for Efficient CASH

Authors: Beicheng Xu, Weitong Qian, Lingching Tung, Yupeng Lu, Bin Cui

Organizations: School of Computer Science, Peking University, Beijing, China

Abstract

To lower the expertise barrier in machine learning, the AutoML community has focused on the CASH problem, which jointly automates algorithm selection and hyperparameter tuning. While traditional methods like Bayesian Optimization (BO) struggle with cold-start issues, Large Language Models (LLMs) can mitigate these through semantic priors. However, existing LLM-based optimizers generalize poorly to high-dimensional, structured CASH spaces. CASH requires coordinated exploration and exploitation both across candidate algorithms and within their algorithm-specific hyperparameter spaces. Yet existing approaches often rely on flat optimization histories that mix observations across algorithms and obscure configuration-refinement trajectories. In this paper, we propose LB-MCTS, a trajectory-structured optimization framework that uses a Monte Carlo Tree Search tree as a shared state for algorithm selection, hyperparameter refinement, and BO-LLM proposer synergy. Within this shared state, BO provides algorithm-specific surrogate modeling for quantitative search, while the LLM exploits path-aware selective memory to generate semantic proposals and reflections. As the surrogate model improves, a reliability-aware proposer policy adaptively shifts from LLM-driven to BO-driven proposals within a unified search trajectory. Experiments on 104 AMLB datasets demonstrate that LB-MCTS consistently outperforms BO-based, LLM-based, and hybrid baselines. Code: https://github.com/PKU-DAIR/open-box.

Figures & tables

Appendix figures & tables13 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Can LLMs Beat Classical Hyperparameter Optimization Algorithms? A Study on autoresearch

    Mar 25, 2026Fabio Ferreira, Lucca Wobbe, Arjun Krishnakumar +2Hyperparameter OptimizationLarge Language Model-Guided Optimization

  2. DASH: Decoupled Adaptive Surrogate - Acquisition Harness for Automated Bayesian Optimization

    Aug 1, 2026Changquan Zhao, Yuxiang Sun, Ruihao Zhu +2Bayesian OptimizationLarge Language Model-Guided Optimization

  3. Bayesian Optimization with Rich Auxiliary Information via LLMs

    Sep 16, 2026Efe Mert Karagözlü, Rohit Sonker, Tejus Gupta +2Bayesian OptimizationLarge Language Model-Guided Optimization