arXiv AI By Jianguo Huang, Zhiyi Lyu, Fuxiang Zhang, Yanchen Deng, Deheng Ye, Bo An

Uncertainty-Aware Budget Allocation for Efficient LLM Tree Search

Read the original on arXiv AI →

The paper introduces Uncertainty-Aware Allocation (UAA), a training‑free framework that adjusts the per‑step budget in large language model tree search based on sibling action dispersion. By estimating uncertainty from sibling dispersion, UAA decouples value from uncertainty, allowing selective expansion and pruning. Experiments on mathematical reasoning, logical reasoning, and planning tasks show that UAA can cut token usage by up to 90% and latency by up to 9× while maintaining comparable accuracy across multiple tree search algorithms and both open‑ and closed‑source LLMs.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

arXiv Computation and Language
Oct 9

Uncertainty-Aware Budget Allocation for Adaptive Test-Time Reasoning

The paper introduces Uncertainty-Aware Budget Allocation (UAB), a concave integer optimization framework that reallocates a fixed sampling budget for language model reasoning based on uncertainty estimated from initial samples. In Phase‑1, each question receives a small number of generations, and the disagreement among these answers (vote entropy) signals difficulty. Phase‑2 uses a marginal‑greedy algorithm to allocate the remaining budget, focusing on questions with high initial disagreement, leading to a 2.3% average accuracy improvement over uniform allocation across five open‑weight models and five reasoning benchmarks, with up to 5.5% gains on individual benchmarks, especially in low‑resource settings.

By Manh Nguyen, Sunil Gupta, Hung Le
arXiv Machine Learning
5d ago

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

The paper introduces LB-MCTS, a trajectory-structured optimization framework that combines Monte Carlo Tree Search, Bayesian Optimization, and Large Language Models to address the CASH problem in AutoML. It uses a shared tree state for algorithm selection, hyperparameter refinement, and synergy between BO and LLM proposals, with a reliability-aware policy that shifts from LLM-driven to BO-driven suggestions as the surrogate model improves. Experiments on 104 AMLB datasets show that LB-MCTS consistently outperforms existing BO-based, LLM-based, and hybrid baselines.

By Beicheng Xu, Weitong Qian, Lingching Tung, Yupeng Lu, Bin Cui
arXiv AI
Aug 26

Exploit More, Explore Smarter for Budget-Constrained Agentic Search

The paper introduces ExTS, a tree‑search policy designed for budget‑constrained agentic search where evaluation and generation costs are high. ExTS treats expansion as a value‑of‑information decision, combining discriminative reward shaping, a stochastic virtual child, and quality‑conditioned branching to allocate budget more effectively. Experiments on prompt optimization, code generation, molecular structure elucidation, and agentic workflow optimization show ExTS matching or surpassing task‑specific baselines with an average gain of +5.5% using a single configuration, and the authors also present pilot‑run diagnostics to guide adaptation to different problem structures.

By Haoyang Fang, Bernie Wang