Implicit State Estimation via Video Replanning

TL;DR

ISE combines online embedding refinement and failed-plan rejection to improve video replanning on five Meta-World tasks with 400 trials each.

cs.RO 🔴 Advanced 2025-10-20 24 views
Po-Chen Ko Jiayuan Mao Yu-Hsiang Fu Hsien-Jeng Yeh Chu-Rong Chen Wei-Chiu Ma Yilun Du Shao-Hua Sun
robot learning video planning implicit state estimation diffusion models online adaptation

Key Findings

Methodology

Implicit State Estimation (ISE) encodes interaction videos with CLIP or DINOv2, retrieves and refines embeddings that implicitly represent hidden dynamics, and conditions a diffusion Video Plan Generator on them. A shared-parameter Identification Module uses successful and failed videos for denoising-based embedding optimization. The Rejection Module samples multiple hypotheses, stores failed plans, and selects the candidate farthest from prior failures. AVDC point tracking converts the selected video into executable actions.

Key Results

  • Evaluation uses the Meta-World System Identification Suite: Push Bar, Pick Bar, Slide Brick, Open Box, and Turn Faucet. Each task has 400 trials and permits at most M=14 replanning attempts. The paper reports substantially fewer replanning failures than non-adaptive AVDC and reinforcement-learning baselines, but the supplied text does not contain the numerical result table.
  • Plan fidelity is evaluated against ground-truth interaction videos using PSNR, LPIPS, CLIP-score, and DINO-score. This complements task success by testing whether generated trajectories reflect unknown centers of mass, interaction modes, and external forces.
  • Ablations compare DINOv2 retrieval, refinement initialized from a Gaussian prior, and original AVDC. The full retrieval-plus-refinement pipeline is the principal method; exact per-variant scores are not included in the provided excerpt.

Significance

ISE reframes failed interaction as useful evidence rather than wasted execution. It addresses a central weakness of video planning: a planner conditioned only on the current image cannot reliably resolve hidden mass, friction, functionality, or interaction direction. Academically, the work connects classical system identification, latent-context adaptation, and generative video planning without requiring a hand-designed parameterization of the hidden state. Industrially, it suggests a practical route to deployment-time adaptation in manipulation and navigation, where objects and contact conditions are difficult to model exhaustively.

Technical Contribution

The framework contributes an implicit alternative to explicit belief-state estimation. First, object-level interaction embeddings serve as hypotheses about physical configuration. Second, a shared Identification Module refines the embedding by minimizing L_refine=E[||ε−εθ(x_t(v),t,e)||²] while freezing model weights. Third, rejection-based replanning selects vp*=argmax_i min_F K(vp_i,F), explicitly discouraging repetition of failed behavior. Together, these components couple representation adaptation, hypothesis diversity, and action execution while retaining a fixed AVDC-based action interface.

Novelty

Relative to video planners that rely primarily on the initial frame, ISE incorporates deployment-time failed interactions into subsequent generation and adapts an internal embedding rather than explicitly estimating named physical variables. Its fundamental novelty is the integration of retrieval, generative refinement, and failed-plan rejection in one video-planning loop. It therefore differs from Hidden-Parameter MDPs and explicit system-identification methods, which generally assume a known latent-variable structure or maintain an interpretable belief.

Limitations

  • The method assumes that interaction videos contain enough visual evidence to distinguish hidden dynamics and that the offline dataset covers relevant configurations. Weak visual changes, ambiguous failures, or noisy action execution can make retrieval and gradient refinement unreliable.
  • Evidence is currently simulation-centered: the Meta-World suite uses an AVDC tracking-based Action Module and heuristic controllers. Real robots introduce occlusion, camera shift, contact uncertainty, collision risk, latency, and embodiment mismatch.
  • The supplied paper text omits full numerical tables, preventing precise claims about percentage gains, success rates, and metric differences.

Future Work

Important directions include real-robot validation, cross-embodiment transfer, calibrated uncertainty, and active selection of trials that maximize information gain. The authors also suggest links to out-of-distribution detection. Future studies should compare ISE with explicit belief models, verifier-based adaptation, and retrieval-based subtrajectory reuse on long-horizon tasks, while measuring safety, inference cost, and memory growth.

AI Executive Summary

Robots often fail not because they lack a plan, but because they do not know which physical world they are in. A box lid may lift or slide; a bar’s center of mass may shift; a faucet may turn clockwise or counterclockwise. Existing video planners commonly generate a fresh rollout from the current frame while discarding previous failures. This makes them prone to repeating physically inappropriate behavior in partially observed environments.

Implicit State Estimation (ISE) treats interaction history as evidence. CLIP or DINOv2 encodes videos into object-level embeddings. A Retrieval Module samples plausible prior embeddings, while a shared-parameter Identification Module refines them with a diffusion denoising loss. The Video Plan Generator receives the refined embedding as conditioning and produces seven future 128×128 frames. Multiple hypotheses are generated, and the Rejection Module selects the plan most dissimilar to those stored as failures. AVDC point or wrist tracking then translates the video into actions.

The Meta-World System Identification Suite contains Push Bar, Pick Bar, Slide Brick, Open Box, and Turn Faucet. Each task is evaluated for 400 trials with up to 14 replans, using replanning count plus PSNR, LPIPS, CLIP-score, and DINO-score. The paper reports fewer replanning failures and more accurate plans than original AVDC and RL baselines, although the supplied text does not include exact tables. The broader contribution is a practical bridge between generative video planning and online adaptation without explicit system identification. Real-world robustness, safety, computational cost, and long-horizon scaling remain open.

Deep Analysis

Background

Video has become a planning medium because it jointly represents appearance, actions, geometry, and temporal dynamics. Systems such as AVDC (Ko et al., 2024), video dynamics models, diffusion planners, and point-track controllers have enabled manipulation and navigation. Classical system identification estimates physical parameters, while Hidden-Parameter MDPs and meta-RL infer latent context. However, most video planners still assume that the current observation is sufficient and do not exploit failed deployment interactions.

Core Problem

Let θ denote unknown task or system properties such as mass, friction, center of mass, external force, or interaction mode. The offline dataset D contains video v, object ID o, and binary success s, but θ is not directly observed or even parameterized. The challenge is to use failures to update the planner, avoid repeating invalid strategies, and generate executable visual trajectories when the current frame alone cannot disambiguate the correct action.

Innovation

  • ��Implicit state estimation represents hidden dynamics through learned embeddings rather than hand-specified beliefs.
  • ��Retrieval plus refinement uses CLIP/DINOv2 similarity for initialization and diffusion-loss gradients for adaptation.
  • ��Rejection-based replanning samples diverse hypotheses and chooses a candidate farthest from failed plans.
  • ��A shared generative backbone supports both successful-plan generation and successful/failed interaction identification, while preserving a modular AVDC Action Module.

Methodology

  • ��Encoding: uniformly sample F=8 frames and concatenate CLIP or DINOv2 features into ev.
  • ��Canonicalization: choose one successful embedding eo per object ID.
  • ��Retrieval: compute di=||ev−evi||² and sample with pi=exp(−di/τ)/Σj exp(−dj/τ).
  • ��Refinement: update e^(k+1)=e^(k)−η∇eLrefine using frozen εθ and the diffusion loss.
  • ��Planning: condition the diffusion Video Plan Generator on the embedding; generate M=7 future frames at 128×128 and concatenate them with the initial frame.
  • ��Execution: reject candidates close to the failed-plan buffer and use AVDC object or wrist tracking plus heuristic control.

Experiments

The benchmark is the Meta-World System Identification Suite, with five tasks sharing the four-dimensional positional action space (dx,dy,dz,g). Hidden variation includes randomized centers of mass, two-mode mechanisms, and external forces. Each task uses 400 trials and a maximum of M=14 replans. Baselines include original non-adaptive AVDC, DINOv2-based Ours, Gaussian-prior refinement, and RL methods. Metrics are average replans per success and PSNR, LPIPS, CLIP-score, and DINO-score against interaction videos.

Results

Qualitative examples show changed plans after failed interactions across all five tasks. The reported quantitative conclusion is that ISE significantly reduces replanning failures and improves video-plan accuracy relative to original AVDC and RL baselines. Retrieval followed by refinement is the intended full pipeline, while Gaussian-prior refinement tests whether adaptation can work without retrieval. Exact success rates, replanning averages, and metric values are unavailable in the supplied text.

Applications

The method targets manipulation under unknown friction, payload, center of mass, mechanism direction, or external disturbance. It can support flexible manufacturing, warehouse handling, household robotics, and exploratory navigation. Prerequisites include an offline interaction-video dataset, a pretrained vision encoder, a diffusion planner, and an action module capable of tracking generated trajectories. Its main operational benefit is reducing repeated trial-and-error without building a complete explicit physics model.

Limitations & Outlook

ISE assumes that hidden dynamics are visually inferable and represented in the offline dataset. Visual similarity or plan distance may not correlate with physical feasibility, so rejection can encourage novelty without guaranteeing safety. Diffusion inference, multiple candidate samples, and gradient refinement add computation. Current validation is in simulated Meta-World, with training-free AVDC tracking and heuristic control; camera shifts, occlusions, contact dynamics, latency, collision avoidance, uncertainty calibration, and long-horizon real-robot transfer remain unresolved.

Plain Language Accessible to non-experts

Imagine a cook working in an unfamiliar kitchen. The cook sees a jar and a goal—open it—but does not know whether the lid twists, lifts, or slides. A first attempt fails. A sensible cook does not erase that experience and repeat the same motion; the failure narrows the possibilities. The cook remembers how the jar moved, finds similar experiences, adjusts the guess, and tries a different method.

ISE gives a robot this kind of practical memory. It turns operation videos into compact records of an object’s behavior: perhaps it is heavy, slippery, off-balance, or designed to move in a particular direction. After a failed attempt, the robot updates this record and imagines several new action movies. It then avoids movies that look too much like earlier failures.

The robot still needs a way to act. It follows the moving object or its own hand in the imagined movie and converts that movement into commands. In five simulated tasks—pushing, picking, sliding, opening, and turning—the authors tested 400 trials per task and allowed up to 14 replans. They report fewer failures than a planner that ignores experience. The important idea is simple: failure becomes a useful clue, not merely a setback.

ELI14 Explained like you're 14

Picture a new video game where a door blocks your character, but the game never tells you whether to push or pull it. You push, nothing happens. Would you keep pushing forever? Of course not! You would remember the failed move, try pulling, or search for another route.

This paper teaches a robot to behave more like that. The robot watches old gameplay-like robot videos and creates a hidden “personality card” for each object. The card does not literally say “heavy” or “slippery”; it stores patterns showing how the object reacts. When a real attempt fails, the robot edits the card using what just happened.

Next, it makes several mini-movies of possible futures. One module imagines the plans, another keeps a list of failed plans, and a rejection module picks an option that looks most different from the bad ones. A tracking method called AVDC turns the chosen movie into hand and object movements. So the robot is not simply guessing once—it is guessing, checking, remembering, and changing strategy!

Researchers tested this on five Meta-World challenges: pushing a bar, picking a bar, sliding a brick, opening a box, and turning a faucet. Each task had 400 trials and up to 14 replans. The authors report fewer replanning failures than older video planning and RL baselines. The catch? These tests were simulated, so real robots still need to handle messy cameras, bumps, delays, and safety.

Glossary

Implicit State Estimation

A way to represent hidden environment properties without explicitly naming or estimating each physical variable. The representation is an adaptable internal embedding learned from interaction evidence.

The central principle of the ISE framework.

Video Plan Generator

A diffusion model that predicts plausible future visual trajectories conditioned on the current frame and an environment embedding. It generates seven future frames at 128×128 resolution.

Produces candidate plans for execution.

Identification Module

A shared-parameter generative model trained on both successful and failed interactions. At test time, its weights are frozen and the conditioning embedding is optimized.

Refines the implicit state representation.

Rejection Module

A selector that compares candidate plans with a buffer of failed plans and chooses the least similar candidate. It encourages behavioral exploration after failure.

Prevents direct repetition during replanning.

Canonical Embedding

A representative vector selected for an object ID from a successful interaction. It summarizes interaction dynamics and provides a retrieval or planning condition.

Constructed from the offline experience dataset.

AVDC

A training-free video-to-action approach based on dense object or wrist tracking and heuristic trajectory control. It provides the action interface in this paper.

Converts generated videos into executable robot commands.

Open Questions Unanswered questions from this research

  • 1 Will embedding refinement remain reliable on real robots with occlusion, camera shift, noisy contacts, and safety constraints? Hardware experiments and risk-aware rejection are needed.
  • 2 The supplied text lacks complete result tables, so the exact task-wise gains and statistical significance remain unclear. Future reports should include confidence intervals and inference cost.
  • 3 Distance from failed plans is only a behavioral proxy, not a physical safety guarantee. Learning feasibility-aware and uncertainty-calibrated distances remains open.

Applications

Immediate Applications

Flexible object manipulation

Robots can use prior push, pick, and slide videos to adapt to objects with different centers of mass or friction. Users need an offline video corpus, a vision encoder, and a tracking controller; the expected benefit is fewer repeated failures after limited trials.

Ambiguous mechanisms

For lids that lift or slide, or faucets with uncertain rotation direction, the system can record failed modes and generate contrasting alternatives. It suits laboratory and warehouse prototypes, provided collision checks and retry limits are enforced.

Long-term Vision

Open-world adaptive robots

A future robot could continuously convert deployment videos into reusable implicit dynamics knowledge and transfer it across novel objects, environments, and embodiments. Achieving this requires real-world data, persistent memory, calibrated uncertainty, and safe information-seeking actions.

Abstract

Video-based representations have gained prominence in planning and decision-making due to their ability to encode rich spatiotemporal dynamics and geometric relationships. These representations enable flexible and generalizable solutions for complex tasks such as object manipulation and navigation. However, existing video planning frameworks often struggle to adapt to failures at interaction time due to their inability to reason about uncertainties in partially observed environments. To overcome these limitations, we introduce a novel framework that integrates interaction-time data into the planning process. Our approach updates model parameters online and filters out previously failed plans during generation. This enables implicit state estimation, allowing the system to adapt dynamically without explicitly modeling unknown state variables. We evaluate our framework through extensive experiments on a new simulated manipulation benchmark, demonstrating its ability to improve replanning performance and advance the field of video-based decision-making.

cs.RO