用单步生成替代多步去噪,提升离线强化学习效率与稳定性
Fast and Highly Expressive Policy Learning for Offline Reinforcement Learning via Bootstrapped Flow Q-Learning

- 通过分治思想学习短程位移,直接构建噪声到动作的映射
- 在D4RL上实现更高性能,计算成本降低显著
- 无需额外网络或蒸馏,方法简洁且鲁棒性强
基于扩散的Q-learning已成为离线强化学习的强大范式,但其依赖多步去噪导致训练和推理计算开销大且脆弱。现有加速方法常引入辅助网络、策略蒸馏或多阶段训练,往往牺牲简单性、稳定性或性能。为此,我们提出自举流Q学习(BFQ),一种新颖框架,可在训练和推理中实现准确的单步动作生成,无需辅助网络或蒸馏。BFQ将流路径上的位移向量视为分治问题:先学习可由流匹配边际速度准确估计的短程位移,并以此为基础自举出单步噪声到动作的映射。该设计消除了多步去噪,使学习过程更快速、简单、鲁棒。大量D4RL实验表明,相较于多步扩散基线,BFQ在显著降低计算成本的同时提升了性能,证明单步动作生成足以实现高性能离线强化学习。
原文摘要 · Abstract (English)
Diffusion-based Q-learning has emerged as a powerful paradigm for offline reinforcement learning, but its reliance on multi-step denoising makes both training and inference computationally expensive and brittle. Recent efforts to accelerate diffusion Q-learning toward single-step action generation typically introduce auxiliary networks, policy distillation, or multi-phase training, which frequently compromise simplicity, stability, or performance. To address these limitations, we introduce Bootstrapped Flow Q-Learning (BFQ), a novel framework that enables accurate single-step action generation during both training and inference, without auxiliary networks or distillation procedures. BFQ adopts a divide-and-conquer view of the displacement vector along the flow path: it begins by learning short-range displacements that can be accurately estimated from the Flow Matching marginal velocity, and bootstraps these components to directly learn a noise-to-action mapping in a single step. This formulation eliminates multi-step denoising, resulting in a learning procedure that is substantially faster, simpler, and more robust. Extensive D4RL evaluations show that BFQ improves performance while significantly reducing computational cost compared to multi-step diffusion baselines, demonstrating that single-step action generation suffices for high-performance offline Reinforcement Learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。