arXiv:2505.20781cs.ROcs.LG2025-05NeurIPS被引 3

用扩散模型生成目标策略轨迹,提升高维长序列离线评估精度

STITCH-OPE: Trajectory Stitching with Guided Diffusion for Off-Policy Evaluation

  • 基于行为数据预训练扩散模型,用目标策略得分函数引导生成新轨迹
  • 在D4RL和Gym上误差、相关性和后悔值均显著优于现有方法
  • 通过轨迹拼接与行为策略去正则化,有效降低长序列评估方差

离线策略评估(OPE)利用行为策略收集的离线数据估计目标策略性能,在机器人或医疗等高成本或高风险领域至关重要。现有方法在高维、长时序问题上表现不佳,源于重要性权重带来的方差指数增长或动态模型误差累积。为此,我们提出STITCH-OPE,一种基于生成模型的模型式框架,利用去噪扩散模型实现高维状态与动作空间中的长时序OPE。该方法以行为数据预训练的扩散模型为基础,通过目标策略的得分函数引导去噪过程生成合成轨迹。STITCH-OPE提出两项关键技术:(1)在引导过程中减去行为策略得分以防止过正则化;(2)通过端到端拼接部分轨迹生成长序列。理论分析表明,在温和假设下,该方法相比传统长序列扩散可实现方差指数级下降。在D4RL与OpenAI Gym基准测试中,其在均方误差、相关性与后悔值指标上均显著优于当前最优OPE方法。

原文摘要 · Abstract (English)

Off-policy evaluation (OPE) estimates the performance of a target policy using offline data collected from a behavior policy, and is crucial in domains such as robotics or healthcare where direct interaction with the environment is costly or unsafe. Existing OPE methods are ineffective for high-dimensional, long-horizon problems, due to exponential blow-ups in variance from importance weighting or compounding errors from learned dynamics models. To address these challenges, we propose STITCH-OPE, a model-based generative framework that leverages denoising diffusion for long-horizon OPE in high-dimensional state and action spaces. Starting with a diffusion model pre-trained on the behavior data, STITCH-OPE generates synthetic trajectories from the target policy by guiding the denoising process using the score function of the target policy. STITCH-OPE proposes two technical innovations that make it advantageous for OPE: (1) prevents over-regularization by subtracting the score of the behavior policy during guidance, and (2) generates long-horizon trajectories by stitching partial trajectories together end-to-end. We provide a theoretical guarantee that under mild assumptions, these modifications result in an exponential reduction in variance versus long-horizon trajectory diffusion. Experiments on the D4RL and OpenAI Gym benchmarks show substantial improvement in mean squared error, correlation, and regret metrics compared to state-of-the-art OPE methods.

离线评估扩散模型轨迹生成强化学习

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