Model-Based Offline Planning with Trajectory Pruning
On the map: Offline RL and imitation learning from data → Offline RL algorithms
Abstract (arXiv)
The recent offline reinforcement learning (RL) studies have achieved much progress to make RL usable in real-world systems by learning policies from pre-collected datasets without environment interaction. Unfortunately, existing offline RL methods still face many practical challenges in real-world system control tasks, such as computational restriction during agent training and the requirement of extra control flexibility. The model-based planning framework provides an attractive alternative. However, most model-based planning algorithms are not designed for offline settings. Simply combining the ingredients of offline RL with existing methods either provides over-restrictive planning or leads to inferior performance. We propose a new light-weighted model-based offline planning framework, namely MOPP, which tackles the dilemma between the restrictions of offline learning and high-performance planning. MOPP encourages more aggressive trajectory rollout guided by the behavior policy learned from data, and prunes out problematic trajectories to avoid potential out-of-distribution samples. Experimental results show that MOPP provides competitive performance compared with existing model-based offline planning and RL approaches.
Key facts. Cited by 30 papers in the map (data through 14 Oct 2026). Placed in Offline RL algorithms, one of 13 directions in Offline RL and imitation learning from data. Builds on 24 of its 35 references that are in the map. First posted on arXiv 16 May 2021.
Builds on
24 of 35 references.
Closest ancestors · 7
Same direction, newest first
- 2021Offline Reinforcement Learning with Fisher Divergence Critic Regularization14 Mar
- 2020EMaQ: Expected-Max Q-Learning Operator for Simple Yet Effective Offline and Online RL21 Jul
- 2020Conservative Q-Learning for Offline Reinforcement Learning8 Jun
- 2020MOPO: Model-based Offline Policy Optimization27 May
- 2020MOReL : Model-Based Offline Reinforcement Learning12 May
- 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learning15 Apr
- 2019Behavior Regularized Offline Reinforcement Learning26 Nov
Other references · 17
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2013Playing Atari with Deep Reinforcement Learningcited by 3,274
- 2018Off-Policy Deep Reinforcement Learning without Explorationcited by 1,241
- 2015End-to-End Training of Deep Visuomotor Policiescited by 875
- 2019Stabilizing Off-Policy Q-Learning via Bootstrapping Error Reductioncited by 840
- 2018Deep Reinforcement Learning in a Handful of Trials using Probabilistic Dynamics Modelscited by 774
- 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 773
- 2017Learning Complex Dexterous Manipulation with Deep Reinforcement Learning and Demonstrationscited by 715
- 2015MADE: Masked Autoencoder for Distribution Estimationcited by 307
- 2019Way Off-Policy Batch Deep Reinforcement Learning of Implicit Human Preferences in Dialogcited by 284
- 2019Batch Policy Learning under Constraintscited by 244
- 2019Deep Dynamics Models for Learning Dexterous Manipulationcited by 206
5 more
- 2020Model-Based Offline Planningcited by 129
- 2018Plan Online, Learn Offline: Efficient Learning and Exploration via Model-Based Controlcited by 113
- 2020The Importance of Pessimism in Fixed-Dataset Policy Optimizationcited by 99
- 2019Exploring Model-based Planning with Policy Networkscited by 90
- 2021DeepThermal: Combustion Optimization for Thermal Power Generating Units Using Offline Reinforcement Learningcited by 48
Built on by
Citing papers per year: 2021: 2; 2022: 7; 2023: 9; 2024: 6; 2025: 5; 2026: 1.
30 papers.
Sort:
By area:
- 2021Constraints Penalized Q-learning for Safe Offline Reinforcement Learningcited by 91
- 2022When to Trust Your Simulator: Dynamics-Aware Hybrid Offline-and-Online Reinforcement Learningcited by 60
- 2022A Policy-Guided Imitation Approach for Offline Reinforcement Learningcited by 59
- 2023Offline Multi-Agent Reinforcement Learning with Implicit Global-to-Local Value Regularizationcited by 43
- 2021UMBRELLA: Uncertainty-Aware Model-Based Offline Reinforcement Learning Leveraging Planningcited by 23
- 2023Waypoint Transformer: Reinforcement Learning via Supervised Learning with Intermediate Targetscited by 22
- 2022Double Check Your State Before Trusting It: Confidence-Aware Bidirectional Offline Model-Based Imaginationcited by 21
- 2023GOPlan: Goal-conditioned Offline Reinforcement Learning by Planning with Learned Modelscited by 20
- 2023Look Beneath the Surface: Exploiting Fundamental Symmetry for Sample-Efficient Offline RLcited by 18
- 2022Discriminator-Guided Model-Based Offline Imitation Learningcited by 17
- 2022Model-based Trajectory Stitching for Improved Offline Reinforcement Learningcited by 15
- 2023Uncertainty-driven Trajectory Truncation for Data Augmentation in Offline Reinforcement Learningcited by 13
18 more
- 2024Bayes Adaptive Monte Carlo Tree Search for Offline Model-based Reinforcement Learningcited by 11
- 2025Closing the Train-Test Gap in World Models for Gradient-Based Planningcited by 11
- 2022Model-based trajectory stitching for improved behavioural cloning and its applicationscited by 9
- 2023Offline Multi-Agent Reinforcement Learning with Coupled Value Factorizationcited by 7
- 2024Zero-shot Model-based Reinforcement Learning using Large Language Modelscited by 6
- 2023Model-based Offline Reinforcement Learning with Local Misspecificationcited by 5
- 2023Get Back Here: Robust Imitation by Return-to-Distribution Planningcited by 4
- 2025Data Center Cooling System Optimization Using Offline Reinforcement Learningcited by 4
- 2025Reflect-then-Plan: Offline Model-Based Planning through a Doubly Bayesian Lenscited by 4
- 2022Behavior Estimation from Multi-Source Data for Offline Reinforcement Learningcited by 2
- 2023Multi-timestep models for Model-based Reinforcement Learningcited by 1
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learningcited by 1
- 2024Offline Trajectory Optimization for Offline Reinforcement Learningcited by 1
- 2024State-Constrained Offline Reinforcement Learningcited by 1
- 2025Long-Horizon Model-Based Offline Reinforcement Learning Without Explicit Conservatismcited by 1
- 2026Receding-Horizon Control via Drifting Modelscited by 1
- 2024M3PC: Test-time Model Predictive Control for Pretrained Masked Trajectory Modelcited by 0
- 2025Model-Based Offline Reinforcement Learning with Reliability-Guaranteed Sequence Modelingcited by 0
- 2026Receding-Horizon Control via Drifting Models6 Apr 2026
- 2025Closing the Train-Test Gap in World Models for Gradient-Based Planning10 Dec 2025
- 2025Long-Horizon Model-Based Offline Reinforcement Learning Without Explicit Conservatism4 Dec 2025
- 2025Reflect-then-Plan: Offline Model-Based Planning through a Doubly Bayesian Lens6 Jun 2025
- 2025Model-Based Offline Reinforcement Learning with Reliability-Guaranteed Sequence Modeling10 Feb 2025
- 2025Data Center Cooling System Optimization Using Offline Reinforcement Learning25 Jan 2025
- 2024M3PC: Test-time Model Predictive Control for Pretrained Masked Trajectory Model7 Dec 2024
- 2024Zero-shot Model-based Reinforcement Learning using Large Language Models15 Oct 2024
- 2024Bayes Adaptive Monte Carlo Tree Search for Offline Model-based Reinforcement Learning15 Oct 2024
- 2024State-Constrained Offline Reinforcement Learning23 May 2024
- 2024Offline Trajectory Optimization for Offline Reinforcement Learning16 Apr 2024
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learning5 Feb 2024
18 more
- 2023GOPlan: Goal-conditioned Offline Reinforcement Learning by Planning with Learned Models30 Oct 2023
- 2023Multi-timestep models for Model-based Reinforcement Learning9 Oct 2023
- 2023Offline Multi-Agent Reinforcement Learning with Implicit Global-to-Local Value Regularization21 Jul 2023
- 2023Waypoint Transformer: Reinforcement Learning via Supervised Learning with Intermediate Targets24 Jun 2023
- 2023Offline Multi-Agent Reinforcement Learning with Coupled Value Factorization15 Jun 2023
- 2023Look Beneath the Surface: Exploiting Fundamental Symmetry for Sample-Efficient Offline RL7 Jun 2023
- 2023Get Back Here: Robust Imitation by Return-to-Distribution Planning2 May 2023
- 2023Uncertainty-driven Trajectory Truncation for Data Augmentation in Offline Reinforcement Learning10 Apr 2023
- 2023Model-based Offline Reinforcement Learning with Local Misspecification26 Jan 2023
- 2022Model-based trajectory stitching for improved behavioural cloning and its applications8 Dec 2022
- 2022Behavior Estimation from Multi-Source Data for Offline Reinforcement Learning29 Nov 2022
- 2022Model-based Trajectory Stitching for Improved Offline Reinforcement Learning21 Nov 2022
- 2022A Policy-Guided Imitation Approach for Offline Reinforcement Learning15 Oct 2022
- 2022Discriminator-Guided Model-Based Offline Imitation Learning1 Jul 2022
- 2022When to Trust Your Simulator: Dynamics-Aware Hybrid Offline-and-Online Reinforcement Learning27 Jun 2022
- 2022Double Check Your State Before Trusting It: Confidence-Aware Bidirectional Offline Model-Based Imagination16 Jun 2022
- 2021UMBRELLA: Uncertainty-Aware Model-Based Offline Reinforcement Learning Leveraging Planning22 Nov 2021
- 2021Constraints Penalized Q-learning for Safe Offline Reinforcement Learning19 Jul 2021
- 2021Constraints Penalized Q-learning for Safe Offline Reinforcement Learningcited by 91
- 2022When to Trust Your Simulator: Dynamics-Aware Hybrid Offline-and-Online Reinforcement Learningcited by 60
- 2022A Policy-Guided Imitation Approach for Offline Reinforcement Learningcited by 59
- 2023Offline Multi-Agent Reinforcement Learning with Implicit Global-to-Local Value Regularizationcited by 43
- 2021UMBRELLA: Uncertainty-Aware Model-Based Offline Reinforcement Learning Leveraging Planningcited by 23
- 2023Waypoint Transformer: Reinforcement Learning via Supervised Learning with Intermediate Targetscited by 22
- 2022Double Check Your State Before Trusting It: Confidence-Aware Bidirectional Offline Model-Based Imaginationcited by 21
- 2023GOPlan: Goal-conditioned Offline Reinforcement Learning by Planning with Learned Modelscited by 20
- 2023Look Beneath the Surface: Exploiting Fundamental Symmetry for Sample-Efficient Offline RLcited by 18
- 2022Discriminator-Guided Model-Based Offline Imitation Learningcited by 17
- 2022Model-based Trajectory Stitching for Improved Offline Reinforcement Learningcited by 15
- 2023Uncertainty-driven Trajectory Truncation for Data Augmentation in Offline Reinforcement Learningcited by 13
12 most cited of 27 All 27 (JSON)
- 2025Long-Horizon Model-Based Offline Reinforcement Learning Without Explicit Conservatism4 Dec 2025
- 2025Reflect-then-Plan: Offline Model-Based Planning through a Doubly Bayesian Lens6 Jun 2025
- 2025Model-Based Offline Reinforcement Learning with Reliability-Guaranteed Sequence Modeling10 Feb 2025
- 2025Data Center Cooling System Optimization Using Offline Reinforcement Learning25 Jan 2025
- 2024M3PC: Test-time Model Predictive Control for Pretrained Masked Trajectory Model7 Dec 2024
- 2024Bayes Adaptive Monte Carlo Tree Search for Offline Model-based Reinforcement Learning15 Oct 2024
- 2024State-Constrained Offline Reinforcement Learning23 May 2024
- 2024Offline Trajectory Optimization for Offline Reinforcement Learning16 Apr 2024
- 2024Deep autoregressive density nets vs neural ensembles for model-based offline reinforcement learning5 Feb 2024
- 2023GOPlan: Goal-conditioned Offline Reinforcement Learning by Planning with Learned Models30 Oct 2023
- 2023Multi-timestep models for Model-based Reinforcement Learning9 Oct 2023
- 2023Offline Multi-Agent Reinforcement Learning with Implicit Global-to-Local Value Regularization21 Jul 2023
12 newest of 27 All 27 (JSON)
- 2026Receding-Horizon Control via Drifting Models6 Apr 2026
- 2024Zero-shot Model-based Reinforcement Learning using Large Language Models15 Oct 2024
Similar, not cited
Close in wording, no citation either way
- 2023A Survey on Offline Model-Based Reinforcement Learning
- 2025Learning from Reward-Free Offline Data: A Case for Planning with Latent Dynamics Models
- 2025A Tutorial: An Intuitive Explanation of Offline Reinforcement Learning Theory
- 2021Revisiting Design Choices in Offline Model-Based Reinforcement Learning
- 2021A Workflow for Offline Model-Free Robotic Reinforcement Learning
- 2022Hierarchical Planning Through Goal-Conditioned Offline Reinforcement Learning
- 2026Shortcut Trajectory Planning for Efficient Offline Reinforcement Learning
- 2022Comparing Model-free and Model-based Algorithms for Offline Reinforcement Learning