← Xiaoyu Zhu / Technical Notes · Multimodal Reasoning
Read our paper ↗

Beyond Visual CoT: Internalized Visual Thinking for Proactive Video Reasoning

Internalized Visual Thinking trains a multimodal LLM to predict next latent representations of future frames from unlabeled videos, then removes the image generation pathway at inference. The model keeps the foresight, but not the pixel-generation bill.

Paper: Zhu et al., 2026· Estimated reading time: ~14 min· Topics: video reasoning, world models, multimodal post-training
next-embedding prediction Visual CoT predictive world modeling proactive video reasoning
Core result. Across six dataset–task settings, Internalized Visual Thinking (IVT) improves every reported answer-quality metric over text-only post-training. Besides, IVT beats Visual CoT in four of six settings, and keeps mean inference latency at 1.22 s versus 6.56 s for Visual CoT.

Quick Summary

  • Visual CoT externalizes foresight. It generates an intermediate future image, re-encodes it, and then predicts the answer.
  • IVT internalizes foresight. During training, the model predicts both the textual target and latent embeddings of future frames.
  • The future-prediction branch disappears at inference. Inference is the same direct observed-video → text path as Answer-Only SFT.
  • The training recipe matters. Flux-VAE targets, sustained predictive updates, joint optimization, and carefully chosen prediction horizons are all important.
  • The broader idea: world modeling can be useful as a training-time representational constraint even when no world-model rollout is executed at test time.

A common intuition in multimodal reasoning is that if a model needs to reason about the visual world, it should show its work visually: generate a sketch, a frame, or an imagined future, then reason from it. We ask a sharper question: is rendering the visual thought actually necessary, or is learning to predict it enough?

1. The Problem: Reason Before the Future Arrives

Most video understanding benchmarks are retrospective. The evidence needed to answer the question is already somewhere in the clip. Proactive video reasoning is different: the model must make a useful prediction from an incomplete observation.

Early-event prediction (EEP)

The target action has started but is not finished. The observation boundary is placed inside an annotated action segment:

t = ts + ρ(te − ts),   ρ ∈ {0.1, 0.3, 0.5, 0.7}

The model must identify the action from only 10–70% of its execution.

Next-event prediction (NEP)

The target action has not started. The observation ends at least one second before the next annotated event:

t ≤ tnext − τa,   τa = 1 s

This is closer to anticipation: infer what comes next from current objects, motion, context, and latent intent.

For proactive systems, being correct is not enough. A prediction is less useful if it comes only after the event has ended.

This latency constraint makes the form of reasoning important. A slow “thinking” procedure can erase the benefit of a more accurate model.

2. Why Not Just Generate the Future?

Predictive Visual Chain-of-Thought follows an intuitive pipeline:

observed video → generated future image → re-encoded visual tokens → textual prediction.

That explicit future frame can carry information that language is bad at compressing: object state, spatial arrangement, motion direction, contact relationships, and fine visual details. But it also adds a dense generation problem to every inference request.

Three paradigms: Answer-Only SFT, Visual CoT, and Internalized Visual Thinking.
Figure 1. The three post-training paradigms we compare. IVT uses future visual prediction only during training; inference collapses back to direct text generation.

The oracle-frame diagnosis

We run a control experiment here. Instead of asking only whether generated future frames help, we separate two questions:

  1. Is future visual information useful?
  2. Can a generated image deliver that information faithfully and cheaply enough?

To answer the first question, we replace the generated future frame with the ground-truth future frame at the same checkpoint. This “oracle” is impossible at deployment, but diagnostically clean.

SettingMean latencyROUGE-LMETEORCIDEr
EEP · Answer-Only SFT1.07 s34.424.31.43
EEP · Visual CoT, generated6.56 s36.429.81.69
EEP · Visual CoT, oracle41.934.22.05
NEP · Answer-Only SFT1.32 s47.235.32.14
NEP · Visual CoT, generated6.55 s46.836.92.13
NEP · Visual CoT, oracle50.440.42.38
The conceptual hinge: the oracle future helps a lot, especially for early-event prediction. Therefore the problem is not that “future visual context is useless.” The problem is that externalizing that context as pixels is both expensive and imperfect.

This suggests a different use for the future: do not sample it at inference; use it as a training target that shapes the representation from which the language model answers.

3. What Makes Predictive Supervision Work?

3.1 What should the model predict?

We compare four target spaces:

  • Flux-VAE latents: reconstruction-oriented; retain fine-grained appearance and spatial information.
  • DINOv2 features: semantic, higher-level visual features.
  • SigLIP2-adaptive: semantic target encoder that can be jointly optimized through the text objective.
  • SigLIP2-frozen: a stationary semantic target space.
Comparison of VAE, DINOv2, and SigLIP2 target representations.
Figure 2. Flux-VAE future targets are the strongest and most consistent across both early-event and next-event prediction.

The result is notable: reconstruction-oriented VAE targets beat the more semantic alternatives. DINOv2 still helps, so high-level semantic prediction is useful.

3.2 Predictive supervision cannot be an occasional regularizer

Next, we vary how frequently the model sees language-understanding versus future-prediction examples. Ratios are understanding:prediction = 1:1, 3:1, and 5:1.

Training curves for different understanding to prediction sampling ratios.
Figure 3. Understanding-heavy mixtures improve faster at first, but plateau; the balanced 1:1 mixture keeps improving and wins at 20k steps.

The 3:1 and 5:1 mixtures look better early. This is easy to understand: more answer-supervision updates accelerate adaptation to the downstream text target. But by roughly 12–18k steps they saturate or decline. The 1:1 mixture starts slower and finishes much stronger.

At 20k steps, the balanced mixture reaches ROUGE-L 34.5, BERTScore 36.5, METEOR 27.4, and CIDEr 2.02 on the Ego-Exo4D early-event ablation.

This suggests that future prediction is not behaving like a tiny regularization term. It needs enough optimization mass to keep influencing the shared representation instead of being overwritten by task-specific text supervision.

3.3 The best predictive loss depends on the target space

Flow matching versus direct regression for VAE and DINOv2 targets.
Figure 4. Direct regression is decisively better for DINOv2; for VAE targets, regression and flow matching are much closer.

For DINOv2, direct regression consistently dominates flow matching. For Flux-VAE, the gap is much smaller: regression often leads during training, while flow matching catches up on some metrics at the final checkpoint.

The practical lesson is that representation and objective should be selected jointly. A target inherited from a generative model does not automatically imply that its original generative training objective is best for transferring predictive structure into a reasoner.

3.4 Joint training beats “predict first, reason later”

A natural alternative is a two-stage curriculum:

Stage 1: learn next-embedding prediction → Stage 2: fine-tune only on text answers.

It fails surprisingly badly.

MethodROUGE-LBERTScoreMETEORCIDEr
Answer-Only SFT30.632.522.61.60
Two-stage training28.6 (−6.5%)30.7 (−5.6%)21.3 (−5.5%)1.54 (−3.5%)
Joint training34.5 (+12.6%)36.5 (+12.4%)27.4 (+21.4%)2.02 (+26.3%)

The two-stage model is worse than never doing predictive training at all. This is a useful negative result: the benefit is not merely “better pretraining.” It comes from continuously forcing the representation used for answer generation to also carry predictive visual structure.

3.5 Shared decoder or separate experts?

We study two ways to couple the objectives:

  • Dense: understanding tokens and predictive tokens update the same decoder parameters.
  • MoE: the two token types use separate feed-forward experts, selected deterministically by token type.
Dense and MoE architectures across prediction horizons.
Figure 5. Dense sharing is strongest at short horizons; MoE is flatter as the horizon grows.

Dense wins most clearly when the future target is short and relatively constrained. At H=2, for example, Dense reaches ROUGE-L 37.4 versus 33.8 for MoE. But Dense degrades as more future frames are predicted, while MoE is less sensitive and tends to peak around H=3.

The key takeaways are:

A shared decoder allows gradients from next-embedding prediction to directly shape the representations used for answer generation, which is consistent with its clear advantage at H=1 and H=2. As the horizon increases, however, future prediction becomes more ambiguous, and tightly coupling this objective to the language decoder may introduce less task-aligned supervision. Separating the objectives through MoE appears to reduce this sensitivity, although it does not match the strongest short-horizon Dense configuration.

4. Main Results: Quality Without the Inference Rollout

We use the same BAGEL-7B-based initialization and matching training setup for Answer-Only SFT, Text CoT, Visual CoT, and IVT. At inference, IVT follows the same direct observed-video → text path as Answer-Only SFT. Across the three datasets and two tasks, IVT improves all four metrics over Answer-Only SFT in every setting.

Dataset / taskAnswer-Only SFT
ROUGE-L
Visual CoT
ROUGE-L
Best IVT
ROUGE-L
Ego-Exo4D · Early event28.832.733.3
Ego4D · Early event33.336.134.8
EPIC-KITCHENS-100 · Early event36.540.239.0
Ego-Exo4D · Next event46.342.648.5
Ego4D · Next event51.952.252.7
EPIC-KITCHENS-100 · Next event45.542.849.1

The pattern is revealing. Explicit Visual CoT remains strong for two early-event settings, where the action is already unfolding and generating a near-future frame may be easier. But on all three next-event datasets—where the target event has not begun—IVT is stronger.

Across the three next-event benchmarks, we report average ROUGE-L increasing from 45.9 for Visual CoT to 49.7 for IVT.

Inference efficiency

The quality result is only half the story. IVT reformulates visual foresight from an explicit inference-time operation into the model’s internal representations.

Answer-Only SFT
1.20 s
Visual CoT
6.56 s
IVT
1.22 s

Measured end-to-end on a single NVIDIA B200 at batch size 1 and bf16, Visual CoT averages 6.56 seconds per sample across the six settings, versus 1.20 seconds for Answer-Only SFT and 1.22 seconds for IVT. The mean P95 latency is 7.92 s, 1.92 s, and 1.77 s, respectively.

Quality–efficiency frontier. IVT is not simply a cheaper approximation to Visual CoT. In four of six settings it is also more accurate, because the model is not forced to condition on a possibly wrong self-generated future frame.

5. Limitations and Open Questions

Short-horizon focus. Our experiments intentionally emphasize near-future anticipation, where local motion and object interaction provide useful predictive signal. Long-horizon futures are more multimodal and depend more heavily on goals, procedures, and language priors.

5.1 One future target versus many plausible futures

Directly regressing toward one observed future frame is well behaved when the horizon is short. As the horizon grows, the conditional future distribution becomes multimodal. A deterministic feature target may average incompatible futures, while flow matching or other distributional objectives may become more important.

5.2 What should be internalized?

We study frame-level latent targets. Other possibilities include object-centric state, depth, optical flow, contact maps, segmentation, action-conditioned state transitions, or learned dynamics tokens. The choice of target representation appears to matter substantially, and there is still no clear answer to what information is most useful to internalize for proactive reasoning.

We also experimented with V-JEPA 2.1 features as the predictive target, motivated by their strong video-centric representation learning objective. In our preliminary experiments, however, we did not find V-JEPA 2.1 targets to provide a clear improvement for these proactive reasoning tasks. This negative result reinforces an important point: a representation that is strong for video understanding or predictive pretraining is not automatically the best supervision target for transferring future information into an MLLM's reasoning pathway.

5.3 Does the gain survive stronger base models?

Our controlled comparisons are intentionally centered on matched BAGEL-based systems. We also report external 7B–8B baselines, but those differ in pretraining, backbone, and optimization. A natural next step is to test whether internalized predictive supervision scales with much stronger multimodal foundation models.

5.4 Can latent prediction support test-time reasoning too?

IVT deliberately removes the prediction branch at inference. That is its efficiency advantage. But there may be a continuum between “no rollout” and “full pixel rollout”: a model could perform a small number of compressed latent-state predictions or updates only on hard examples, instead of generating full-resolution future images. Such compressed latent reasoning could provide adaptive test-time compute while retaining much of IVT's efficiency advantage and avoiding expensive image synthesis and re-encoding.

5.5 Evaluation of open-ended futures

Top-5 decoding partially acknowledges future uncertainty, but lexical metrics still compare generated descriptions to finite annotations. More fundamentally, we found that rigorous evaluation of multiple plausible futures is itself difficult. Given the same partial observation, different human annotators can anticipate different next events, and even stronger LLM-based judges may produce different yet reasonable expectations about what happens next.

This makes a single-reference notion of correctness inherently incomplete. Better evaluations may need semantic sets of valid futures, calibrated uncertainty, multiple independent annotations, and decision-oriented metrics that ask whether a prediction is plausible and useful rather than merely textually similar to one annotated continuation.

Longer-horizon anticipation is substantially harder

We also evaluate a more challenging setting in which the model must anticipate an action three hops into the future (hop = 3). In this regime, we do not observe a clear benefit from IVT over the corresponding answer-only baseline.

This result suggests that IVT is most effective for short-horizon anticipation, where future events remain strongly grounded in the current visual evidence—for example, ongoing motion, hand–object interactions, object state, and local scene dynamics. As the prediction horizon grows, future actions depend increasingly on latent goals, higher-level plans, and multiple branching possibilities that may not be recoverable from visual input alone.

In other words, internalizing near-future visual dynamics appears useful when the future is still tightly coupled to what the model can currently see. For more distant events, stronger modeling of intent, procedure, memory, or multimodal world knowledge may be required in addition to visual prediction.

6. Takeaways

  1. Future visual context is valuable; generated pixels are the bottleneck. The oracle-frame experiment makes this distinction explicit.
  2. Predicting a future representation can improve reasoning even when that representation is never produced at inference.
  3. Predictive supervision must remain coupled to language learning. Sparse predictive updates and sequential “predict then fine-tune” curricula weaken or erase the benefit.
  4. Parameter sharing is a bias. Dense sharing transfers short-horizon predictive gradients strongly; MoE isolation becomes more robust as future uncertainty increases.
  5. Reasoning need not always be externalized. Some of the value of visual chain-of-thought can be moved from inference-time generation into training-time representation learning.

References

  1. Zhu, X. et al. (2026). Beyond Visual CoT: Internalized Visual Thinking for Proactive Video Reasoning. arXiv:2608.15869.
  2. Wei, J. et al. (2022). Chain-of-Thought Prompting Elicits Reasoning in Large Language Models.
  3. Zhao, H. & Wildes, R. (2021). Review of video predictive understanding: Early action recognition and future action prediction.
  4. Deng, X. et al. (2025). BAGEL: a unified multimodal model used as the controlled backbone family in the IVT study.
  5. Black Forest Labs (2024). FLUX latent autoencoder representation, used as the reconstruction-oriented future target.
  6. Oquab, M. et al. (2023). DINOv2: Learning Robust Visual Features without Supervision.
  7. Tschannen, M. et al. (2025). SigLIP2 visual representations.
  8. For full benchmark, baseline, and metric citations, see the bibliography in our paper.
Contact. If you're interested in discussing the research, feel free to reach out at zgzxy001@gmail.com.