Planning in Stochastic Environments with a Learned Model
Planning in Stochastic Environments with a Learned Model
复制标题
使用学习模型在随机环境中进行规划
DOI:
--
复制
发表时间:
2022
期刊:
影响因子:
--
通讯作者:
David Silver
中科院分区:
文献类型:
--
作者:
Ioannis Antonoglou;Julian Schrittwieser;Sherjil Ozair;Thomas K. Hubert;David Silver
Model-based reinforcement learning has proven highly successful. However, learning a model in isolation from its use during planning is problematic in complex environments. To date, the most effective techniques have instead combined value-equivalent model learning with powerful tree-search methods. This approach is exemplified by MuZero , which has achieved state-of-the-art performance in a wide range of domains, from board games to visually rich environments, with discrete and continuous action spaces, in online and offline settings. However, previous instantiations of this approach were limited to the use of deterministic models. This limits their performance in environments that are inherently stochastic, partially observed, or so large and complex that they appear stochastic to a finite agent. In this paper we extend this approach to learn and plan with stochastic models. Specifically, we introduce a new algorithm, Stochastic MuZero , that learns a stochastic model incorporating afterstates, and uses this model to perform a stochastic tree search. Stochastic MuZero matched or exceeded the state of the art in a set of canonical single and multi-agent environments, including 2048 and backgammon, while maintaining the superhuman performance of standard MuZero in the game of Go.