Video diffusion models generate visually compelling short-term sequences but often struggle to preserve scene geometry and object structure over time. In this paper, we first demonstrate that current video attention cannot identify reliable correspondences across frames, with responses dispersed across multiple regions even after recognizable scene structure has emerged. We argue that reliable cross-frame associations are essential for information exchange as scene structure forms in video generation. Thus, we introduce Geometry Attention Distillation (GAD), which transfers correspondences from a frozen geometry model's attention to video attention. Extracted from clean videos, these correspondences link the same static surface or moving object across frames. Since structure formation begins in deeper layers at higher noise and shallower layers at lower noise, we apply attention distillation at a student layer selected according to the noise timestep. After training, video attention focuses more consistently on corresponding locations. Experiments across all three video backbones demonstrate improved long-range scene consistency and dynamic temporal consistency, with geometric gains across generation lengths. On Wan2.2 5B, GAD improves cross-frame matching accuracy by at least 30% over the Wan baseline on both long-range benchmarks. Notably, GAD achieves these gains without additional inference modules, offering a promising approach to efficient and consistent video-based world simulation. Project page: https://fantasy-amap.github.io/GAD/.
Figures & tables
Figure 1: Geometry attention distillation for consistent video generation. Correspondences from a geometry teacher guide video attention during training (left), improving scene geometry under viewpoint changes and object structure during motion (right), without added inference cost.
Figure 2: Correspondence patterns in video generation and geometry foundation models. (a) In the structure-formation region (red boxes), recognizable scene structure and cross-frame associations begin to emerge, at deeper layers under higher noise and shallower layers under lower noise. However, attention remains ambiguous even after structure becomes clear. (b) Selected VGGT- Ω heads track the same static surface or moving object across frames.
Figure 3: Overview of GAD . Correspondences extracted by a frozen geometry teacher from clean videos guide the video model’s attention during denoising. The correspondence supervision complements the generation objective during training, without changing the inference procedure.
Method
RealEstate10K
DL3DV-10K
FID ↓
FVD ↓
CLIP ↑
Rdist↓
Tdist↓
Mrate↑
Mdist↓
FID ↓
FVD ↓
CLIP ↑
Rdist↓
Tdist↓
Mrate↑
Mdist↓
GF
61.46
464.8
30.03
6.229
0.134
0.188
55.66
104.1
813.7
29.42
8.145
0.126
0.209
54.50
Gen3R
33.35
305.3
30.88
5.167
0.103
0.321
36.52
45.23
373.7
32.66
6.457
0.088
0.310
39.27
Voyager
58.38
410.3
30.47
4.779
0.099
0.334
33.01
65.77
608.4
32.16
4.740
0.073
0.313
41.02
Gen3C
31.07
264.7
31.11
4.288
0.076
0.315
35.07
38.63
325.6
33.46
4.732
0.056
0.289
42.19
Uni3C
34.46
289.6
31.32
5.146
0.096
0.384
32.11
43.28
351.1
33.30
5.589
0.078
0.331
38.42
Table 1: Long-range camera-controlled video generation on RealEstate10K and DL3DV-10K. GAD improves all seven metrics over the Wan baselines across all three backbones. Top-3 results are highlighted as the first , second and third .
Method
Rendering
VBench Metrics
Inference Cost
FID ↓
FVD ↓
CLIP ↑
Subj. Cons. ↑
Bg. Cons. ↑
Motion Sm. ↑
Aesth. Q. ↑
Image Q. ↑
Time (s) ↓
Peak Mem. (GB) ↓
Avg. Mem. (GB) ↓
GF
162.0
1543
20.48
0.698
0.851
0.961
0.397
0.380
194.6
48.96
33.80
Gen3R
81.91
870.1
26.84
0.817
0.907
0.957
0.482
0.505
209.3
50.66
37.05
Voyager
103.6
987.7
26.09
0.805
0.906
0.963
0.492
0.521
2583.2
67.45
62.27
Gen3C
70.41
774.8
27.55
0.810
0.908
0.970
0.493
0.527
825.9
63.71
57.20
Uni3C
80.63
856.6
27.56
0.804
0.908
0.968
0.504
0.553
444.0
49.77
37.68
Table 2: Dynamic camera-controlled video generation on OmniWorld-Game. GAD improves subject and background consistency and motion smoothness, without added inference cost.
Table 6
Figure 4: Qualitative comparison of camera-controlled video generation. GAD better preserves scene geometry over 15 seconds (left) and moving-object structure over 5 seconds (right).
CAT
TWS
PWS
FID ↓
FVD ↓
CLIP ↑
Rdist↓
Tdist↓
Mrate↑
Mdist↓
×
×
×
38.60
322.1
33.85
9.573
0.108
0.257
48.90
×
✓
✓
39.04
329.4
33.92
8.753
0.092
0.230
42.94
✓
✓
×
38.61
321.4
34.01
6.028
0.082
0.274
40.20
✓
×
✓
38.35
320.3
34.05
5.312
0.076
0.291
39.67
✓
✓
✓
37.93
318.0
34.24
4.890
0.065
0.335
38.85
Table 5: Distillation component ablation. Confidence-aware teacher weighting (CAT) combined with trackwise supervision (TWS) and pairwise supervision (PWS) gives the best results across all seven metrics.
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
Exp. ID
Loss Assignment
FID ↓
FVD ↓
CLIP ↑
Rdist↓
Tdist↓
Mrate↑
Mdist↓
R1 (Wan)
Off @ all t
38.60
322.1
33.85
9.573
0.108
0.257
48.90
R2
L5 @ all t
38.03
321.0
34.24
7.440
0.086
0.295
43.89
R3
L15 @ all t
38.18
319.1
34.12
7.346
0.090
0.286
42.71
R4
L25 @ all t
38.07
320.3
34.17
6.540
0.075
0.289
43.55
R5
L5 @ t≥900
38.64
323.9
34.09
8.932
0.098
0.262
46.36
R6
L25 @ t<900
38.80
322.9
34.03
9.290
0.106
0.258
47.17
Appendix
Table 6: Complete noise-timestep routing results on DL3DV-10K with Wan2.2 5B. R1–R8 follow Table 4 ; R8 is GAD . Layer sequences run from low to high noise; values after @ are model-timestep boundaries, assigned to the higher-noise interval. For inequalities, supervision is off outside the stated interval. Best results are bold.
Duration
FID ↓
FVD ↓
CLIP ↑
Rdist↓
Tdist↓
Mrate↑
Mdist↓
5s
26.85
285.7
34.22
3.020
0.064
0.529
21.84
5s + GAD
27.63
285.1
34.26
1.684
0.053
0.571
19.31
10s
33.68
287.4
34.03
5.948
0.072
0.371
38.34
10s + GAD
33.49
288.4
34.14
3.376
0.056
0.453
33.33
15s
38.60
322.1
33.85
9.573
0.108
0.257
48.90
15s + GAD
37.93
318.0
34.24
4.890
0.065
0.335
38.85
Appendix
Table 7: Complete generation-duration results on DL3DV-10K with Wan2.2 5B. Separately trained Wan/ GAD pairs show geometric gains at every duration, with small FID or FVD trade-offs at 5 and 10 seconds. Best results within each duration are bold.
Figure 5: Video attention and generation results with geometry attention distillation. Compared with the Wan baseline, GAD produces more localized attention responses around corresponding regions and better preserves scene and object structure in static (top) and dynamic (bottom) examples. Red dots mark source queries in the input frame; heatmaps show their attention responses in subsequent target views. GT denotes reference frames.
Figure 6: VGGT- Ω attention and correspondence reliability. Static (top) and dynamic (bottom) examples show that ambiguous or incorrect responses can have low attention mass despite visible peaks, supporting its use as a reliability cue. Red dots mark source queries; heatmaps show target-view responses, with the total attention mass assigned to each target frame reported below.
Figure 7: More 15-second camera-controlled generation results in static scenes.
Figure 8: More 5-second camera-controlled generation results in dynamic scenes.
Video diffusion models lack explicit geometric supervision during training, leading to inconsistency artifacts such as object deformation, spatial drift, and depth violations in generated videos. To address this limitation, we propose a geometry-based reward model that leverages pretrained geometric foundation models to evaluate multi-view consistency through cross-frame reprojection error. Unlike previous geometric metrics that measure inconsistency in pixel space, where pixel intensity may introduce additional noise, our approach conducts error computation in a pointwise fashion, yielding a more physically grounded and robust error metric. Furthermore, we introduce a geometry-aware sampling strategy that filters out low-texture and non-semantic regions, focusing evaluation on geometrically meaningful areas with reliable correspondences to improve robustness. We apply this reward model to align video diffusion models through two complementary pathways: post-training of a bidirectional model via SFT or Reinforcement Learning and inference-time optimization of a Causal Video Model (e.g., Streaming video generator) via test-time scaling with our reward as a path verifier. Experimental results validate the effectiveness of our design, demonstrating that our geometry-based reward provides superior robustness compared to other variants. By enabling efficient inference-time scaling, our method offers a practical solution for enhancing open-source video models without requiring extensive computational resources for retraining.
Tengjiao Yin, Jinglei Shi, Heng Guo +1
VCIP & TMCC & DISSec, College of Computer Science, Nankai University · Beijing University of Posts and Telecommunications · LIX, École Polytechnique, IP Paris
Generating geometrically consistent videos remains an open challenge: text-to-video diffusion models trained on web-scale data treat geometry only implicitly, leading to object deformation, texture drift, and non-rigid backgrounds under camera motion. Existing solutions either improve consistency as a byproduct, apply only to static scenes or realign the latent space of the model completely. We introduce a geometry-consistency reward that directly measures whether motion in a generated video is compatible with a coherent scene. Our key insight is that in physically consistent videos, background motion should be explainable by rigid camera-induced flow, while independently moving objects should preserve appearance identity along motion trajectories. We operationalize this using optical flow, depth--pose predictions, and feature-based correspondence to separate rigid and dynamic regions and evaluate their respective consistency. Integrating this reward with reinforcement fine-tuning transforms geometric consistency from an emergent property into an explicit optimization objective for video generators. The approach is model agnostic and applies to diverse dynamic scenes containing both camera and object motion. Experiments show substantial reductions in temporal geometric artifacts over strong baselines while preserving perceptual quality. Code and model weights are published.
Large-scale video diffusion models often fail to preserve 3D structure over time, causing geometric drift and implausible motion under viewpoint changes. Existing methods usually enforce geometric consistency by using explicit geometry reconstructions, such as depth maps, point clouds, or reconstructed 3D structures, to define conditions, supervision, or reward signals, making the generator sensitive to errors from upstream geometry pipelines. We propose VideoWeave, a latent-space post-training framework that uses implicit geometry-model features to constrain the generative distribution, providing a more flexible and non-rigid form of guidance that mitigates the impact of reconstruction errors from geometry models. Specifically, VideoWeave adapts these features into geometry latents and jointly models them with video latents in a shared denoising space, allowing geometry to shape the generative distribution during training. To support this process, we build GeoVid-80K, an 80K-video dataset with paired appearance and geometry representations. Experiments on text-to-video and image-to-video generation show that VideoWeave improves geometric coherence while preserving strong visual quality. VideoWeave project page at https://videoweave.github.io/
Xunzhi Xiang, Zixuan Duan, Yabo Chen +8
1Nanjing University · Institute of Artificial Intelligence, China Telecom (TeleAI) · 3Chinese University of Hong Kong, Shenzhen