Sample-Efficient Reinforcement Learning for Policy Improvement via Flow-Reversed Action Anchors
Organizations: Amazon · UT Austin · UC San Diego · Brown University ·
Abstract
Generalist robot policies offer strong behavioral priors, but achieving deployment-level reliability and throughput remains difficult. Reinforcement learning (RL) can improve these policies, yet its effectiveness is constrained by the cost of online interaction. Existing methods adapt frozen flow-matching policies by treating noise space as a continuous action space. We show that this behaviorally unstructured space makes exploration inefficient, often preventing these methods from extracting meaningful learning signals on challenging tasks. We introduce AnchorQ, which enables coarse-to-fine policy improvement through a structured, refinable latent action representation. At each decision step, AnchorQ uses flow reversal to map reference behaviors to state-conditioned latent anchors, samples local particles around those anchors, and applies value-guided sampling to explore promising refinements. Our analysis shows that anchors reduce identification complexity at the cost of support bias; particle refinement trades this bias against action-class complexity; and critic accuracy determines the reliability of KL-regularized, value-guided selection. Across simulated and real-world manipulation tasks, AnchorQ outperforms state-of-the-art post-training baselines in sample efficiency, delivering a 37% improvement in real-world success rate and 97% higher simulated task throughput under the same interaction budget. Additional visualizations are available on our website https://anchorq-rl.com/.
Figures & tables
| Method | LIBERO-15T | LIBERO Hard-10 | RoboTwin-6T | ||
| zero-shot | |||||
| VLM+FRS zero-shot | |||||
| Online RL env. steps | k | k | k | k | k |
| DSRL | |||||
| Residual RL | |||||
| AnchorQ (Ours) | |||||
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Short Key | LIBERO-90 | RoboTwin | Real world | Meaning |
| Environment and action space | |||||
| pi0_replan_steps | substeps per outer decision | ||||
| pi0_flow_steps | Euler steps in the flow ODE | ||||
| num_anchors | reference candidates including defer | ||||
| noise_shape | noise trajectory shape | ||||
| Critic, replay, and exploration | |||||
| Method | Task 41 | Task 81 | Task 75 | Task 73 |
| zero-shot | ||||
| VLM-FRS zero-shot | ||||
| DSRL | ||||
| Residual RL | ||||
| AnchorQ -AdaBeta-AdaSig | ||||
| AdaBeta, |
| Mechanism | Effect |
| online flow-reversal | small structured online action class |
| local particles | controlled relaxation of flow-reversed anchor support |
| exponential tilting | value-directed exploitation of that support |
| adaptive | epistemically calibrated exploitation |
Robot Policy Learning
Robot policy learning develops controllers that map sensory observations, task goals, and interaction history to actions for embodied robots.
See all 377 papers