Paper: 2604.19730 Authors: Perry Dong, Alexander Swerdlow, Dorsa Sadigh, Chelsea Finn (Stanford) Categories: cs.LG, cs.RO, cs.AI

Problem

Most performant RL algorithms use test-time scaling methods like sampling multiple action candidates and selecting the best - but these are prohibitively expensive.

Key Insight

Model the denoising of multiple action candidates + selecting the best as a Markov Decision Process (MDP):

  • Goal: progressively filter action candidates before denoising is complete
  • Learn policy and value function in denoising space
  • Predict downstream value of action candidates

FASTER Method

A method for getting benefits of sampling-based test-time scaling without the computational cost:

  1. Early Filtering: Trace performance gain of action samples back to earlier denoising stages
  2. Value Prediction: Learn to predict which candidates will perform best
  3. Lightweight: Can be plugged into existing generative RL algorithms

Results

Across challenging long-horizon manipulation tasks (online and batch-online RL):

SettingResult
Task PerformanceConsistent improvement
Training/Inference ComputeSubstantially reduced
VLA ApplicationSame performance, less compute

Takeaways

  • Sampling-based scaling benefits achievable without expensive sampling
  • MDP formulation enables principled early filtering
  • Plug-and-play design works with diverse RL algorithms

论文: 2604.19730 作者: Perry Dong, Alexander Swerdlow, Dorsa Sadigh, Chelsea Finn(斯坦福) 分类: cs.LG, cs.RO, cs.AI

问题

大多数高性能RL算法使用测试时缩放方法,如采样多个动作候选并选择最佳——但这些方法计算成本极高

关键洞察

将多个动作候选的去噪+选择最佳建模为马尔可夫决策过程(MDP)

  • 目标:在去噪完成前逐步过滤动作候选
  • 在去噪空间中学习策略和价值函数
  • 预测动作候选的下游价值

FASTER方法

一种在无计算成本的情况下获得基于采样的测试时缩放好处的方法:

  1. 早期过滤:将动作样本的性能增益追踪到早期去噪阶段
  2. 价值预测:学习预测哪些候选将表现最佳
  3. 轻量级:可插入现有生成式RL算法

实验结果

跨挑战性长时程操作任务(在线和批量在线RL):

设置结果
任务性能一致提升
训练/推理计算大幅减少
VLA应用相同性能,更少计算

要点总结

  • 采样缩放好处可在无昂贵采样的情况下实现
  • MDP公式化实现有原则的早期过滤
  • 即插即用设计适用于多种RL算法