atlas

Research map / Video / Video segmentation and tracking

Taming generative video models for zero-shot optical flow extraction

NeurIPS 2025 · Video › Video segmentation and tracking · approach: Stochastic video prediction · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Shows that frozen LRAS generative video model can be prompted zero-shot for optical flow via KL-tracing, a counterfactual perturbation method comparing logit distributions, competitive with supervised and unsupervised flow methods on TAP-Vid DAVIS and Kubric.

Problem

Zero-shot optical flow extraction from frozen self-supervised video generators without fine-tuning, avoiding synthetic-data supervision and photometric-loss heuristics.

Approach

Inject a small white Gaussian tracer into the first frame, run an LRAS autoregressive next-frame predictor with partial second-frame patches, and locate flow as the patch with maximal KL divergence between clean and perturbed predictive logit distributions, which avoids sampling noise in RGB differences.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use KL divergence of logits instead of RGB difference (ablation)LRAS flow extraction · TAP-Vid DAVIS subset (3%)EPE: 8.4797 -> 5.0762Helped
Add KL-tracing to LRASLRAS · TAP-Vid DAVISAD: 13.41 -> 11.18Helped
Add KL-tracing to LRASLRAS · TAP-Vid DAVISOA: 76.63 -> 74.58Hurt
Add KL-tracing to LRASLRAS · TAP-Vid KubricAD: 5.31 -> 5.06Helped
Add KL-tracing to LRASLRAS · TAP-Vid KubricAJ: 75.06 -> 65.49Hurt
Condition Cosmos on top 10% raster tokens (ablation)Cosmos flow extraction · TAP-Vid DAVIS subset (3%)EPE: 35.4338Hurt
Overwrite 10% of Cosmos rollout tokens with ground-truth tokens (ablation)Cosmos flow extraction · TAP-Vid DAVIS subset (3%)EPE: 37.7552Hurt
Provide Cosmos the full second frame (ablation)Cosmos flow extraction · TAP-Vid DAVIS subset (3%)EPE: 66.5521Hurt
Use a global latent code for frame generationStable Video Diffusion · Optical flow extractionflow precisionHurt
Use deterministic single-state predictionCWM flow extraction · Optical flow extractionflow precisionHurt

Surprising findings

Limitations

Experimental setup

Datasets: TAP-Vid DAVIS TAP-Vid Kubric

Benchmarks: TAP-Vid DAVIS TAP-Vid Kubric

Metrics: AJ AD <delta_avg OA EPE

Baselines: RAFT SEA-RAFT Doduo SMURF DiffTrack CWM Stable Video Diffusion Cosmos CoTracker-v3 AllTracker

Builds on: Counterfactual World Model LRAS VideoShop

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy