Summary of Mastering Atari, Go, Chess and Shogi by Planning with a Learned Model

yo3nglau

2026/05/18

Categories: Paper Reading Tags: Model-Based RL MCTS Planning Reinforcement Learning

查看中文版本

Impact and Reproducibility

MuZero’s publication in Nature (2020) cemented DeepMind’s model-based RL line as one of the most impactful in modern AI research: it demonstrated that an algorithm with no knowledge of game rules could match specialist systems in Go, chess, and shogi while simultaneously achieving state of the art across all 57 Atari games, influencing a generation of work on planning under learned models. DeepMind has not officially open-sourced MuZero, but detailed community implementations such as OpenMuZero and muzero-general are actively maintained and closely match the paper’s results. Reproducing Atari benchmarks is tractable with 4–8 NVIDIA V100s or A100s, though full training runs can take several days. Reproducing the board game results is substantially more demanding, as the original experiments relied on large TPU pods running distributed self-play; a single-machine GPU setup can approximate these results but requires weeks of training.

Summary

MuZero is a model-based reinforcement learning algorithm from DeepMind that achieves superhuman performance across both visually complex environments and precision planning tasks using a single approach. The central insight is that an agent does not need a model that reconstructs the full environment: it only needs to predict the quantities directly relevant to planning, namely the policy, value function, and immediate reward. MuZero builds an internal world model with three learned components, a representation function that encodes observations into a hidden state, a dynamics function that rolls this state forward under hypothetical actions, and a prediction function that reads off the policy and value at each step. At decision time it applies Monte Carlo Tree Search over this learned model to select actions.

Paper: Julian Schrittwieser, Ioannis Antonoglou, Thomas Hubert, Karen Simonyan, Laurent Sifre, Simon Schmitt, Arthur Guez, Edward Lockhart, Demis Hassabis, Thore Graepel, Timothy Lillicrap, David Silver. “Mastering Atari, Go, Chess and Shogi by Planning with a Learned Model.” Nature 588, 604–609 (2020). https://doi.org/10.1038/s41586-020-03051-4

Code: https://github.com/werner-duvaud/muzero-general

    / [pdf]