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:
- Early Filtering: Trace performance gain of action samples back to earlier denoising stages
- Value Prediction: Learn to predict which candidates will perform best
- Lightweight: Can be plugged into existing generative RL algorithms
Results
Across challenging long-horizon manipulation tasks (online and batch-online RL):
| Setting | Result |
|---|---|
| Task Performance | Consistent improvement |
| Training/Inference Compute | Substantially reduced |
| VLA Application | Same 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方法
一种在无计算成本的情况下获得基于采样的测试时缩放好处的方法:
- 早期过滤:将动作样本的性能增益追踪到早期去噪阶段
- 价值预测:学习预测哪些候选将表现最佳
- 轻量级:可插入现有生成式RL算法
实验结果
跨挑战性长时程操作任务(在线和批量在线RL):
| 设置 | 结果 |
|---|---|
| 任务性能 | 一致提升 |
| 训练/推理计算 | 大幅减少 |
| VLA应用 | 相同性能,更少计算 |
要点总结
- 采样缩放好处可在无昂贵采样的情况下实现
- MDP公式化实现有原则的早期过滤
- 即插即用设计适用于多种RL算法