AdaPolicy: Action-Driven Anticipatory Robot Policies via Next-Chunk Prediction
Organizations: Tianjin University · University of Bologna
Abstract
Endowing robot policies with anticipatory capabilities has proven effective in improving control performance. However, existing methods predominantly rely on explicit supervision derived from future observations, in either pixel or latent space, at the cost of additional architectural and computational complexity. Departing from explicit future supervision, we ask whether such foresight can be elicited directly from the action signal. To this end, we propose AdaPolicy, an Action-driven anticipatory policy framework that shapes future-aware representations via cascaded next-chunk prediction. Specifically, an Online Action Head jointly produces the current action chunk and an evolved latent state. Building on this state, a cascaded Unrolled Action Head forecasts the next chunk, and the resulting loss flows back through the evolved state to instill implicit foresight into the online policy. Importantly, this training-only unroll requires no future-observation encoding, keeping AdaPolicy lightweight to train with zero inference-time overhead. Experiments on MetaWorld, RoboTwin 2.0, LIBERO, and real-world deployments demonstrate consistent gains across diverse settings, including up to percentage points on millimeter-precision tasks such as earphone assembly and shoelace threading. More details and visualizations are available at https://teee111.github.io/AdaPolicy/.
Figures & tables
| Method | Earphone Assembly | Shoelace Threading | Stacking Blocks | Stacking Cups | Button Pressing | Lid Opening | Avg. |
|---|---|---|---|---|---|---|---|
| Baseline | 74 | 36 | 82 | 100 | 70 | 100 | 77.0 |
| AdaPolicy | 86 | 50 | 94 | 100 | 88 | 100 | 86.3 |
| Method | TinyVLA [ 74 ] | [ 75 ] | SmolVLA [ 76 ] | VPP [ 24 ] | RoMAN-Flow [ 77 ] | AdaPolicy |
|---|---|---|---|---|---|---|
| Avg. | 31.6 | 50.8 | 68.2 | 68.2 | 72.8 | 81.2 |
| (a) 50 tasks, 50 clean + 500 randomized demos per task | |||||
|---|---|---|---|---|---|
| Method | X-VLA | ABot-M0 | Baseline | AdaPolicy | |
| Avg. ( clean ) | 65.9 | 72.8 | 81.2 | 64.4 | 84.92 |
| Avg. ( randomized ) | 58.4 | 72.8 | 80.4 | 64.5 | 84.04 |
| Method | Goal | Object | Spatial | Long | Avg. |
|---|---|---|---|---|---|
| OpenVLA [ 1 ] | 79.2 | 88.4 | 84.7 | 53.7 | 76.5 |
| SmolVLA [ 76 ] | 91.0 | 94.0 | 93.0 | 77.0 | 88.8 |
| DreamVLA [ 78 ] | 89.5 | 94.0 | 97.5 | 89.5 | 92.6 |
| GR00T-N1 [ 79 ] | 94.4 | 97.6 | 93.0 | 90.6 | 93.9 |
| [ 75 ] | 96.8 | 98.8 | 95.8 | 85.2 | 94.1 |
| Baseline | 93.0 | 96.4 | 96.0 | 83.0 | 92.1 |
| Variant | EMA | Avg. | ||
|---|---|---|---|---|
| w/o next-chunk prediction | ||||
| Baseline | 8 | – | – | 67.8 |
| Baseline (extended horizon) | 16 | – | – | 74.3 |
| Baseline (extended horizon) | 24 | – | – | 60.4 |
| Baseline + future-feature supervision | 8 | – | – | 68.8 |
| w/ next-chunk prediction | ||||
| (a) Loss weight on MetaWorld ( ) | ||||||
|---|---|---|---|---|---|---|
| 0 | 0.5 | 0.6 | 0.7 | 0.8 | 0.9 | |
| Avg. | 67.8 † | 76.4 | 74.4 | 76.4 | 76.7 | 75.9 |
| (b) EMA momentum on MetaWorld ( ) | ||||||
| 0.95 | 0.96 | 0.97 | 0.98 | 0.99 | 0.995 | |
| Avg. | 76.0 | 76.2 | 76.7 | 76.7 | 81.2 | 81.4 |
| (c) EMA momentum on RoboTwin 2.0 | ||||||
| MetaWorld | RoboTwin 2.0 | |||||||
| 1-step | 2-step | 1-step | 2-step | |||||
| , | , | , | , | , | ||||
| Avg. | 81.2 | 75.6 | 77.4 | 75.1 | 72.4 | 67.0 | 66.2 | 61.0 |
| Noise Sampling Strategy | Condition on | RoboTwin 2.0 | MetaWorld |
|---|---|---|---|
| Baseline (w/o next-chunk) | – | 63.8 | 67.8 |
| Higher next-chunk noise | ( ) | 63.8 | 76.2 |
| Lower next-chunk noise | ( ) | 69.2 | 79.0 |
| Shared noise timestep | – | 75.4 | |
| Independent sampling (AdaPolicy) | (default) | 72.4 | 81.2 |
| Temporal Offset | Next-chunk Interval | Avg. |
|---|---|---|
| 16 | 58.6 | |
| 32 (default) | 61.8 | |
| 48 | 59.4 |
| Model | Probed features | RoboTwin 2.0 | MetaWorld | |
| Baseline | 8 | condition state | 16.38 0.12 | 2.89 0.09 |
| evolved state | 15.76 0.07 | 2.67 0.05 | ||
| Baseline + future-feature sup. | 8 | condition state | 15.10 0.19 | 2.78 0.05 |
| evolved state | 14.07 0.23 | 2.55 0.05 | ||
| Baseline + extended horizon | 16 | condition state | 15.28 0.16 | 3.00 0.07 |
| evolved state | 14.06 0.25 | 2.73 0.09 |
| Cosine similarity | Attention mass | ||
| cond action | |||
| (current) | (next) | (avg.) | |
| Baseline | 0.9977 | 0.6458 | 0.148 |
| AdaPolicy | 0.9650 | 0.9424 | 0.195 |
| Method | reach@2cm | reach@3cm | reach@5cm |
|---|---|---|---|
| Baseline | 6.0 | 8.2 | 17.8 |
| AdaPolicy | 12.8 | 15.5 | 29.5 |
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
| Variant | ms/step | Memory | Params | GFLOPs |
|---|---|---|---|---|
| (a) Hidden size | ||||
| AdaPolicy | 1238 | 17.1 GB | 93.1M | 648.2 |
| Chunk-32 baseline | 1236 | 13.7 GB | 93.1M | 611.5 |
| Chunk-64 baseline | 1263 | 14.9 GB | 93.1M | 623.6 |
| Future-feature sup. | 2415 | 17.2 GB | 111.7M | 1163.8 |
| (b) Hidden size | ||||
| Model | Probed features | Avg. | |||
|---|---|---|---|---|---|
| Baseline ( ) | condition state | 11.85 0.13 | 19.00 0.41 | 18.29 0.22 | 16.38 0.12 |
| evolved state | 11.60 0.13 | 18.05 0.05 | 17.62 0.27 | 15.76 0.07 | |
| Baseline ( ) + future-feature sup. | condition state | 10.95 0.18 | 16.93 0.59 | 17.43 0.17 | 15.10 0.19 |
| evolved state | 10.36 0.10 | 15.75 0.70 | 16.11 0.24 | 14.07 0.23 | |
| Baseline ( ) | condition state | 11.02 0.17 | 17.26 0.67 | 17.57 0.08 | 15.28 0.16 |
| evolved state | 10.41 0.09 | 16.06 0.82 | 15.70 0.02 | 14.06 0.25 |
| Model | Probed features | 4 | 8 | 12 | 16 | Avg. |
|---|---|---|---|---|---|---|
| Baseline ( ) | condition state | 1.44 0.03 | 2.54 0.06 | 3.42 0.10 | 4.16 0.15 | 2.89 0.09 |
| evolved state | 1.33 0.02 | 2.32 0.04 | 3.16 0.06 | 3.87 0.09 | 2.67 0.05 | |
| Baseline ( ) + future-feature sup. | condition state | 1.42 0.02 | 2.48 0.05 | 3.29 0.06 | 3.93 0.06 | 2.78 0.05 |
| evolved state | 1.33 0.04 | 2.26 0.06 | 2.99 0.06 | 3.61 0.05 | 2.55 0.05 | |
| Baseline ( ) | condition state | 1.60 0.04 | 2.72 0.06 | 3.55 0.10 | 4.12 0.11 | 3.00 0.07 |
| evolved state | 1.48 0.05 | 2.48 0.07 | 3.22 0.09 | 3.73 0.15 | 2.73 0.09 |
| Setting | Embodiment | Action space | Camera views |
|---|---|---|---|
| RoboTwin 2.0 | ALOHA-AgileX bimanual | 14-D absolute joint + gripper (7-D per arm: 6 joints + 1 gripper) | one fixed third-person (head) camera |
| MetaWorld | Sawyer single arm | 4-D end-effector displacement + gripper | two third-person cameras (corner, behindGripper) |
| LIBERO | Franka Panda single arm | 7-D operational-space delta + gripper | wrist camera only |
| Real world | AgileX Piper single arm | 7-D absolute joint + gripper (6 joints + 1 gripper) | two fixed third-person RealSense D435 cameras |
| Category | Setting | MetaWorld & LIBERO | RoboTwin 2.0 |
|---|---|---|---|
| Input | Image resolution | ||
| Architecture | Feature dimension | 512 | |
| Adapter query number | 32 | 64 | |
| Adapter layers | 4 | ||
| Optimization | Optimizer | AdamW | |
| Learning rate | |||
| Unrolled Head Variant | Next-Chunk Gradient Norm (Mean Std) | Variance |
|---|---|---|
| w/o EMA (detached online copy) | ||
| w/ EMA (AdaPolicy) |
| lever-pull | basketball | peg-insert-side | stick-push | pick-out-of-hole | sweep | shelf-place | Avg. | |
|---|---|---|---|---|---|---|---|---|
| Baseline | 54 | 34 | 14 | 10 | 2 | 4 | 4 | 17.4 |
| AdaPolicy | 64 | 24 | 16 | 26 | 16 | 12 | 4 | 23.1 |
| X-VLA | AdaPolicy | X-VLA | AdaPolicy | X-VLA | AdaPolicy | |||||||||
| Task | Clean | Rand. | Clean | Rand. | Task | Clean | Rand. | Clean | Rand. | Task | Clean | Rand. | Clean | Rand. |
| Adjust Bottle | 100 | 99 | 100 | 100 | Open Microwave | 79 | 71 | 82 | 58 | Place Object Stand | 86 | 88 | 84 | 94 |
| Beat Block Hammer | 92 | 88 | 94 | 80 | Pick Diverse Bottles | 58 | 36 | 72 | 84 | Place Phone Stand | 88 | 87 | 82 | 88 |
| Blocks Ranking RGB | 83 | 83 | 100 | 96 | Pick Dual Bottles | 47 | 36 | 92 | 92 | Place Shoe | 96 | 95 | 100 | 100 |
| Blocks Ranking Size | 67 | 74 | 62 | 74 | Place A2B Left | 48 | 49 | 58 | 76 | Press Stapler | 92 | 98 | 98 | 96 |
| Click Alarmclock | 99 | 99 | 92 | 88 | Place A2B Right | 36 | 36 | 56 | 54 | Put Bottles Dustbin | 74 | 77 | 90 | 92 |
Robot Policy Learning
Robot policy learning develops controllers that map sensory observations, task goals, and interaction history to actions for embodied robots.
See all 372 papers