用统计检验方法加速强化学习中的动作采样,提升效率与适应性。
Moment Matching Q-Learning

- 基于最大均值差异(MMD)匹配分布各阶统计量,实现稳定采样。
- 在D4RL多个任务中达到接近或优于基线的性能,计算效率更高。
- 特别适合需要快速在线微调的离线到在线强化学习场景。
基于得分和流的生成模型在捕捉复杂分布方面表现出强大能力,已广泛应用于从图像生成到强化学习的任务中。然而,这些模型存在推理延迟长的问题,在需要迭代采样的强化学习中构成显著计算瓶颈。为此,我们提出一种名为矩匹配Q学习(MoMa QL)的新框架,利用统计假设检验中的最大均值差异(MMD)技术,旨在匹配原始分布与目标分布的所有阶统计量。通过强正则化所有阶矩,该算法保证了条件得分函数的分布级收敛,并在多种超参数下保持稳定。实验证明,我们的方法在多个D4RL任务中具有更高的计算效率,性能相当甚至更优。尤为突出的是,通过加速基于流的策略的动作采样过程,MoMa QL在离线到在线强化学习任务中展现出更优性能,得益于更快更强的在线交互微调适应能力。
原文摘要 · Abstract (English)
Score-based and flow-based generative models exhibit remarkable expressive capacity in capturing complex distributions, and have been extensively deployed in tasks ranging from image generation to reinforcement learning. Nevertheless, these models suffer from prolonged inference latency, which imposes a significant computational bottleneck in RL with iterative sampling. To overcome this limitation, we propose a new framework named Moment Matching Q-Learning (MoMa QL), which utilizes a technique from statistical hypothesis testing known as maximum mean discrepancy (MMD) that intend to match all orders of statistics between the original and target distribution. By enforcing strong regularization on all moment statistics, this algorithm guarantees distribution-level convergence for conditional score function and remains stable under various hyperparameters. Empirically, we show that our method MoMa QL is more computationally efficient with a comparable if not competitive performance in various D4RL tasks. Remarkably, by accelerating the action sampling process for flow-based policies, MoMa QL demonstrates superior performance in offline-to-online RL tasks because of faster and stronger adaptability for online interactive finetuning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。