arXiv:2603.22430cs.LG2026-03被引 1

让离线强化学习在推理时动态优化策略,提升性能。

Inference Time Policy Optimization for Offline RL with Differentiable World Models

  • 用可微世界模型在推理时通过想象轨迹反向传播优化策略参数。
  • 在D4RL连续控制任务上优于主流离线RL基线,性能持续提升。
  • 提出轻量采样器,在降低计算开销的同时保留大部分收益,适合实际部署。

离线强化学习从固定数据集学习最优策略,训练一次后在推理时直接部署,不再调整。受模型预测控制(MPC)启发,本文提出一种推理时适应框架,结合预训练策略与学习到的世界模型。现有世界模型和扩散规划方法仅在训练中生成想象轨迹或在推理时采样候选计划,但未利用推理时信息动态优化策略参数。本文设计的可微世界模型(DWM)管道支持通过想象轨迹进行端到端梯度计算,实现推理时策略优化(ITPO)。在D4RL连续控制基准(MuJoCo运动任务和AntMaze)上验证,利用推理时信息优化策略参数能稳定超越强基线。然而,推理时适应计算成本高,轨迹生成与反向传播占主导。本文显式分析该权衡,发现一种改进的一步均值流采样器(tilted one-step MeanFlow)能在极小计算代价下恢复大部分性能增益。

原文摘要 · Abstract (English)

Offline Reinforcement Learning (RL) learns optimal policies from fixed datasets, training a policy once and deploying it at inference time without further refinement. Inspired by model predictive control (MPC), we introduce an inference time adaptation framework that utilizes a pretrained policy along with a learned world model. While existing world model and diffusion-planning methods use learned dynamics to generate imagined trajectories during training, or to sample candidate plans at inference time, they do not use inference-time information to *optimize* the policy parameters on the fly. In contrast, our design is a Differentiable World Model (DWM) pipeline that enables end-to-end gradient computation through imagined rollouts for inference time policy optimization (ITPO). We evaluate our algorithm on D4RL continuous-control benchmarks (MuJoCo locomotion tasks and AntMaze), and show that exploiting inference-time information to optimize the policy parameters yields consistent gains over strong offline RL baselines. Inference-time adaptation, however, is expensive: rollout generation and backpropagation dominate per-step compute. We study this tradeoff explicitly, showing that a suitable tilted version of one-step MeanFlow sampler recovers much of the gains at a fraction of the cost.

离线RL策略优化可微模型推理加速

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