EpiFlow:通过上图引导流匹配在离线强化学习中兼顾安全与性能
EpiFlow: Bridging Safety and Performance in Offline Reinforcement Learning via Epigraph-Guided Flow Matching
AI 导读
EpiFlow 将安全离线强化学习建模为带硬状态约束的最优控制问题,用上图可行性值函数配合流匹配从固定数据集中提取策略,被 NeurIPS 2026 接收。在船只导航、Safety-Gymnasium 和 Bullet-Safety-Gym 等安全关键基准上,该方法取得有竞争力的回报,同时约束违规接近于零,在安全性与性能之间取得良好平衡。
正文 · 原文
Abstract:Offline reinforcement learning (RL) offers a promising framework for deploying autonomous systems in safety-critical settings without the risks of online exploration. However, learning policies that simultaneously achieve high performance and strong safety guarantees from fixed datasets remains a fundamental challenge. Many existing safe offline RL approaches typically rely on soft constraint formulations, which may permit safety violations and are sensitive to distributional shift. In contrast, formal methods such as Hamilton--Jacobi (HJ) reachability and Control Barrier Functions (CBFs) provide rigorous safety guarantees, but often yield overly conservative solutions, often neglecting performance. In this work, we bridge this gap by formulating safe offline RL as a state-constrained optimal control problem, where safety is enforced through hard state constraints and performance is captured via a reward function. The resulting value function satisfies a Hamilton--Jacobi--Bellman (HJB) equation, which we approximate using offline RL on fixed datasets by learning an epigraph feasibility value function and extracting policies via flow matching. This formulation enables principled integration of safety guarantees with data-driven policy optimization. Empirically, across safety-critical benchmarks including boat navigation, Safety-Gymnasium, and Bullet-Safety-Gym tasks, our approach achieves competitive returns while exhibiting near-zero constraint violations, demonstrating a favorable balance between safety and performance in the offline setting.
| Comments: | 27 pages, 10 figures, first two authors have contributed equally. Paper accepted at NeurIPS 2026 |
| Subjects: | Machine Learning (cs.LG); Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2602.08054 [cs.LG] |
| (or arXiv:2602.08054v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2602.08054 arXiv-issued DOI via DataCite |
Submission history
From: Manan Tayal [view email]
[v1]
Sun, 8 Feb 2026 16:56:21 UTC (3,086 KB)
[v2]
Mon, 26 Oct 2026 11:57:00 UTC (3,374 KB)
来源:arXiv cs.AI · arxiv.org