RiskZero: Plan More to Risk Less with a Learned Model
Yousef Yassin, Junfeng Wen
Abstract
AlphaZero and MuZero have demonstrated superhuman performance across a range of strategic tasks. Yet their reliance on maximizing expected returns limits their use in real-world settings, where even high-return policies may incur rare but catastrophic failures. We introduce RiskZero to address this limitation; the first MuZero -family method for risk-sensitive decision-making, and planning with zero prior knowledge of environment dynamics. RiskZero learns distributional quantities to estimate trajectory-level risk, guiding search toward policies that explicitly avoid rare but severe outcomes. We establish theoretical convergence to optimal, stationary risk-sensitive policies and validate our approach on environments designed to test risk-sensitive learning from pixels, as well as on larger-scale combinatorial tasks. Across all settings, RiskZero consistently outperforms state-of-the-art risk-sensitive baselines, and improves sample efficiency, providing a general framework for safer and reliable model-based reinforcement learning under uncertainty.
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 3a909260-72ef-4a87-9a50-7b70af264e07Builds on21
- Mastering Atari Games with Limited DataWeirui Ye, Shaohuai Liu, Thanard Kurutach, Pieter Abbeel et al.NeurIPS 2021 · 345 citations
- Learning and Planning in Complex Action SpacesThomas Hubert, Julian Schrittwieser, Ioannis Antonoglou, Mohammadamin Barekatain et al.ICML 2021 · 99 citations
- Learning What to Defer for Maximum Independent SetsSungsoo Ahn, Younggyo Seo, Jinwoo ShinICML 2020 · 90 citations
- Being Optimistic to Be Conservative: Quickly Learning a CVaR PolicyRamtin Keramati, Christoph Dann, Alex Tamkin, Emma BrunskillAAAI 2020 · 86 citations
- Policy improvement by planning with GumbelIvo Danihelka, Arthur Guez, Julian Schrittwieser, David SilverICLR 2022 · 84 citations
Related papers
- Reinforcement Learning of Risk-Constrained Policies in Markov Decision ProcessesTomás Brázdil, Krishnendu Chatterjee, Petr Novotný, Jiri VahalaAAAI 2020 · 5 citations
- RobustZero: Enhancing MuZero Reinforcement Learning Robustness to State PerturbationsYushuai Li, Hengyu Liu, Torben Bach Pedersen, Yuqiang He et al.ICML 2025
- Multiagent Gumbel MuZero: Efficient Planning in Combinatorial Action SpacesXiaotian Hao, Jianye Hao, Chenjun Xiao, Kai Li et al.AAAI 2024 · 5 citations
- One Risk to Rule Them All: A Risk-Sensitive Perspective on Model-Based Offline Reinforcement LearningMarc Rigter, Bruno Lacerda, Nick HawesNeurIPS 2023 · 26 citations
- Epistemic Monte Carlo Tree SearchYaniv Oren, Viliam Vadocz, Matthijs T. J. Spaan, Wendelin BoehmerICLR 2025
