arXiv:2604.19730cs.LGcs.AI2026-04被引 3

用价值引导采样加速扩散策略强化学习,不增加计算开销。

FASTER: Value-Guided Sampling for Fast RL

论文配图:FASTER: Value-Guided Sampling for Fast RL
图 1 · 摘自论文原文
  • 将动作采样与去噪过程建模为马尔可夫决策过程,提前筛选候选动作。
  • 在长时序操作任务中提升性能,优于对比方法且推理更高效。
  • 可无缝接入现有生成式强化学习框架,适合算力受限场景。

当前表现优异的强化学习算法常因测试时扩展(test-time scaling)方法而计算成本高昂,例如多次采样动作候选并选择最优者。本文提出FASTER,一种无需额外计算代价即可获得基于扩散模型策略的采样优势的方法。其核心思想是将多个动作候选的去噪过程与选择机制建模为马尔可夫决策过程(MDP),目标是在去噪完成前逐步过滤候选动作。通过在去噪空间中学习策略与价值函数,预测动作候选的下游价值,实现高效筛选以最大化回报。该方法轻量且可嵌入现有生成式强化学习算法。在在线与批在线强化学习的复杂长时序操控任务中,FASTER持续提升基线策略性能,并取得最佳综合表现。应用于预训练视觉-语言-动作模型(VLA)时,达到相当性能的同时显著降低训练与推理计算需求。代码已开源。

原文摘要 · Abstract (English)

Some of the most performant reinforcement learning algorithms today can be prohibitively expensive as they use test-time scaling methods such as sampling multiple action candidates and selecting the best one. In this work, we propose FASTER, a method for getting the benefits of sampling-based test-time scaling of diffusion-based policies without the computational cost by tracing the performance gain of action samples back to earlier in the denoising process. Our key insight is that we can model the denoising of multiple action candidates and selecting the best one as a Markov Decision Process (MDP) where the goal is to progressively filter action candidates before denoising is complete. With this MDP, we can learn a policy and value function in the denoising space that predicts the downstream value of action candidates in the denoising process and filters them while maximizing returns. The result is a method that is lightweight and can be plugged into existing generative RL algorithms. Across challenging long-horizon manipulation tasks in online and batch-online RL, FASTER consistently improves the underlying policies and achieves the best overall performance among the compared methods. Applied to a pretrained VLA, FASTER achieves the same performance while substantially reducing training and inference compute requirements. Code is available at https://github.com/alexanderswerdlow/faster .

强化学习扩散模型采样优化

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