Video Replanning via Latent Embedding Refinement and Rejection-Based Sampling
Organizations: National Taiwan University · MIT CSAIL · Cornell University · Harvard University
Abstract
Video planning has emerged as a flexible framework for robot manipulation, in which a generative model predicts a video of task completion, and a downstream module translates the predicted frames into actions. However, existing methods typically ignore information from past interactions, limiting their ability to adapt to latent physical properties that can only be revealed through trial and error, such as whether a door should be pushed or pulled, or how friction affects object dynamics. When a plan fails, these methods usually replan from scratch without leveraging the information revealed by the failure. To address this limitation, we introduce RELIC, REplanning with Latent embedding refInement and Candidate rejection, a video planning framework that adapts to hidden physical properties from test-time failures. RELIC optimizes a latent embedding that captures the environment's hidden physical properties from interaction videos and introduces a rejection-based sampling mechanism that filters out hypotheses inconsistent with prior failures. Across eight tasks in two simulation suites, RELIC consistently reduces the number of replanning steps required for success, and linear probes show that its embedding captures the hidden parameters from the interaction itself rather than from scene appearance. Across four challenging real-world robotic manipulation tasks involving hidden interaction modes, e.g., friction, center of mass, and object mass, RELIC raises the one-shot replanning success rate of a video planning baseline from 30.0% to 63.8% after a single physical interaction.
Figures & tables
| Meta-World | ManiSkill3 | ||||||||
| Method | PushBar | PickBar | SlideBrick | OpenBox | TurnFaucet | Seesaw | Bounce | Strike | Overall [-2pt] (all 8, norm.) |
| CQL | 0.20 | 0.19 | 0.18 | 0.18 | 0.21 | 0.18 | 0.68 | 0.37 | 0.08 |
| BC | 0.30 | 0.32 | 0.33 | 0.19 | 0.27 | 0.09 | 0.71 | 0.20 | 0.06 |
| AVDC | 0.27 | 0.22 | 0.27 | 0.16 | 0.16 | 0.47 | 0.48 | 0.76 | 0.06 |
| VLM verifier | 0.40 | 0.38 | 0.45 | 0.36 | 0.16 | 0.48 | 0.43 | 0.41 | 0.06 |
| RELIC (Refine FS) | 0.23 | 0.21 | 0.26 | 0.16 | 0.15 | 0.37 | 0.38 | 0.26 | 0.04 |
| Suite | Tasks | Videos | Successful |
|---|---|---|---|
| Meta-World | 5 | 14,270 | 670 |
| ManiSkill3 | 3 | 18,750 | 750 |
| Method | PSNR | LPIPS | CLIP | DINO |
|---|---|---|---|---|
| AVDC | 19.426 | 0.271 | 0.870 | 0.705 |
| RELIC (R3M) | 19.632 | 0.261 | 0.875 | 0.705 |
| RELIC (CLIP) | 19.584 | 0.262 | 0.871 | 0.707 |
| RELIC (DINOv2, Ours) | 19.817 | 0.255 | 0.882 | 0.717 |
| Latent-conditioned policy | Latent in RELIC pipeline | ||||
|---|---|---|---|---|---|
| Task | HiP-MDP | PEARL | HiP-MDP | PEARL | RELIC |
| PushBar | 0.24 | 0.23 | 0.24 | 0.21 | 0.20 |
| PickBar | 0.23 | 0.24 | 0.23 | 0.21 | 0.18 |
| SlideBrick | 0.22 | 0.13 | 0.23 | 0.23 | 0.26 |
| OpenBox | 0.15 | 0.14 | 0.13 | 0.04 | 0.10 |
| TurnFaucet | 0.14 | 0.09 | 0.14 | 0.05 | 0.14 |
| Task | Hidden | Metric | Post | Pre |
|---|---|---|---|---|
| PushBar | center of mass | 0.77 | ||
| PickBar | center of mass | 0.82 | ||
| SlideBrick | friction | 0.84 | ||
| OpenBox | push vs. pull | acc. (%) | 100 | 65 |
| TurnFaucet | CW vs. CCW | acc. (%) | 92.5 | 55 |
| Method | OpenDoor | SlideBlock | Balance | Pinball | Overall |
|---|---|---|---|---|---|
| AVDC | 8/20 | 4/20 | 7/20 | 5/20 | 30.0% |
| RELIC (Ours) | 14/20 | 7/20 | 17/20 | 13/20 | 63.8% |
Appendix figures & tables19 assets
Supplementary material from the paper’s appendix.
Appendix
| Task | Hidden parameter | Robot | Trials (S/F) |
|---|---|---|---|
| Open Door | Push vs. pull direction | WidowX (Mobile ALOHA) | 10/10 |
| Slide Block | Block friction coefficient | Piper-X | 5/20 |
| Balance | Bar’s center of mass | Piper-X | 6/12 |
| Pinball | Ball mass | Piper-X | 6/12 |
| Success Rate | |
|---|---|
| 4/10 | |
| (default) | 7/10 |
| 3/10 |
| num_parameters | 125M |
|---|---|
| diffusion_resolution | (32, 32) |
| target_resolution | (128, 128) |
| base_channels | 128 |
| num_res_block | 2 |
| attention_resolutions | (2, 4, 8) |
| channel_mult | (1, 2, 3, 4) |
| PushBar | PickBar | SlideBrick | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Method | SR (%) | Trials | Fails | SR (%) | Trials | Fails | SR (%) | Trials | Fails |
| Latest failure only(Ours) | 95.8 | 4.28 0.20 | 17 | 98.0 | 3.46 0.17 | 8 | 82.8 | 6.64 0.54 | 69 |
| Full failure history | 98.0 | 4.08 0.19 | 8 | 99.0 | 3.46 0.17 | 4 | 77.2 | 7.39 0.51 | 91 |
| Component | AVDC | Ours |
|---|---|---|
| Retrieve | N/A | 0.062 0.001 |
| Refine | N/A | 2.713 0.010 |
| DDIM | 0.542 0.001 | 0.546 0.002 |
| Reject | N/A | 0.078 0.001 |
| Optical Flow | 0.097 0.000 | 0.097 0.000 |
| Total | 0.639 0.001 | 3.496 0.010 |
| PushBar | PickBar | SlideBrick | OpenBox | TurnFaucet | Total | |
| Main exp (100%) | 6000/250 | 6000/250 | 1560/130 | 20/20 | 20/20 | 13620/650 |
| 28% | 1000/250 | 1000/250 | 1300/130 | 20/20 | 20/20 | 3360/650 |
| Method Dataset Size | ||||
|---|---|---|---|---|
| (Successful, Failed) | (2, 14) | (10, 70) | (40, 280) | (120, 840) |
| Random | 11.87 1.28 | 11.87 1.28 | 11.87 1.28 | 11.87 1.28 |
| Ours (Retrieval) | 18.59 1.54 | 30.94 1.83 | 53.12 1.97 | 67.66 1.85 |
| Method | PushBar | PickBar | SlideBrick | OpenBox | TurnFaucet | Overall (Normalized) |
|---|---|---|---|---|---|---|
| Attract | ||||||
| Reject (Ours) |
| Method | SlideBrick Noisy | OpenBox Noisy |
|---|---|---|
| No Rejection | ||
| DINO | ||
| L2 error | ||
| Seg + DINO | ||
| Seg + L2 error |
| Method | PushBar | OpenBox | PickBar | TurnFaucet | SlideBrick |
|---|---|---|---|---|---|
| Random Guess | |||||
| RELIC-Seen | |||||
| RELIC-Unseen |