Trajectory Forcing Paper

NeurIPS 2026

Trajectory Forcing: Exploiting Diffusion Trajectories for Autoregressive Long Video Generation

TraF · 30-second generationPigeon
Original speed 0:00 / 0:30

Selected qualitative examples. Compare this scene

01Idea

Denoising trajectories are context for what comes next.

Trajectory Forcing turns intermediate denoising predictions into useful context for generating subsequent video chunks, across supervised training and self-rollout training.

Paper figure. (a) Standard autoregression: tokens 1, 2 and 3 form the history used to predict token 4. (b) Autoregressive diffusion: each chunk is generated from noise to clean through several steps, so chunks 1 to 3 each carry a trajectory that conditions the prediction of chunk 4.
(a) Tokens are generated as single units. (b) Video chunks leave a history of denoising predictions.
Denoising trajectories as context (conceptual view) Conceptual view, not sampled data or an execution trace. Three columns are video chunks in order. Each column is a denoising trajectory from Gaussian noise at the top to the chunk's final output at the bottom; its points darken as they become cleaner. Chunks 1 and 2 are complete. Chunk 3 starts from Gaussian noise, which is sampled without conditioning, and its current denoising step is marked by a ring. The shaded context of this step holds the complete trajectories of chunks 1 and 2, including their points below the current step, and chunk 3's own earlier states; chunk 3's later steps are outside it. One arrow leads from this context to the current step. With Play, the shading first covers only the endpoints of chunks 1 and 2 and then extends to their full trajectories; chunk 3's denoising updates then appear one at a time, each drawn with this context and arrow. x1 x2 x3 noise clean endpoints only full trajectories context for this step x3,t
Conceptual view Endpoint-only conditioning uses each earlier chunk’s final output; trajectory conditioning also draws on intermediate predictions.

Each column is one chunk’s denoising trajectory, from Gaussian noise at the top to its final output at the bottom.

In this conceptual view, the shaded history includes complete earlier trajectories and the current chunk’s earlier states. The arrow shows how this history conditions the current update. The current chunk’s initial noise is sampled without conditioning.

In practice, each update uses one selected clean-signal prediction from each retained preceding chunk. Selection details.

02METHOD

From trajectory learning to practical conditioning

Ground-truth videos have no model-generated denoising history, so TraF learns in two stages: pseudo-trajectories first, then trajectories from its own rollouts.

Stage 1 · SL

Pseudo-trajectories from ground truth

Source
Ground-truth videos
Context
At a sampled noise level, one velocity prediction per chunk gives a clean-signal estimate. Sampling different noise levels across training yields the pseudo-trajectory points.
Objective
Flow matching
Algorithm 1 pseudo-code from the paper Algorithm 1 from the paper, SL: training on ground truth. For each chunk, sample a noise level, interpolate with Gaussian noise and use one velocity prediction to estimate the clean signal without gradients. Then train the model with flow matching to predict velocities for freshly noised chunks using those estimates as context.

Stage 2 · RL

Real trajectories from self-rollout

Source
Initialized from the SL model, the student generates chunks autoregressively with a four-step self-rollout
Context
Its four denoising predictions form each chunk’s real trajectory; selected predictions condition later chunks
Objective
Distribution Matching Distillation (DMD) with a frozen teacher and a learned critic
Algorithm 2 pseudo-code from the paper Algorithm 2 from the paper, RL: training on self-rollouts. For each chunk, run the denoising schedule; at each step select trajectory context from preceding chunks, predict the clean signal, and cache it in the trajectory; then compute the DMD loss and update the student.

Selection · practical conditioning

Using trajectory context efficiently

Full-trajectory conditioning adds attention and memory cost. TraF uses one selected prediction per retained preceding chunk at each update.

Paper figure. Left: clean-signal estimates predicted at successive denoising steps of a preceding chunk. Right: the current chunk being denoised, which receives one of the preceding chunk's intermediate estimates as context. Arrows illustrate the selection options shown in the paper.
A preceding chunk’s trajectory supplies context to the current update. Arrows are illustrative, not a timestep mapping.
Cache
Clean-signal predictions from preceding chunks, not noisy latents.
Context
One selected prediction from each retained preceding chunk at each denoising update.
Rule
On the finite denoising grid, select the available prediction nearest to the regularized target progress.
Selection rule continuous target & discrete steps

The continuous target scales the current step’s amplitude signal-to-noise ratio by γ.

Sγ(p) = γp1 + (γ − 1)p

p = 1 − σ is denoising progress, from noise (0) to clean signal (1); γ ≥ 1.

On the finite grid, TraF adds a regularization term to the target ratio and converts it to progress before selecting the nearest available prediction. See Appendix A of the paper.

γ = 1 is step-aligned. γ → ∞ approaches endpoint-only conditioning.

03Results

Trajectory context for long video generation

We first examine 30-second generation, then generation at TraF’s five-second training duration. These comparisons from the paper span different model and training configurations.

Indigo row markers identify Self Forcing, Causal Forcing and TraF. See their training and inference settings.

Two evaluation settings. Different prompt sets and scoring protocols: the 5-second and 30-second scores are not directly comparable.

30-second generation

6× TraF’s training duration

We evaluated each method on the same 128 MovieGenBench prompts. Nine VBench dimensions are grouped into Temporal Quality, Frame Quality and Text Alignment; Total is the equal-weighted average of these three axes.

Scatter plot of Frame Quality and Temporal Quality for six methods in the 30-second evaluation: CausVid, Causal Forcing, Self Forcing, Trajectory Forcing, Rolling Forcing and LongLive. Hollow markers identify Rolling Forcing and LongLive, which use long-video-specific training.
30-second evaluation. Frame Quality and Temporal Quality are higher-is-better. Labels show Frame / Temporal scores. Axes are zoomed; hollow markers denote long-video-specific training.
30-second VBench axes, total score, VideoScore2 and Quality Drift for six selected methods, with long-video-specific training shown as a separate group.
MethodTotal ↑Temporal
Quality ↑
Frame
Quality ↑
Text
Alignment ↑
VideoScore2 ↑Quality
Drift ↓
Main comparisons
CausVid1.3B · Wan-14B teacher68.5083.9762.6658.883.5492.72
Causal Forcing1.3B · Wan-14B teacher70.4889.2659.5862.613.5894.36
Self Forcing1.3B · Wan-14B teacher71.0685.3561.4966.333.7192.21
Trajectory Forcing Ours1.3B · Wan-14B teacher73.9890.3965.4666.083.7711.38
Long-video-specific training
Rolling Forcing1.3B · Wan-14B teacher71.3982.0465.3666.773.8040.56
LongLive1.3B · Wan-14B teacher71.4881.3366.0767.043.8010.60

Rolling Forcing and LongLive use long-video-specific training and their own inference architectures. SF, CF and TraF use the same sliding-window inference with an attention sink in this evaluation.

TraF’s 30-second Total is 2.92 points higher than Self Forcing (95% paired prompt-bootstrap CI: +2.06 to +3.78).

Quality Drift is the absolute imaging-quality difference between the first and last 5-second segments; lower is better. VideoScore2 reports its text-alignment sub-score. All other columns are higher-is-better.

Bold marks the best value in each column across all displayed methods. TraF leads Total and Temporal Quality; LongLive leads Frame Quality and Text Alignment; Rolling Forcing leads VideoScore2 and has the lowest Quality Drift. See the paper for all methods.

5-second generation

VBench · short video

946 prompts, all 16 VBench dimensions. Total = (4 × Quality + Semantic) / 5. Throughput and latency are reported in FPS and seconds, respectively.

5-second VBench scores and inference speed for twelve selected methods across three model categories.
MethodTotal ↑Quality ↑Semantic ↑Throughput ↑FPSLatency ↓seconds
Bidirectional Video Diffusion
LTX-Video1.9B · Teacher —79.8381.8871.628.9813.5
Wan2.1-1.3B1.3B · Teacher —83.3784.3079.650.78103
Autoregressive Video Diffusion
MAGI-14.5B · Teacher —78.8881.6767.720.19282
NOVA0.6B · Teacher —80.3180.6678.920.884.1
Pyramid Flow2B · Teacher —80.7583.4170.116.702.5
SkyReels-V21.3B · Teacher —81.9783.9674.010.49112
Distilled Autoregressive Video Diffusion
CausVid1.3B · Wan-14B teacher81.3383.9870.7217.00.69
Rolling Forcing1.3B · Wan-14B teacher83.0783.6180.8915.80.76
LongLive1.3B · Wan-14B teacher83.0983.6980.6918.00.65
Causal Forcing1.3B · Wan-14B teacher84.0484.5981.8417.00.69
Self Forcing1.3B · Wan-14B teacher84.3185.0781.2817.00.69
Trajectory Forcing Ours1.3B · Wan-14B teacher85.1585.8982.2015.00.78

We evaluated TraF, Rolling Forcing and LongLive. The Self Forcing row comes from its original paper. The other rows shown here come from the Causal Forcing paper. See the full table and sources in the paper.

Bold marks the best reported value in each column. Among the methods shown, TraF leads on Total, Quality and Semantic. LongLive has the highest reported throughput and lowest reported latency.

SF / CF / TraF setup

Base model
Wan2.1-T2V-1.3B
Inference
4 steps
Resolution
832×480
Video frame rate
16 fps

Self Forcing, Causal Forcing and Trajectory Forcing use the same backbone architecture and teacher during RL. TraF starts from its SL checkpoint, while SF and CF use their original initializations. In the 30-second evaluation, all three use the same sliding-window inference with an attention sink.

The following three ablations examine the information in trajectory context, learning before distillation, and the choice of conditioning predictions.

Evidence · 5-second VBench

Testing the information in trajectory context

Adding Gaussian noise to endpoint predictions lowers Total at both tested magnitudes. The unperturbed endpoint scores 84.24.

VBench Total ↑ higher is better

Gaussian
σ = 0.22
83.87
Gaussian
σ = 0.11
83.95
Trajectory
predictions
85.15
View all three VBench metrics

5-second VBench ablation. Endpoint predictions with added isotropic Gaussian noise serve as the context for the two noise controls. All three metrics are higher-is-better.

5-second VBench noise versus trajectory ablation.
ContextTotal ↑Quality ↑Semantic ↑
Noisy endpoint, σ = 0.2283.8784.7680.29
Noisy endpoint, σ = 0.1183.9584.9280.08
Trajectory85.1585.8982.20

Training evidence

SL stage, before distillation

Same architecture and data, evaluated before distillation with 50-step denoising on 5-second VBench. All three metrics are higher-is-better.

SL-stage ablation before distillation, evaluated with 50-step denoising.
SL strategyTotal ↑Quality ↑Semantic ↑
Teacher Forcing83.4784.0981.02
Diffusion Forcing84.0784.6681.68
Trajectory84.3985.2580.96

Trajectory has the highest Total and Quality; Diffusion Forcing has the highest Semantic score. These SL-stage scores use a different evaluation stage and denoising schedule from the final four-step model.

Conditioning evidence

Choosing trajectory context

5-second VBench full-pipeline ablation. Each γ setting uses a separately trained and evaluated model. See Figure 2 of the paper. All three metrics are higher-is-better.

5-second VBench full-pipeline selection ablation from Figure 2.
γTotal ↑Quality ↑Semantic ↑
184.8285.4982.10
285.1585.8982.20
385.1385.9181.97
584.5885.4781.03
1084.6285.4081.50
∞84.2485.0481.07

Every finite γ reported here has a higher Total than endpoint-only conditioning (γ → ∞). γ = 2 and 3 give the highest Total scores. Bold marks the best value in each column.

04Videos

30-second comparisons

Selected qualitative examples from SF, CF and TraF, played together at the original speed.

0:00 / 0:30

Prompt

The comparison plays the first 30 seconds of each video at original speed. “Full video” opens the complete file. Comparisons with other methods are in the paper.

05Cite

BibTeX

@inproceedings{wang2026trajectoryforcing,
  title     = {Trajectory Forcing: Exploiting Diffusion Trajectories for
               Autoregressive Long Video Generation},
  author    = {Wang, Bohan and Chen, Shuo and Li, Yunzhi and
               Chen, Zhaozheng and Zhang, Junzhe and Zhang, Hanwang},
  booktitle = {Advances in Neural Information Processing Systems},
  year      = {2026}
}