Spending Thinking Time Wisely: Accelerating MCTS with Virtual Expansions
Weirui Ye, Pieter Abbeel, Yang Gao
Abstract
One of the most important AI research questions is to trade off computation versus performance since ``perfect rationality"exists in theory but is impossible to achieve in practice. Recently, Monte-Carlo tree search (MCTS) has attracted considerable attention due to the significant performance improvement in various challenging domains. However, the expensive time cost during search severely restricts its scope for applications. This paper proposes the Virtual MCTS (V-MCTS), a variant of MCTS that spends more search time on harder states and less search time on simpler states adaptively. We give theoretical bounds of the proposed method and evaluate the performance and computations on Go board games and Atari games. Experiments show that our method can achieve comparable performances to the original search algorithm while requiring less than search time on average. We believe that this approach is a viable alternative for tasks under limited time and resources. The code is available at https://github.com/YeWR/V-MCTS.git.
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 9ca4d422-929a-400f-ad45-15b1962622f0Cited by top-tier papers2
- Planning for Sample Efficient Imitation LearningZhao-Heng Yin, Weirui Ye, Qifeng Chen, Yang GaoNeurIPS 2022 · 32 citations
- Accelerating Monte Carlo Tree Search with Probability Tree State AbstractionYangqing Fu, Ming Sun, Buqing Nie, Yue GaoNeurIPS 2023 · 5 citations
Builds on4
- Mastering Atari Games with Limited DataWeirui Ye, Shaohuai Liu, Thanard Kurutach, Pieter Abbeel et al.NeurIPS 2021 · 345 citations
- Policy improvement by planning with GumbelIvo Danihelka, Arthur Guez, Julian Schrittwieser, David SilverICLR 2022 · 84 citations
- Monte-Carlo Tree Search as Regularized Policy OptimizationJean-Bastien Grill, Florent Altché, Yunhao Tang, Thomas Hubert et al.ICML 2020 · 84 citations
- Learning to Stop: Dynamic Simulation Monte-Carlo Tree SearchLi-Cheng Lan, Ti-Rong Wu, I-Chen Wu, Cho-Jui HsiehAAAI 2021 · 7 citations
Related papers
- Watch the Unobserved: A Simple Approach to Parallelizing Monte Carlo Tree SearchAnji Liu, Jianshu Chen, Mingze Yu, Yu Zhai et al.ICLR 2020 · 39 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
- Bilevel MCTS for Amortized O(1) Node Selection in Classical PlanningMasataro AsaiAAAI 2026
- Monte Carlo Tree Search With Iteratively Refining State AbstractionsSamuel Sokota, Caleb Ho, Zaheen Farraz Ahmad, J. Zico KolterNeurIPS 2021 · 22 citations
- Monte Carlo Tree Search based Variable Selection for High Dimensional Bayesian OptimizationLei Song, Ke Xue, Xiaobin Huang, Chao QianNeurIPS 2022 · 57 citations
