Paper Overview
Research Area: Machine Learning Authors: Perry Dong, Alexander Swerdlow, Dorsa Sadigh, Chelsea Finn Released: 2026-04-21 arXiv: 2604.19730
Summary
Some of the most performant reinforcement learning algorithms today can be prohibitively expensive because they use test-time scaling methods such as sampling multiple action candidates and selecting the best one. In this work, the authors 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 stages in the denoising process.
Key Insight
The denoising of multiple action candidates and selecting the best one can be modeled as a Markov Decision Process (MDP), where the goal is to progressively filter action candidates before denoising is complete. With this MDP formulation, a policy and value function can be learned in the denoising space that predicts the downstream value of candidates and filters them to maximize return.
Results
- The result is a lightweight method that works plug-and-play with existing generative RL algorithms.
- On challenging long-horizon manipulation tasks in both online and batch online RL settings, FASTER consistently improves the underlying policy and achieves the best overall performance among compared methods.
- Applied to a pretrained vision-language-action (VLA) model, FASTER matches performance while substantially reducing both training and inference compute requirements.
- Paper: arXiv:2604.19730
- Code: https://github.com/alexanderswerdlow/faster