Self-Correcting Models for Model-Based Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
When an agent cannot represent a perfectly accurate model of its environment's dynamics, model-based reinforcement learning (MBRL) can fail catastrophically. Planning involves composing the predictions of the model; when flawed predictions are composed, even minor errors can compound and render the model useless for planning. Hallucinated Replay (Talvitie 2014) trains the model to "correct" itself when it produces errors, substantially improving MBRL with flawed models. This paper theoretically analyzes this approach, illuminates settings in which it is likely to be effective or ineffective, and presents a novel error bound, showing that a model's ability to self-correct is more tightly related to MBRL performance than one-step prediction error. These results inspire an MBRL algorithm for deterministic MDPs with performance guarantees that are robust to model class limitations.
Key facts. Cited by 40 papers in the map (data through 7 Oct 2026). First posted on arXiv 19 Dec 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2018: 1; 2019: 0; 2020: 0; 2021: 5; 2022: 5; 2023: 8; 2024: 8; 2025: 6; 2026: 7.
40 papers.
Sort:
By area:
- 2021Learning One Representation to Optimize All Rewardscited by 110
- 2022Value Gradient weighted Model-Based Reinforcement Learningcited by 31
- 2022Investigating the Properties of Neural Network Representations in Reinforcement Learningcited by 24
- 2022Goal-Space Planning with Subgoal Modelscited by 12
- 2023Off-Policy RL Algorithms Can be Sample-Efficient for Continuous Control via Sample Multiple Reusecited by 12
- 2024Locality Sensitive Sparse Encoding for Learning World Models Onlinecited by 12
- 2021High-dimensional separability for one- and few-shot learningcited by 11
- 2021Temporally Abstract Partial Modelscited by 11
- 2023Harnessing Discrete Representations For Continual Reinforcement Learningcited by 10
- 2025On Rollouts in Model-Based Reinforcement Learningcited by 10
- 2023A Unified View on Solving Objective Mismatch in Model-Based Reinforcement Learningcited by 9
- 2024A Note on Loss Functions and Error Compounding in Model-based Reinforcement Learningcited by 7
28 more
- 2021On-Policy Model Errors in Reinforcement Learningcited by 6
- 2022No More Pesky Hyperparameters: Offline Hyperparameter Tuning for RLcited by 6
- 2025Theoretical Modeling of Large Language Model Self-Improvement Training Dynamics Through Solver-Verifier Gapcited by 6
- 2018Learning the Reward Function for a Misspecified Modelcited by 4
- 2026Laplacian Representations for Decision-Time Planningcited by 4
- 2023Monte Carlo Tree Search in the Presence of Transition Uncertaintycited by 3
- 2026WIMLE: Uncertainty-Aware World Models with IMLE for Sample-Efficient Continuous Controlcited by 3
- 2026A Definition and Roadmap for World Modelscited by 3
- 2023A Tractable Inference Perspective of Offline RLcited by 2
- 2023Maximum Entropy Model Correction in Reinforcement Learningcited by 2
- 2025Calibrated Value-Aware Model Learning with Probabilistic Environment Modelscited by 2
- 2026Looped World Modelscited by 2
- 2026The Intervention Gap in Latent World Modelscited by 2
- 2023λ-models: Effective Decision-Aware Reinforcement Learning with Latent Modelscited by 1
- 2023Multi-timestep models for Model-based Reinforcement Learningcited by 1
- 2024A Multi-step Loss Function for Robust Learning of the Dynamics in Model-based Reinforcement Learningcited by 1
- 2024Sequence Compression Speeds Up Credit Assignment in Reinforcement Learningcited by 1
- 2024Bounding-Box Inference for Error-Aware Model-Based Reinforcement Learningcited by 1
- 2024Why long model-based rollouts are no reason for bad Q-value estimatescited by 1
- 2024Rejecting Hallucinated State Targets during Planningcited by 1
- 2025Situationally-Aware Dynamics Learningcited by 1
- 2021Planning with Expectation Models for Controlcited by 0
- 2022Reward-Predictive Clusteringcited by 0
- 2024Meta-Gradient Search Control: A Method for Improving the Efficiency of Dyna-style Planningcited by 0
- 2025Model-Based Offline Reinforcement Learning with Reliability-Guaranteed Sequence Modelingcited by 0
- 2025Planning as Descent: Goal-Conditioned Latent Trajectory Synthesis in Learned Energy Landscapescited by 0
- 2026Beyond Myopic World Models: Long-Horizon End-to-End Training for Direct Future Predictioncited by 0
- 2026Dynamics Models for Offline Hyperparameter Selection in Real-World RLcited by 0
- 2026The Intervention Gap in Latent World Models30 Aug 2026
- 2026Dynamics Models for Offline Hyperparameter Selection in Real-World RL11 Aug 2026
- 2026Beyond Myopic World Models: Long-Horizon End-to-End Training for Direct Future Prediction7 Aug 2026
- 2026A Definition and Roadmap for World Models7 Jul 2026
- 2026Looped World Models16 Jun 2026
- 2026WIMLE: Uncertainty-Aware World Models with IMLE for Sample-Efficient Continuous Control15 Feb 2026
- 2026Laplacian Representations for Decision-Time Planning4 Feb 2026
- 2025Planning as Descent: Goal-Conditioned Latent Trajectory Synthesis in Learned Energy Landscapes19 Dec 2025
- 2025Theoretical Modeling of Large Language Model Self-Improvement Training Dynamics Through Solver-Verifier Gap29 Jun 2025
- 2025Calibrated Value-Aware Model Learning with Probabilistic Environment Models28 May 2025
- 2025Situationally-Aware Dynamics Learning26 May 2025
- 2025Model-Based Offline Reinforcement Learning with Reliability-Guaranteed Sequence Modeling10 Feb 2025
28 more
- 2025On Rollouts in Model-Based Reinforcement Learning28 Jan 2025
- 2024Rejecting Hallucinated State Targets during Planning9 Oct 2024
- 2024Why long model-based rollouts are no reason for bad Q-value estimates16 Jul 2024
- 2024Meta-Gradient Search Control: A Method for Improving the Efficiency of Dyna-style Planning27 Jun 2024
- 2024Bounding-Box Inference for Error-Aware Model-Based Reinforcement Learning23 Jun 2024
- 2024Sequence Compression Speeds Up Credit Assignment in Reinforcement Learning6 May 2024
- 2024A Note on Loss Functions and Error Compounding in Model-based Reinforcement Learning15 Apr 2024
- 2024A Multi-step Loss Function for Robust Learning of the Dynamics in Model-based Reinforcement Learning5 Feb 2024
- 2024Locality Sensitive Sparse Encoding for Learning World Models Online23 Jan 2024
- 2023Monte Carlo Tree Search in the Presence of Transition Uncertainty18 Dec 2023
- 2023Harnessing Discrete Representations For Continual Reinforcement Learning2 Dec 2023
- 2023Maximum Entropy Model Correction in Reinforcement Learning29 Nov 2023
- 2023A Tractable Inference Perspective of Offline RL31 Oct 2023
- 2023A Unified View on Solving Objective Mismatch in Model-Based Reinforcement Learning10 Oct 2023
- 2023Multi-timestep models for Model-based Reinforcement Learning9 Oct 2023
- 2023λ-models: Effective Decision-Aware Reinforcement Learning with Latent Models30 Jun 2023
- 2023Off-Policy RL Algorithms Can be Sample-Efficient for Continuous Control via Sample Multiple Reuse29 May 2023
- 2022Reward-Predictive Clustering7 Nov 2022
- 2022Goal-Space Planning with Subgoal Models6 Jun 2022
- 2022No More Pesky Hyperparameters: Offline Hyperparameter Tuning for RL18 May 2022
- 2022Value Gradient weighted Model-Based Reinforcement Learning4 Apr 2022
- 2022Investigating the Properties of Neural Network Representations in Reinforcement Learning30 Mar 2022
- 2021On-Policy Model Errors in Reinforcement Learning15 Oct 2021
- 2021Temporally Abstract Partial Models6 Aug 2021
- 2021High-dimensional separability for one- and few-shot learning28 Jun 2021
- 2021Planning with Expectation Models for Control17 Apr 2021
- 2021Learning One Representation to Optimize All Rewards14 Mar 2021
- 2018Learning the Reward Function for a Misspecified Model29 Jan 2018
- 2024Locality Sensitive Sparse Encoding for Learning World Models Onlinecited by 12
- 2023Harnessing Discrete Representations For Continual Reinforcement Learningcited by 10
- 2026WIMLE: Uncertainty-Aware World Models with IMLE for Sample-Efficient Continuous Controlcited by 3
- 2026A Definition and Roadmap for World Modelscited by 3
- 2026Looped World Modelscited by 2
- 2026The Intervention Gap in Latent World Modelscited by 2
- 2026Beyond Myopic World Models: Long-Horizon End-to-End Training for Direct Future Predictioncited by 0
- 2026The Intervention Gap in Latent World Models30 Aug 2026
- 2026Beyond Myopic World Models: Long-Horizon End-to-End Training for Direct Future Prediction7 Aug 2026
- 2026A Definition and Roadmap for World Models7 Jul 2026
- 2026Looped World Models16 Jun 2026
- 2026WIMLE: Uncertainty-Aware World Models with IMLE for Sample-Efficient Continuous Control15 Feb 2026
- 2024Locality Sensitive Sparse Encoding for Learning World Models Online23 Jan 2024
- 2023Harnessing Discrete Representations For Continual Reinforcement Learning2 Dec 2023
- 2023A Tractable Inference Perspective of Offline RLcited by 2
- 2023Multi-timestep models for Model-based Reinforcement Learningcited by 1
- 2024Why long model-based rollouts are no reason for bad Q-value estimatescited by 1
- 2025Model-Based Offline Reinforcement Learning with Reliability-Guaranteed Sequence Modelingcited by 0
- 2025Model-Based Offline Reinforcement Learning with Reliability-Guaranteed Sequence Modeling10 Feb 2025
- 2024Why long model-based rollouts are no reason for bad Q-value estimates16 Jul 2024
- 2023A Tractable Inference Perspective of Offline RL31 Oct 2023
- 2023Multi-timestep models for Model-based Reinforcement Learning9 Oct 2023
- 2018Learning the Reward Function for a Misspecified Modelcited by 4
- 2018Learning the Reward Function for a Misspecified Model29 Jan 2018
- 2021Learning One Representation to Optimize All Rewardscited by 110
- 2022Value Gradient weighted Model-Based Reinforcement Learningcited by 31
- 2022Investigating the Properties of Neural Network Representations in Reinforcement Learningcited by 24
- 2022Goal-Space Planning with Subgoal Modelscited by 12
- 2023Off-Policy RL Algorithms Can be Sample-Efficient for Continuous Control via Sample Multiple Reusecited by 12
- 2021High-dimensional separability for one- and few-shot learningcited by 11
- 2021Temporally Abstract Partial Modelscited by 11
- 2025On Rollouts in Model-Based Reinforcement Learningcited by 10
- 2023A Unified View on Solving Objective Mismatch in Model-Based Reinforcement Learningcited by 9
- 2024A Note on Loss Functions and Error Compounding in Model-based Reinforcement Learningcited by 7
- 2021On-Policy Model Errors in Reinforcement Learningcited by 6
- 2022No More Pesky Hyperparameters: Offline Hyperparameter Tuning for RLcited by 6
12 most cited of 27 All 27 (JSON)
- 2026Dynamics Models for Offline Hyperparameter Selection in Real-World RL11 Aug 2026
- 2026Laplacian Representations for Decision-Time Planning4 Feb 2026
- 2025Planning as Descent: Goal-Conditioned Latent Trajectory Synthesis in Learned Energy Landscapes19 Dec 2025
- 2025Calibrated Value-Aware Model Learning with Probabilistic Environment Models28 May 2025
- 2025Situationally-Aware Dynamics Learning26 May 2025
- 2025On Rollouts in Model-Based Reinforcement Learning28 Jan 2025
- 2024Rejecting Hallucinated State Targets during Planning9 Oct 2024
- 2024Meta-Gradient Search Control: A Method for Improving the Efficiency of Dyna-style Planning27 Jun 2024
- 2024Bounding-Box Inference for Error-Aware Model-Based Reinforcement Learning23 Jun 2024
- 2024Sequence Compression Speeds Up Credit Assignment in Reinforcement Learning6 May 2024
- 2024A Note on Loss Functions and Error Compounding in Model-based Reinforcement Learning15 Apr 2024
- 2024A Multi-step Loss Function for Robust Learning of the Dynamics in Model-based Reinforcement Learning5 Feb 2024
12 newest of 27 All 27 (JSON)