Safe Offline Reinforcement Learning with Real-Time Budget Constraints
On the map: Offline RL and imitation learning from data → Safe offline RL
Abstract (arXiv)
Aiming at promoting the safe real-world deployment of Reinforcement Learning (RL), research on safe RL has made significant progress in recent years. However, most existing works in the literature still focus on the online setting where risky violations of the safety budget are likely to be incurred during training. Besides, in many real-world applications, the learned policy is required to respond to dynamically determined safety budgets (i.e., constraint threshold) in real time. In this paper, we target at the above real-time budget constraint problem under the offline setting, and propose Trajectory-based REal-time Budget Inference (TREBI) as a novel solution that models this problem from the perspective of trajectory distribution and solves it through diffusion model planning. Theoretically, we prove an error bound of the estimation on the episodic reward and cost under the offline setting and thus provide a performance guarantee for TREBI. Empirical results on a wide range of simulation tasks and a real-world large-scale advertising application demonstrate the capability of TREBI in solving real-time budget constraint problems under offline settings.
Key facts. Cited by 28 papers in the map (data through 14 Oct 2026). Placed in Safe offline RL, one of 13 directions in Offline RL and imitation learning from data. Builds on 40 of its 54 references that are in the map. First posted on arXiv 1 Jun 2023.
Builds on
40 of 54 references.
Closest ancestors · 2
Same direction, newest first
- 2023Constrained Decision Transformer for Offline Safe Reinforcement Learning14 Feb
- 2021Constraints Penalized Q-learning for Safe Offline Reinforcement Learning19 Jul
Other references · 38
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,920
- 2020Conservative Q-Learning for Offline Reinforcement Learningcited by 1,809
- 2016OpenAI Gymcited by 1,736
- 2021Decision Transformer: Reinforcement Learning via Sequence Modelingcited by 1,546
- 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learningcited by 1,427
- 2018Off-Policy Deep Reinforcement Learning without Explorationcited by 1,241
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,061
- 2017Constrained Policy Optimizationcited by 983
- 2021A Minimalist Approach to Offline Reinforcement Learningcited by 899
- 2019Stabilizing Off-Policy Q-Learning via Bootstrapping Error Reductioncited by 840
26 more
- 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
- 2017Sim-to-Real Transfer of Robotic Control with Dynamics Randomizationcited by 672
- 2019Behavior Regularized Offline Reinforcement Learningcited by 662
- 2021Offline Reinforcement Learning as One Big Sequence Modeling Problemcited by 644
- 2020MOPO: Model-based Offline Policy Optimizationcited by 636
- 2020MOReL : Model-Based Offline Reinforcement Learningcited by 590
- 2018Reinforcement Learning and Control as Probabilistic Inference: Tutorial and Reviewcited by 482
- 2018Reward Constrained Policy Optimizationcited by 417
- 2021COMBO: Conservative Offline Model-Based Policy Optimizationcited by 370
- 2015Risk-Constrained Reinforcement Learning with Percentile Risk Criteriacited by 337
- 2020Responsive Safety in Reinforcement Learning by PID Lagrangian Methodscited by 316
- 2019Batch Policy Learning under Constraintscited by 244
- 2019Deep Dynamics Models for Learning Dexterous Manipulationcited by 206
- 2022A Review of Safe Reinforcement Learning: Methods, Theory and Applicationscited by 181
- 2020CRPO: A New Approach for Safe Reinforcement Learning with Convergence Guaranteecited by 134
- 2022RAMBO-RL: Robust Adversarial Model-Based Offline Reinforcement Learningcited by 115
- 2022Mildly Conservative Q-Learning for Offline Reinforcement Learningcited by 112
- 2021Generalized Decision Transformer for Offline Hindsight Information Matchingcited by 106
- 2022Constrained Variational Policy Optimization for Safe Reinforcement Learningcited by 71
- 2022COptiDICE: Offline Constrained Reinforcement Learning via Stationary Distribution Correction Estimationcited by 71
- 2022Saute RL: Almost Surely Safe Reinforcement Learning Using State Augmentationcited by 69
- 2022Constrained Policy Optimization via Bayesian World Modelscited by 57
- 2022On the Robustness of Safe Reinforcement Learning under Observational Perturbationscited by 38
- 2022Trustworthy Reinforcement Learning Against Intrinsic Vulnerabilities: Robustness, Safety, and Generalizabilitycited by 36
- 2022CUP: A Conservative Update Policy Algorithm for Safe Reinforcement Learningcited by 20
Built on by
Citing papers per year: 2023: 1; 2024: 8; 2025: 9; 2026: 10.
28 papers.
Sort:
By area:
- 2024Safe Offline Reinforcement Learning with Feasibility-Guided Diffusion Modelcited by 64
- 2023Constraint-Conditioned Policy Optimization for Versatile Safe Reinforcement Learningcited by 18
- 2024Off-Policy Primal-Dual Safe Reinforcement Learningcited by 14
- 2024Constraint-Adaptive Policy Switching for Offline Safe Reinforcement Learningcited by 13
- 2024OASIS: Conditional Distribution Shaping for Offline Safe Reinforcement Learningcited by 10
- 2025One Subgoal at a Time: Zero-Shot Generalization to Arbitrary Linear Temporal Logic Requirements in Multi-Task Reinforcement Learningcited by 8
- 2024Cooperative Backdoor Attack in Decentralized Reinforcement Learning with Theoretical Guaranteecited by 6
- 2024Offline Goal-Conditioned Reinforcement Learning for Safety-Critical Tasks with Recovery Policycited by 5
- 2024An Offline Adaptation Framework for Constrained Multi-Objective Reinforcement Learningcited by 5
- 2025V-OCBF: Learning Safety Filters from Offline Data via Value-Guided Offline Control Barrier Functionscited by 5
- 2026Safe Flow Q-Learning: Offline Safe Reinforcement Learning with Reachability-Based Flow Policiescited by 4
- 2024FOSP: Fine-tuning Offline Safe Policy through World Modelscited by 3
16 more
- 2025From Uncertain to Safe: Conformal Adaptation of Diffusion Models for Safe PDE Controlcited by 3
- 2025A Provable Approach for End-to-End Safe Reinforcement Learningcited by 2
- 2025Expert-Guided Diffusion Planner for Auto-Biddingcited by 2
- 2026Epigraph-Guided Flow Matching for Safe and Performant Offline Reinforcement Learningcited by 2
- 2025Semi-gradient DICE for Offline Constrained Reinforcement Learningcited by 1
- 2025Online Optimization for Offline Safe Reinforcement Learningcited by 1
- 2025Don't Trade Off Safety: Diffusion Regularization for Constrained Offline RLcited by 0
- 2025Learning Multi-Timescale Interventions under Safety and Resource Constraintscited by 0
- 2026Conditional Sequence Modeling for Safe Reinforcement Learningcited by 0
- 2026Decoupled Guidance Diffusion for Adaptive Offline Safe Reinforcement Learningcited by 0
- 2026Efficient and Uncertainty-Aware Diffusion Framework for Offline-to-Online Reinforcement Learningcited by 0
- 2026Temporal Self-Imitation Learningcited by 0
- 2026Generative Optimization for Incentivized Advertising with Global Level Constraintscited by 0
- 2026Q-learning Penalized Transformer for Safe Offline Reinforcement Learningcited by 0
- 2026A Safe Action Is Not Enough: Feasible-Future Decoding for Vision-Language-Action Policiescited by 0
- 2026Reachability-Aware Diffusion Policy Optimizationcited by 0
- 2026Reachability-Aware Diffusion Policy Optimization5 Oct 2026
- 2026A Safe Action Is Not Enough: Feasible-Future Decoding for Vision-Language-Action Policies4 Oct 2026
- 2026Q-learning Penalized Transformer for Safe Offline Reinforcement Learning28 Sep 2026
- 2026Generative Optimization for Incentivized Advertising with Global Level Constraints5 Aug 2026
- 2026Temporal Self-Imitation Learning18 Jun 2026
- 2026Efficient and Uncertainty-Aware Diffusion Framework for Offline-to-Online Reinforcement Learning29 May 2026
- 2026Decoupled Guidance Diffusion for Adaptive Offline Safe Reinforcement Learning4 May 2026
- 2026Safe Flow Q-Learning: Offline Safe Reinforcement Learning with Reachability-Based Flow Policies16 Mar 2026
- 2026Conditional Sequence Modeling for Safe Reinforcement Learning9 Feb 2026
- 2026Epigraph-Guided Flow Matching for Safe and Performant Offline Reinforcement Learning8 Feb 2026
- 2025V-OCBF: Learning Safety Filters from Offline Data via Value-Guided Offline Control Barrier Functions11 Dec 2025
- 2025Online Optimization for Offline Safe Reinforcement Learning24 Oct 2025
16 more
- 2025Expert-Guided Diffusion Planner for Auto-Bidding12 Aug 2025
- 2025Learning Multi-Timescale Interventions under Safety and Resource Constraints5 Aug 2025
- 2025One Subgoal at a Time: Zero-Shot Generalization to Arbitrary Linear Temporal Logic Requirements in Multi-Task Reinforcement Learning3 Aug 2025
- 2025Semi-gradient DICE for Offline Constrained Reinforcement Learning10 Jun 2025
- 2025A Provable Approach for End-to-End Safe Reinforcement Learning28 May 2025
- 2025Don't Trade Off Safety: Diffusion Regularization for Constrained Offline RL18 Feb 2025
- 2025From Uncertain to Safe: Conformal Adaptation of Diffusion Models for Safe PDE Control4 Feb 2025
- 2024Constraint-Adaptive Policy Switching for Offline Safe Reinforcement Learning25 Dec 2024
- 2024An Offline Adaptation Framework for Constrained Multi-Objective Reinforcement Learning16 Sep 2024
- 2024OASIS: Conditional Distribution Shaping for Offline Safe Reinforcement Learning19 Jul 2024
- 2024FOSP: Fine-tuning Offline Safe Policy through World Models6 Jul 2024
- 2024Cooperative Backdoor Attack in Decentralized Reinforcement Learning with Theoretical Guarantee24 May 2024
- 2024Offline Goal-Conditioned Reinforcement Learning for Safety-Critical Tasks with Recovery Policy4 Mar 2024
- 2024Off-Policy Primal-Dual Safe Reinforcement Learning26 Jan 2024
- 2024Safe Offline Reinforcement Learning with Feasibility-Guided Diffusion Model19 Jan 2024
- 2023Constraint-Conditioned Policy Optimization for Versatile Safe Reinforcement Learning5 Oct 2023
- 2024Safe Offline Reinforcement Learning with Feasibility-Guided Diffusion Modelcited by 64
- 2024Constraint-Adaptive Policy Switching for Offline Safe Reinforcement Learningcited by 13
- 2024OASIS: Conditional Distribution Shaping for Offline Safe Reinforcement Learningcited by 10
- 2024Offline Goal-Conditioned Reinforcement Learning for Safety-Critical Tasks with Recovery Policycited by 5
- 2025V-OCBF: Learning Safety Filters from Offline Data via Value-Guided Offline Control Barrier Functionscited by 5
- 2026Safe Flow Q-Learning: Offline Safe Reinforcement Learning with Reachability-Based Flow Policiescited by 4
- 2025Online Optimization for Offline Safe Reinforcement Learningcited by 1
- 2025Don't Trade Off Safety: Diffusion Regularization for Constrained Offline RLcited by 0
- 2026Decoupled Guidance Diffusion for Adaptive Offline Safe Reinforcement Learningcited by 0
- 2026Temporal Self-Imitation Learningcited by 0
- 2026Q-learning Penalized Transformer for Safe Offline Reinforcement Learningcited by 0
- 2026Q-learning Penalized Transformer for Safe Offline Reinforcement Learning28 Sep 2026
- 2026Temporal Self-Imitation Learning18 Jun 2026
- 2026Decoupled Guidance Diffusion for Adaptive Offline Safe Reinforcement Learning4 May 2026
- 2026Safe Flow Q-Learning: Offline Safe Reinforcement Learning with Reachability-Based Flow Policies16 Mar 2026
- 2025V-OCBF: Learning Safety Filters from Offline Data via Value-Guided Offline Control Barrier Functions11 Dec 2025
- 2025Online Optimization for Offline Safe Reinforcement Learning24 Oct 2025
- 2025Don't Trade Off Safety: Diffusion Regularization for Constrained Offline RL18 Feb 2025
- 2024Constraint-Adaptive Policy Switching for Offline Safe Reinforcement Learning25 Dec 2024
- 2024OASIS: Conditional Distribution Shaping for Offline Safe Reinforcement Learning19 Jul 2024
- 2024Offline Goal-Conditioned Reinforcement Learning for Safety-Critical Tasks with Recovery Policy4 Mar 2024
- 2024Safe Offline Reinforcement Learning with Feasibility-Guided Diffusion Model19 Jan 2024
- 2024Off-Policy Primal-Dual Safe Reinforcement Learningcited by 14
- 2025One Subgoal at a Time: Zero-Shot Generalization to Arbitrary Linear Temporal Logic Requirements in Multi-Task Reinforcement Learningcited by 8
- 2024Cooperative Backdoor Attack in Decentralized Reinforcement Learning with Theoretical Guaranteecited by 6
- 2024An Offline Adaptation Framework for Constrained Multi-Objective Reinforcement Learningcited by 5
- 2025A Provable Approach for End-to-End Safe Reinforcement Learningcited by 2
- 2026Conditional Sequence Modeling for Safe Reinforcement Learningcited by 0
- 2026Conditional Sequence Modeling for Safe Reinforcement Learning9 Feb 2026
- 2025One Subgoal at a Time: Zero-Shot Generalization to Arbitrary Linear Temporal Logic Requirements in Multi-Task Reinforcement Learning3 Aug 2025
- 2025A Provable Approach for End-to-End Safe Reinforcement Learning28 May 2025
- 2024An Offline Adaptation Framework for Constrained Multi-Objective Reinforcement Learning16 Sep 2024
- 2024Cooperative Backdoor Attack in Decentralized Reinforcement Learning with Theoretical Guarantee24 May 2024
- 2024Off-Policy Primal-Dual Safe Reinforcement Learning26 Jan 2024
- 2025From Uncertain to Safe: Conformal Adaptation of Diffusion Models for Safe PDE Controlcited by 3
- 2025Expert-Guided Diffusion Planner for Auto-Biddingcited by 2
- 2026Epigraph-Guided Flow Matching for Safe and Performant Offline Reinforcement Learningcited by 2
- 2026Efficient and Uncertainty-Aware Diffusion Framework for Offline-to-Online Reinforcement Learningcited by 0
- 2026Reachability-Aware Diffusion Policy Optimizationcited by 0
- 2026Reachability-Aware Diffusion Policy Optimization5 Oct 2026
- 2026Efficient and Uncertainty-Aware Diffusion Framework for Offline-to-Online Reinforcement Learning29 May 2026
- 2026Epigraph-Guided Flow Matching for Safe and Performant Offline Reinforcement Learning8 Feb 2026
- 2025Expert-Guided Diffusion Planner for Auto-Bidding12 Aug 2025
- 2025From Uncertain to Safe: Conformal Adaptation of Diffusion Models for Safe PDE Control4 Feb 2025
- 2023Constraint-Conditioned Policy Optimization for Versatile Safe Reinforcement Learningcited by 18
- 2025Semi-gradient DICE for Offline Constrained Reinforcement Learningcited by 1
- 2025Learning Multi-Timescale Interventions under Safety and Resource Constraintscited by 0
- 2026Generative Optimization for Incentivized Advertising with Global Level Constraintscited by 0
- 2026Generative Optimization for Incentivized Advertising with Global Level Constraints5 Aug 2026
- 2025Learning Multi-Timescale Interventions under Safety and Resource Constraints5 Aug 2025
- 2025Semi-gradient DICE for Offline Constrained Reinforcement Learning10 Jun 2025
- 2023Constraint-Conditioned Policy Optimization for Versatile Safe Reinforcement Learning5 Oct 2023
Similar, not cited
Close in wording, no citation either way
- 2024Offline Safe Reinforcement Learning Using Trajectory Classification
- 2023SaFormer: A Conditional Sequence Modeling Approach to Offline Safe Reinforcement Learning
- 2026Beyond Hard Constraints: Budget-Conditioned Reachability For Safe Offline Reinforcement Learning
- 2022Robust Offline Reinforcement Learning with Gradient Penalty and Constraint Relaxation
- 2024Safe Reinforcement Learning for Constrained Markov Decision Processes with Stochastic Stopping Time
- 2024Towards Fast Safe Online Reinforcement Learning via Policy Finetuning
- 2025A Tutorial: An Intuitive Explanation of Offline Reinforcement Learning Theory
- 2023Bayesian Inverse Transition Learning for Offline Settings