Paper Overview
Field: Machine Learning Authors: Perry Dong, Alexander Swerdlow, Dorsa Sadigh, Chelsea Finn Published: 2026-04-21 arXiv: 2604.19730
Abstract
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, 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 in the denoising process.
Key Idea
The core insight is that denoising 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, FASTER learns a policy and value function in the denoising space that predicts the downstream value of action candidates and filters them to maximize return.
Highlights
- Lightweight, plug-and-play method that works with existing generative RL algorithms.
- On challenging long-horizon manipulation tasks in online and batch online RL, FASTER consistently improves the underlying policy and achieves the best overall performance among compared baselines.
- Applied to a pretrained vision-language-action (VLA) model, FASTER achieves comparable performance while significantly reducing training and inference compute requirements.
- Paper: https://arxiv.org/abs/2604.19730
- Code: https://github.com/alexanderswerdlow/faster