Monte Carlo Tree Search With Iteratively Refining State Abstractions
Samuel Sokota, Caleb Ho, Zaheen Farraz Ahmad, J. Zico Kolter
Abstract
Decision-time planning is the process of constructing a transient, local policy with the intent of using it to make the immediate decision. Monte Carlo tree search (MCTS), which has been leveraged to great success in Go, chess, shogi, Hex, Atari, and other settings, is perhaps the most celebrated decision-time planning algorithm. Unfortunately, in its original form, MCTS can degenerate to one-step search in domains with stochasticity. Progressive widening is one way to ameliorate this issue, but we argue that it possesses undesirable properties for some settings. In this work, we present a method, called abstraction refining, for extending MCTS to stochastic environments which, unlike progressive widening, leverages the geometry of the state space. We argue that leveraging the geometry of the space can offer advantages. To support this claim, we present a series of experimental examples in which abstraction refining outperforms progressive widening, given equal simulation budgets.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext e8882112-ef3f-4239-9f51-ec5a5e310b2aCited by top-tier papers2
- Wider or Deeper? Scaling LLM Inference-Time Compute with Adaptive Branching Tree SearchYuichi Inoue, Kou Misaki, Yuki Imajuku, So Kuroki et al.NeurIPS 2025 · 67 citations
- Accelerating Monte Carlo Tree Search with Probability Tree State AbstractionYangqing Fu, Ming Sun, Buqing Nie, Yue GaoNeurIPS 2023 · 5 citations
Builds on3
- Scalable Methods for Computing State Similarity in Deterministic Markov Decision ProcessesPablo Samuel CastroAAAI 2020 · 171 citations
- Learning Invariant Representations for Reinforcement Learning without ReconstructionAmy Zhang, Rowan Thomas McAllister, Roberto Calandra, Yarin Gal et al.ICLR 2021 · 77 citations
- Monte Carlo Tree Search in Continuous Spaces Using Voronoi Optimistic Optimization with Regret BoundsBeomjoon Kim, Kyungjae Lee, Sungbin Lim, Leslie Pack Kaelbling et al.AAAI 2020 · 55 citations
Related papers
- Bilevel MCTS for Amortized O(1) Node Selection in Classical PlanningMasataro AsaiAAAI 2026
- Learning to Stop: Dynamic Simulation Monte-Carlo Tree SearchLi-Cheng Lan, Ti-Rong Wu, I-Chen Wu, Cho-Jui HsiehAAAI 2021 · 7 citations
- Monte Carlo Tree Search in the Presence of Transition UncertaintyFarnaz Kohankhaki, Kiarash Aghakasiri, Hongming Zhang, Ting-Han Wei et al.AAAI 2024 · 4 citations
- Monte Carlo Tree Diffusion for System 2 PlanningJaesik Yoon, Hyeonseo Cho, Doojin Baek, Yoshua Bengio et al.ICML 2025
- MonteFloor: Extending MCTS for Reconstructing Accurate Large-Scale Floor PlansSinisa Stekovic, Mahdi Rad, Friedrich Fraundorfer, Vincent LepetitICCV 2021 · 43 citations
