arXiv:2602.05051cs.LGcs.AI2026-02中稿 · the fourteenth Int…被引 5

用反射流模型在离线强化学习中同时保证策略表达力与分布外安全。

ReFORM: Reflected Flows for On-support Offline RL via Noise Manipulation

  • 通过构造受约束的支持集,让流模型自然避免分布外行为。
  • 在OGBench 40个任务上超越所有调参基线,性能领先显著。
  • 适合追求高安全性和强泛化能力的离线强化学习研究者。

离线强化学习旨在仅从行为策略生成的固定数据集中学习最优策略,无需额外环境交互。其主要挑战之一是分布外(OOD)误差,即策略偏离训练分布时出现的问题。已有方法通过惩罚统计距离来限制策略接近行为策略,但这会抑制策略改进且无法完全防止分布外动作。另一个挑战是最优策略分布可能为多模态,难以建模。近期工作采用扩散或流模型以提升表达能力,但尚未解决如何在保持表达力的同时避免分布外误差。本文提出ReFORM,一种基于流模型的离线强化学习方法,通过构造性地施加更宽松的支持约束实现此目标。ReFORM首先学习一个行为克隆(BC)流策略,其源分布受限以捕捉动作分布的支持集;随后优化一个反射流,生成受限噪声作用于该BC流,从而在保持支持集的前提下最大化性能。在包含40个挑战性任务的OGBench基准上,使用统一超参数设置,ReFORM在性能曲线表现上全面优于所有经过人工调优的基线方法。

原文摘要 · Abstract (English)

Offline reinforcement learning (RL) aims to learn the optimal policy from a fixed dataset generated by behavior policies without additional environment interactions. One common challenge that arises in this setting is the out-of-distribution (OOD) error, which occurs when the policy leaves the training distribution. Prior methods penalize a statistical distance term to keep the policy close to the behavior policy, but this constrains policy improvement and may not completely prevent OOD actions. Another challenge is that the optimal policy distribution can be multimodal and difficult to represent. Recent works apply diffusion or flow policies to address this problem, but it is unclear how to avoid OOD errors while retaining policy expressiveness. We propose ReFORM, an offline RL method based on flow policies that enforces the less restrictive support constraint by construction. ReFORM learns a behavior cloning (BC) flow policy with a bounded source distribution to capture the support of the action distribution, then optimizes a reflected flow that generates bounded noise for the BC flow while keeping the support, to maximize the performance. Across 40 challenging tasks from the OGBench benchmark with datasets of varying quality and using a constant set of hyperparameters for all tasks, ReFORM dominates all baselines with hand-tuned hyperparameters on the performance profile curves.

离线RL流模型策略安全强化学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。