Tree-Structured Synergy of Large Language Models and Bayesian Optimization for Efficient CASH
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 37 of its 63 references that are in the map. First posted on arXiv 18 Jan 2026.
Builds on
37 of 63 references.
References · 37
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2023Reflexion: Language Agents with Verbal Reinforcement Learningcited by 4,276
- 2019BERT4Rec: Sequential Recommendation with Bidirectional Encoder Representations from Transformercited by 964
- 2023Large Language Models as Optimizerscited by 864
- 2021Revisiting Deep Learning Models for Tabular Datacited by 740
- 2019TabNet: Attentive Interpretable Tabular Learningcited by 648
- 2023ChemCrow: Augmenting large-language models with chemistry toolscited by 631
- 2014OpenML: networked science in machine learningcited by 526
- 2018BOHB: Robust and Efficient Hyperparameter Optimization at Scalecited by 470
- 2019AutoML: A Survey of the State-of-the-Artcited by 390
- 2023Large Language Models as Commonsense Knowledge for Large-Scale Task Planningcited by 231
- 2020Auto-Sklearn 2.0: Hands-free AutoML via Meta-Learningcited by 174
25 more
- 2024Large Language Models to Enhance Bayesian Optimizationcited by 135
- 2023Large Language Models for Automated Data Science: Introducing CAAFE for Context-Aware Automated Feature Engineeringcited by 127
- 2019An Open Source AutoML Benchmarkcited by 85
- 2025ML-Master: Towards AI-for-AI via Integration of Exploration and Reasoningcited by 61
- 2019An ADMM Based Framework for AutoML Pipeline Configurationcited by 47
- 2024ADO-LLM: Analog Design Bayesian Optimization with In-Context Learning of Large Language Modelscited by 44
- 2021LightAutoML: AutoML Solution for a Large Financial Services Ecosystemcited by 32
- 2024LLMs learn governing principles of dynamical systems, revealing an in-context neural scaling lawcited by 32
- 2020Efficient Automatic CASH via Rising Banditscited by 31
- 2024In-the-loop Hyper-Parameter Optimization for LLM-Based Automated Design of Heuristicscited by 28
- 2019When does Diversity Help Generalization in Classification Ensembles?cited by 27
- 2024LICO: Large Language Models for In-Context Molecular Optimizationcited by 26
- 2021VolcanoML: Speeding up End-to-End AutoML via Scalable Search Space Decompositioncited by 25
- 2021AlphaD3M: Machine Learning Pipeline Synthesiscited by 24
- 2024Sequential Large Language Model-Based Hyper-parameter Optimizationcited by 24
- 2025Monte Carlo Planning with Large Language Model for Text-Based Game Agentscited by 22
- 2024Exploring the True Potential: Evaluating the Black-box Optimization Capability of Large Language Modelscited by 19
- 2025Language-Based Bayesian Optimization Research Assistant (BORA)cited by 15
- 2021Efficient Diversity-Driven Ensemble for Deep Neural Networkscited by 11
- 2023OpenBox: A Python Toolkit for Generalized Black-box Optimizationcited by 9
- 2023In-Context Explainers: Harnessing LLMs for Explaining Black Box Modelscited by 9
- 2025Simplifying Bayesian Optimization Via In-Context Direct Optimum Samplingcited by 7
- 2023DivBO: Diversity-aware CASH for Ensemble Learningcited by 6
- 2025Thompson Sampling via Fine-Tuning of LLMscited by 6
- 2025PSEO: Optimizing Post-hoc Stacking Ensemble Through Hyperparameter Tuningcited by 4
Built on by
Citing papers per year: 2026: 2.
2 papers.