Monte-Carlo Tree Search as Regularized Policy Optimization
Not on the map. Why?
Abstract (arXiv)
The combination of Monte-Carlo tree search (MCTS) with deep reinforcement learning has led to significant advances in artificial intelligence. However, AlphaZero, the current state-of-the-art MCTS algorithm, still relies on handcrafted heuristics that are only partially understood. In this paper, we show that AlphaZero's search heuristics, along with other common ones such as UCT, are an approximation to the solution of a specific regularized policy optimization problem. With this insight, we propose a variant of AlphaZero which uses the exact solution to this policy optimization problem, and show experimentally that it reliably outperforms the original algorithm in multiple domains.
Key facts. Cited by 59 papers in the map (data through 7 Oct 2026). First posted on arXiv 24 Jul 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 16; 2022: 8; 2023: 6; 2024: 12; 2025: 10; 2026: 7.
59 papers.
Sort:
By area:
- 2024Monte Carlo Tree Search Boosts Reasoning via Iterative Preference Learningcited by 238
- 2022HyperTree Proof Search for Neural Theorem Provingcited by 192
- 2021Learning and Planning in Complex Action Spacescited by 76
- 2021Scaling Scaling Laws with Board Gamescited by 62
- 2023Don't throw away your value model! Generating more preferable text with Value-Guided Monte-Carlo Tree Search decodingcited by 59
- 2021Modeling Strong and Human-Like Gameplay with KL-Regularized Searchcited by 50
- 2025Dynamic Search for Inference-Time Alignment in Diffusion Modelscited by 36
- 2021Procedural Generalization by Planning with Self-Supervised World Modelscited by 30
- 2025KL-Regularized Reinforcement Learning is Designed to Mode Collapsecited by 29
- 2025On a few pitfalls in KL divergence gradient estimation for RLcited by 27
- 2021Scalable Online Planning via Reinforcement Learning Fine-Tuningcited by 18
- 2024UniZero: Generalized and Efficient Planning with Scalable Latent World Modelscited by 12
38 more
- 2024Large Language Models as Agents in Two-Player Gamescited by 10
- 2024Designing Skill-Compatible AI: Methodologies and Frameworks in Chesscited by 10
- 2025AssistanceZero: Scalably Solving Assistance Gamescited by 10
- 2024Human-aligned Chess with a Bit of Searchcited by 9
- 2021Measuring Progress in Deep Reinforcement Learning Sample Efficiencycited by 8
- 2021Towards an Understanding of Default Policies in Multitask Policy Optimizationcited by 8
- 2022Efficient Offline Policy Optimization with a Learned Modelcited by 8
- 2024Adaptive Draft-Verification for Efficient Large Language Model Decodingcited by 8
- 2023What model does MuZero learn?cited by 7
- 2025iTool: Reinforced Fine-Tuning with Dynamic Deficiency Calibration for Advanced Tool Usecited by 7
- 2021Fragment-based Sequential Translation for Molecular Optimizationcited by 6
- 2022A Unified Perspective on Value Backup and Exploration in Monte-Carlo Tree Searchcited by 6
- 2024Policy Mirror Descent with Lookaheadcited by 6
- 2024Efficient Monte Carlo Tree Search via On-the-Fly State-Conditioned Action Abstractioncited by 6
- 2024Value Improved Actor Critic Algorithmscited by 6
- 2021Scalable Anytime Planning for Multi-Agent MDPscited by 5
- 2022Learning to design without prior data: Discovering generalizable design strategies using deep learning and tree searchcited by 5
- 2025Trust-Region Twisted Policy Improvementcited by 5
- 2021Combining Off and On-Policy Training in Model-Based Reinforcement Learningcited by 4
- 2022Continuous Monte Carlo Graph Searchcited by 4
- 2022Spending Thinking Time Wisely: Accelerating MCTS with Virtual Expansionscited by 4
- 2024SPO: Sequential Monte Carlo Policy Optimisationcited by 4
- 2026EMA Policy Gradient: Taming Reinforcement Learning for LLMs with EMA Anchor and Top-k KLcited by 4
- 2021Planning and Learning Using Adaptive Entropy Tree Searchcited by 3
- 2023Policy-Based Self-Competition for Planning Problemscited by 3
- 2023Modeling Boundedly Rational Agents with Latent Inference Budgetscited by 3
- 2025Weak-to-Strong Generalization with Failure Trajectories: A Tree-based Approach to Elicit Optimal Policy in Strong Modelscited by 3
- 2025Twice Sequential Monte Carlo for Tree Searchcited by 3
- 2021First-Order Problem Solving through Neural MCTS based Reinforcement Learningcited by 2
- 2021Learning compositional programs with arguments and samplingcited by 2
- 2021Procrastinated Tree Search: Black-box Optimization with Delayed, Noisy, and Multi-Fidelity Feedbackcited by 2
- 2022Policy Gradient Algorithms with Monte Carlo Tree Learning for Non-Markov Decision Processescited by 2
- 2022Brick Tic-Tac-Toe: Exploring the Generalizability of AlphaZero to Novel Test Environmentscited by 2
- 2023Autonomous Port Navigation With Ranging Sensors Using Model-Based Reinforcement Learningcited by 2
- 2024Provably Efficient Long-Horizon Exploration in Monte Carlo Tree Search through State Occupancy Regularizationcited by 1
- 2026Revisiting Regularized Policy Optimization for Stable and Efficient Reinforcement Learning in Two-Player Gamescited by 1
- 2021ScrofaZero: Mastering Trick-taking Poker Game Gongzhu by Deep Reinforcement Learningcited by 0
- 2021Dual Monte Carlo Tree Searchcited by 0
50 most cited of 59 All 59 (JSON)
- 2026Towards Optimal Policy Improvement1 Oct 2026
- 2026OPTS-TTPO: Enhancing Finite-Sample Policy-Gradient Learning with Tree Search30 Sep 2026
- 2026The Surprising Effectiveness of Approximate Value Iteration in Self-Play8 Sep 2026
- 2026Particle Monte Carlo Tree Search9 May 2026
- 2026Revisiting Regularized Policy Optimization for Stable and Efficient Reinforcement Learning in Two-Player Games11 Feb 2026
- 2026EMA Policy Gradient: Taming Reinforcement Learning for LLMs with EMA Anchor and Top-k KL4 Feb 2026
- 2026Accelerating Monte-Carlo Tree Search with Optimized Posterior Policies3 Jan 2026
- 2025Twice Sequential Monte Carlo for Tree Search18 Nov 2025
- 2025KL-Regularized Reinforcement Learning is Designed to Mode Collapse23 Oct 2025
- 2025A Markov Decision Process for Variable Selection in Branch & Bound22 Oct 2025
- 2025Weak-to-Strong Generalization with Failure Trajectories: A Tree-based Approach to Elicit Optimal Policy in Strong Models25 Jul 2025
- 2025On a few pitfalls in KL divergence gradient estimation for RL11 Jun 2025
38 more
- 2025When Search Teaches Style: Causal Internalization of Tactical Priors in AlphaZero20 Apr 2025
- 2025AssistanceZero: Scalably Solving Assistance Games9 Apr 2025
- 2025Trust-Region Twisted Policy Improvement8 Apr 2025
- 2025Dynamic Search for Inference-Time Alignment in Diffusion Models3 Mar 2025
- 2025iTool: Reinforced Fine-Tuning with Dynamic Deficiency Calibration for Advanced Tool Use15 Jan 2025
- 2024Human-aligned Chess with a Bit of Search4 Oct 2024
- 2024Provably Efficient Long-Horizon Exploration in Monte Carlo Tree Search through State Occupancy Regularization7 Jul 2024
- 2024Adaptive Draft-Verification for Efficient Large Language Model Decoding27 Jun 2024
- 2024Improved Monte Carlo Planning via Causal Disentanglement for Structurally-Decomposed Markov Decision Processes23 Jun 2024
- 2024UniZero: Generalized and Efficient Planning with Scalable Latent World Models15 Jun 2024
- 2024Value Improved Actor Critic Algorithms3 Jun 2024
- 2024Efficient Monte Carlo Tree Search via On-the-Fly State-Conditioned Action Abstraction2 Jun 2024
- 2024Designing Skill-Compatible AI: Methodologies and Frameworks in Chess8 May 2024
- 2024Monte Carlo Tree Search Boosts Reasoning via Iterative Preference Learning1 May 2024
- 2024Policy Mirror Descent with Lookahead21 Mar 2024
- 2024Large Language Models as Agents in Two-Player Games12 Feb 2024
- 2024SPO: Sequential Monte Carlo Policy Optimisation12 Feb 2024
- 2023Modeling Boundedly Rational Agents with Latent Inference Budgets7 Dec 2023
- 2023Autonomous Port Navigation With Ranging Sensors Using Model-Based Reinforcement Learning17 Nov 2023
- 2023Don't throw away your value model! Generating more preferable text with Value-Guided Monte-Carlo Tree Search decoding26 Sep 2023
- 2023Acceleration in Policy Optimization18 Jun 2023
- 2023Policy-Based Self-Competition for Planning Problems7 Jun 2023
- 2023What model does MuZero learn?1 Jun 2023
- 2022Learning to design without prior data: Discovering generalizable design strategies using deep learning and tree search28 Nov 2022
- 2022Spending Thinking Time Wisely: Accelerating MCTS with Virtual Expansions23 Oct 2022
- 2022Efficient Offline Policy Optimization with a Learned Model12 Oct 2022
- 2022Continuous Monte Carlo Graph Search4 Oct 2022
- 2022Brick Tic-Tac-Toe: Exploring the Generalizability of AlphaZero to Novel Test Environments13 Jul 2022
- 2022Policy Gradient Algorithms with Monte Carlo Tree Learning for Non-Markov Decision Processes2 Jun 2022
- 2022HyperTree Proof Search for Neural Theorem Proving23 May 2022
- 2022A Unified Perspective on Value Backup and Exploration in Monte-Carlo Tree Search11 Feb 2022
- 2021Modeling Strong and Human-Like Gameplay with KL-Regularized Search14 Dec 2021
- 2021Towards an Understanding of Default Policies in Multitask Policy Optimization4 Nov 2021
- 2021Procedural Generalization by Planning with Self-Supervised World Models2 Nov 2021
- 2021Fragment-based Sequential Translation for Molecular Optimization26 Oct 2021
- 2021Procrastinated Tree Search: Black-box Optimization with Delayed, Noisy, and Multi-Fidelity Feedback14 Oct 2021
- 2021Scalable Online Planning via Reinforcement Learning Fine-Tuning30 Sep 2021
- 2021Learning compositional programs with arguments and sampling1 Sep 2021
50 newest of 59 All 59 (JSON)
- 2024Monte Carlo Tree Search Boosts Reasoning via Iterative Preference Learningcited by 238
- 2023Don't throw away your value model! Generating more preferable text with Value-Guided Monte-Carlo Tree Search decodingcited by 59
- 2026EMA Policy Gradient: Taming Reinforcement Learning for LLMs with EMA Anchor and Top-k KLcited by 4
- 2026EMA Policy Gradient: Taming Reinforcement Learning for LLMs with EMA Anchor and Top-k KL4 Feb 2026
- 2024Monte Carlo Tree Search Boosts Reasoning via Iterative Preference Learning1 May 2024
- 2023Don't throw away your value model! Generating more preferable text with Value-Guided Monte-Carlo Tree Search decoding26 Sep 2023
- 2024Large Language Models as Agents in Two-Player Gamescited by 10
- 2024Large Language Models as Agents in Two-Player Games12 Feb 2024
- 2025AssistanceZero: Scalably Solving Assistance Gamescited by 10
- 2025AssistanceZero: Scalably Solving Assistance Games9 Apr 2025
- 2022HyperTree Proof Search for Neural Theorem Provingcited by 192
- 2021Learning and Planning in Complex Action Spacescited by 76
- 2021Scaling Scaling Laws with Board Gamescited by 62
- 2025KL-Regularized Reinforcement Learning is Designed to Mode Collapsecited by 29
- 2025On a few pitfalls in KL divergence gradient estimation for RLcited by 27
- 2021Scalable Online Planning via Reinforcement Learning Fine-Tuningcited by 18
- 2024Designing Skill-Compatible AI: Methodologies and Frameworks in Chesscited by 10
- 2024Human-aligned Chess with a Bit of Searchcited by 9
- 2021Measuring Progress in Deep Reinforcement Learning Sample Efficiencycited by 8
- 2021Towards an Understanding of Default Policies in Multitask Policy Optimizationcited by 8
- 2024Adaptive Draft-Verification for Efficient Large Language Model Decodingcited by 8
- 2023What model does MuZero learn?cited by 7
12 most cited of 49 All 49 (JSON)
- 2026Towards Optimal Policy Improvement1 Oct 2026
- 2026OPTS-TTPO: Enhancing Finite-Sample Policy-Gradient Learning with Tree Search30 Sep 2026
- 2026The Surprising Effectiveness of Approximate Value Iteration in Self-Play8 Sep 2026
- 2026Particle Monte Carlo Tree Search9 May 2026
- 2026Revisiting Regularized Policy Optimization for Stable and Efficient Reinforcement Learning in Two-Player Games11 Feb 2026
- 2026Accelerating Monte-Carlo Tree Search with Optimized Posterior Policies3 Jan 2026
- 2025Twice Sequential Monte Carlo for Tree Search18 Nov 2025
- 2025KL-Regularized Reinforcement Learning is Designed to Mode Collapse23 Oct 2025
- 2025A Markov Decision Process for Variable Selection in Branch & Bound22 Oct 2025
- 2025Weak-to-Strong Generalization with Failure Trajectories: A Tree-based Approach to Elicit Optimal Policy in Strong Models25 Jul 2025
- 2025On a few pitfalls in KL divergence gradient estimation for RL11 Jun 2025
- 2025When Search Teaches Style: Causal Internalization of Tactical Priors in AlphaZero20 Apr 2025
12 newest of 49 All 49 (JSON)