ProAR: Learning Prospective Reasoning with Autoregressive Video Models
Organizations: The Hong Kong Polytechnic University · University of California, Davis · Microsoft
Abstract
Autoregressive (AR) video models excel at causal generation, but their reliance on next-chunk prediction confines them to a short-sighted, reactive paradigm. This limitation is particularly consequential for reasoning-oriented generation, where achieving a target outcome through valid intermediate states matters more than local visual plausibility. To address this challenge, we propose Learning Prospective Reasoning with Autoregressive Video Models (ProAR), a novel framework that transforms autoregressive video generation into a goal-oriented reasoning process. ProAR introduces two key components: (1) To anchor generation to the long-range outcome, we integrate goal-frame prediction into the autoregressive loop via an asymmetric attention mask, enabling the predicted goal frame to guide the generation of intermediate states without being disrupted by them. (2) To guide short-range transitions, we introduce future representation self-alignment to encourage current hidden states to anticipate upcoming temporal dynamics. By leveraging teacher-forcing in AR training, we extract clean future representations in a single forward pass and align current representations with them using a lightweight, training-only predictor. Together, these two mechanisms seamlessly combine explicit, sparse target supervision with implicit, dense step-wise guidance, promoting coherent, goal-directed reasoning progress with modest computational cost. Experiments show that ProAR's complementary components consistently improve performance across diverse visual reasoning benchmarks. The framework proves highly training-efficient, surpassing fully trained standard AR baselines using only 25% of the training steps. This paradigm also demonstrates promising applicability to embodied reasoning tasks.
Figures & tables
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Task | Description |
| Concentric Ring (G194) | Move two circles to the same center while preserving their sizes and colors. |
| Outline Match (G25) | Move each colored shape into its matching outline. |
| Stable Sort (G3) | Group shapes by type and sort each group from smallest to largest. |
| Star Match (G5) | Move each colored object to the star marker of the same color. |
| Domino Chain (O23) | Trigger the first domino and propagate the falling sequence through the complete chain. |
| Ball Eater (O31) | Guide the black ball to consume smaller balls, growing after each step. |
| Task | Description |
| Maze | Move the agent from the green start to the red goal through open cells without crossing any walls. |
| FlowFree | Connect each pair of same-colored endpoints with a continuous path, without overlapping other paths, until the entire grid is filled. |
| Sokoban | Move the agent through free cells and push the box onto the green target without crossing walls or pulling the box. |
Video Reasoning
Video reasoning studies how AI systems use video evidence to infer events, actions, temporal relations, causes, and future states.
See all 208 papers