Value as Memory: Online Value-Guided Post-training for General Robot Policies
Organizations: Shanghai Jiao Tong University · Shanghai Innovation Institute · Noematrix Ltd.
Abstract
Real-world post-training of generalist robot policies faces two challenges: partial observability makes action selection ambiguous, while offline post-training methods delay feedback from deployment experience. We present Value as Memory (VaM), a framework that addresses these challenges through value caching and online real-world reinforcement learning (RL). Value caching conditions actions on recent value predictions as task-progress memory, while online updates enable timely policy improvement by combining offline demonstrations, off-policy interventions, and on-policy rollouts for sample-efficient adaptation. As a result, VaM transitions from offline RL warmup to online autonomous learning with sparse rewards. Across long-horizon tasks involving deformable objects and contact-rich manipulation, VaM yields a mean absolute success-rate gain of with value caching, while achieving a gain with online autonomous learning over matched offline training. The evaluated online schedule performs as many policy iterations under the same gradient-step budget, with more than lower recorded time per gradient step with concurrent data collection and training.
Figures & tables
| Method | Box Assembly | Laundry Folding | Magnetic Bus Assembly |
|---|---|---|---|
| VaM | \mathbf{96.7}\,{\color[rgb]{0.5,0.5,0.5}\pm 2.9} | \mathbf{88.9}\,{\color[rgb]{0.5,0.5,0.5}\pm 10.2} | \mathbf{76.7}\,{\color[rgb]{0.5,0.5,0.5}\pm 5.8} |
| VaM (w/o value caching) | 81.7\,{\color[rgb]{0.5,0.5,0.5}\pm 2.9} | 48.9\,{\color[rgb]{0.5,0.5,0.5}\pm 30.1} | 43.3\,{\color[rgb]{0.5,0.5,0.5}\pm 5.8} |
| VaM (w/o offline RL warmup) | 81.7\,{\color[rgb]{0.5,0.5,0.5}\pm 7.6} | 60.6\,{\color[rgb]{0.5,0.5,0.5}\pm 18.3} | 50.0\,{\color[rgb]{0.5,0.5,0.5}\pm 10.0} |
| RECAP | 71.7\,{\color[rgb]{0.5,0.5,0.5}\pm 16.1} | 64.4\,{\color[rgb]{0.5,0.5,0.5}\pm 7.7} | 43.3\,{\color[rgb]{0.5,0.5,0.5}\pm 5.8} |
| VaM (demo-trained value) | 80.0\,{\color[rgb]{0.5,0.5,0.5}\pm 13.2} | 71.1\,{\color[rgb]{0.5,0.5,0.5}\pm 20.4} | 40.0\,{\color[rgb]{0.5,0.5,0.5}\pm 10.0} |
| VaM (post-warmup value) | 85.0\,{\color[rgb]{0.5,0.5,0.5}\pm 13.2} | 68.9\,{\color[rgb]{0.5,0.5,0.5}\pm 10.2} | 50.0\,{\color[rgb]{0.5,0.5,0.5}\pm 10.0} |
| Method | Box Assembly | Laundry Folding | Magnetic Bus Assembly |
|---|---|---|---|
| VaM | |||
| VaM (w/o value caching) | |||
| VaM (w/o offline RL warmup) | |||
| RECAP |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Task | Offline demonstrations (episodes) | Offline RL warmup (episodes) | Online RL rollouts per round (episodes) |
|---|---|---|---|
| Box Assembly | 200 | 100 | 20 |
| Laundry Folding | 90 | 50 | 15 |
| Magnetic Bus Assembly | 30 | 40 | 10 |
| Task | VaM | VaM (w/o value caching) | Gap | |
|---|---|---|---|---|
| Box Assembly | 8 | |||
| Laundry Folding | 3 | |||
| Magnetic Bus Assembly | 4 |
| Task | Method | # recovered / # required | # episodes | Recovery rate |
|---|---|---|---|---|
| Box Assembly | VaM | 24/44 | 180 | |
| VaM (w/o value caching) | 35/70 | 180 | ||
| VaM (w/o offline RL warmup) | 20/79 | 180 | ||
| RECAP | 36/85 | 180 | ||
| Laundry Folding | VaM | 38/44 | 45 | |
| VaM (w/o value caching) | 30/51 | 45 |
| Task | Setting | Time per gradient step (s) | Total experimental time (hours) |
|---|---|---|---|
| Box Assembly | Offline RL | 4.885\,{\color[rgb]{0.5,0.5,0.5}\pm 0.413} | 6.514 |
| Online RL | 1.887\,{\color[rgb]{0.5,0.5,0.5}\pm 0.107} | 2.516 | |
| Laundry Folding | Offline RL | 5.090\,{\color[rgb]{0.5,0.5,0.5}\pm 0.345} | 7.069 |
| Online RL | 2.253\,{\color[rgb]{0.5,0.5,0.5}\pm 0.057} | 3.129 | |
| Magnetic Bus Assembly | Offline RL | 5.891\,{\color[rgb]{0.5,0.5,0.5}\pm 0.105} | 7.213 |
| Online RL | 2.571\,{\color[rgb]{0.5,0.5,0.5}\pm 0.160} | 3.061 |
| Hyperparameter | Scope | Task | Value |
|---|---|---|---|
| Value-support atoms | Value expert | All tasks | 256 |
| Action horizon (steps) | Action expert | All tasks | 100 |
| Control frequency (Hz) | Deployment | All tasks | 100 |
| Progress-label std. dev. | Value expert | All tasks | |
| Target-network EMA coefficient | Value expert | All tasks | 1.0 |
| Value-training stride (steps) | Value training | All tasks | 30 |
Robotic RL
Robotic reinforcement learning uses trial-and-error learning to help robots control, navigate, and manipulate objects.
See all 283 papers