提出单步生成动作的强化学习方法,训练推理更快更稳。
One-Step Flow Q-Learning: Addressing the Diffusion Policy Bottleneck in Offline Reinforcement Learning
- 将扩散策略重构为流匹配框架,直接学习平均速度场实现单步动作生成。
- 在D4RL上单步生成仍超越多步扩散强化学习,性能领先。
- 无需额外模块或蒸馏,适合追求高效稳定的离线强化学习应用。
扩散Q学习(DQL)已将扩散策略确立为离线强化学习中的高性能范式,但其依赖多步去噪生成动作,导致训练和推理速度慢且不稳定。现有加速方法通常依赖辅助模块或策略蒸馏,牺牲了简洁性或性能。目前尚不清楚能否在不作权衡的情况下直接训练单步策略。为此,我们提出单步流Q学习(OFQL),一种新框架,可在无需辅助模块或蒸馏的前提下,实现训练与推理阶段的高效单步动作生成。OFQL在流匹配(FM)框架内重构DQL策略,但不同于传统FM,通过学习一个平均速度场,直接支持精确的单步动作生成。该设计避免了多步去噪与时间反向传播更新,显著提升训练速度与鲁棒性。在D4RL基准上的大量实验表明,尽管仅用单步生成,OFQL不仅大幅降低训练与推理计算量,还远超多步DQL,且优于所有其他基线,在D4RL上达到最先进性能。
原文摘要 · Abstract (English)
Diffusion Q-Learning (DQL) has established diffusion policies as a high-performing paradigm for offline reinforcement learning, but its reliance on multi-step denoising for action generation renders both training and inference slow and fragile. Existing efforts to accelerate DQL toward one-step denoising typically rely on auxiliary modules or policy distillation, sacrificing either simplicity or performance. It remains unclear whether a one-step policy can be trained directly without such trade-offs. To this end, we introduce One-Step Flow Q-Learning (OFQL), a novel framework that enables effective one-step action generation during both training and inference, without auxiliary modules or distillation. OFQL reformulates the DQL policy within the Flow Matching (FM) paradigm but departs from conventional FM by learning an average velocity field that directly supports accurate one-step action generation. This design removes the need for multi-step denoising and backpropagation-through-time updates, resulting in substantially faster and more robust learning. Extensive experiments on the D4RL benchmark show that OFQL, despite generating actions in a single step, not only significantly reduces computation during both training and inference but also outperforms multi-step DQL by a large margin. Furthermore, OFQL surpasses all other baselines, achieving state-of-the-art performance in D4RL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。