ExTS Improves Agentic Search Under Budget Constraints

Haoyang Fang, Bernie Wang· August 26, 2026 View original

Key takeaways

  • ExTS optimizes agentic search under tight budget constraints.
  • It improves upon standard MCTS by smarter exploration and exploitation.
  • Key mechanisms include reward shaping, virtual children, and quality-conditioned branching.
  • ExTS shows consistent performance gains across various complex tasks.

Who benefits

AI/ML EngineeringDrug DiscoveryMaterials ScienceSoftware DevelopmentResearch & Development

Summary

Researchers introduce ExTS, a tree-search policy that optimizes budget-constrained agentic search by treating expansion as a value-of-information decision, combining discriminative reward shaping, a stochastic virtual child, and quality-conditioned branching to outperform standard MCTS.

A new tree-search policy called ExTS (Exploit More, Explore Smarter) has been developed to enhance agentic search performance under strict budget constraints. Traditional Monte Carlo Tree Search (MCTS) often allocates budget inefficiently in scenarios where evaluations are expensive or require multiple model calls, leading to suboptimal exploration and exploitation. ExTS addresses these issues through three core mechanisms: discriminative reward shaping, which helps differentiate candidates even with narrow score distributions; a stochastic virtual child, which estimates the value of new branches based on parent reward history; and quality-conditioned branching, which only expands nodes when their score justifies the budget cost. Across diverse tasks like prompt optimization, code generation, and molecular structure elucidation, ExTS consistently matched or improved upon task-specific tree-search baselines, demonstrating an average relative gain of +5.5% with a single fixed configuration. The research also provides diagnostics to characterize different budget-constrained problems.

Why it matters

For professionals working with AI agents in resource-limited environments, ExTS offers a more efficient and effective strategy for problem-solving, potentially reducing computational costs and accelerating discovery in complex domains.

How to implement this in your domain

  1. 1Adopt ExTS for agentic workflows: Implement the ExTS tree-search policy in your budget-constrained AI agent applications.
  2. 2Optimize evaluation budgets: Analyze the cost-benefit of different evaluation strategies within agentic search to maximize efficiency.
  3. 3Apply discriminative reward shaping: Design reward functions that provide clearer signals for agent learning, even with subtle differences in candidate quality.
  4. 4Characterize problem spaces: Use the pilot-run diagnostics provided to better understand the structural properties of your agentic search problems.

Original post by Haoyang Fang, Bernie Wang

"arXiv:2608.23848v1 Announce Type: new Abstract: Budget-constrained agentic search arises when an LLM agent must refine candidates under a small evaluation budget, because validation is expensive, generation requires multiple model calls, or both. In this regime, standard MCTS all…"

View on X

Originally posted by Haoyang Fang, Bernie Wang on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses