Text-to-video diffusion models have enabled open-ended video synthesis, but often struggle with generating the correct number of objects specified in a prompt. We introduce NUMINA , a training-free identify-then-guide framework for improved numerical alignment. NUMINA identifies prompt-layout inconsistencies by selecting discriminative self- and cross-attention heads to derive a countable latent layout. It then refines this layout conservatively and modulates cross-attention to guide regeneration. On the introduced CountBench, NUMINA improves counting accuracy by up to 7.4% on Wan2.1-1.3B, and by 4.9% and 5.5% on 5B and 14B models, respectively. Furthermore, CLIP alignment is improved while maintaining temporal consistency. These results demonstrate that structural guidance complements seed search and prompt enhancement, offering a practical path toward count-accurate text-to-video diffusion. The code is available at https://github.com/H-EmbodVis/NUMINA.
Figures & tables
Figure 1: We present NUMINA , a training-free framework that alleviates the misalignment between precise numerals and visual instances in text-to-video diffusion models. We significantly improve counting accuracy while maintaining natural layouts and temporal coherence.
Figure 2: Visualization of the cross-attention maps corresponding to different texts in the prompt. The highlighted areas represent a stronger level of attention between the pixels and the text.
Figure 3: The pipeline of our NUMINA follows a two-phase paradigm. Given a text prompt containing numerals, we first perform the numerical misalignment identification to extract explicitly countable layouts from attention maps. Based on the layout, we further conduct a refinement and a layout-guided generation for the numerically aligned video generation.
Figure 4: The PCA visualization of self-attention maps for Wan2.1-1.3B. (a) Different attention heads naturally capture diverse spatial patterns. (b) We select the head with the highest instance separability for countable layout construction.
Models
CountAcc (%)
TC (%)
CLIP Score
Wan2.1-1.3B [ 59 ] (81 frames, 832 × 480)
Wan2.1-1.3B
42.3
81.2
33.9
+ Seed search
45.5( +3.2 )
82.3( +1.1 )
34.6( +0.7 )
+ Prompt enhancement
47.2( +4.9 )
82.1( +0.9 )
33.7( -0.2 )
+ NUMINA ( ours )
49.7 ( +7.4 )
83.4 ( +2.2 )
35.6 ( +1.7 )
Wan2.2-5B [ 59 ] (81 frames, 1280 × 704)
Table 1: Comparison of NUMINA with other practical strategies. We report Counting Accuracy (CountAcc), Temporal Consistency (TC), and CLIP Score on Wan [ 59 ] of varying scales.
Figure 5: Qualitative comparison of NUMINA with the most advanced commercial models.
Figure 6: The per-numeral accuracies for Wan2.1-1.3B.
Figure 7: Ablation on the reference timesteps t⋆ for head selection.
Method
CountAcc (%)
TC (%)
Baseline
42.3
81.2
GroundingDINO
47.5( +5.2 )
82.8( +1.6 )
Attention ( ours )
49.7 ( +7.4 )
83.4 ( +2.2 )
Table 2: Ablation on the layout construction method.
Co
Cc
Ct
CountAcc (%)
TC (%)
Baseline
42.3
81.2
✓
45.1( +2.8 )
82.1( +0.9 )
✓
✓
46.9( +4.6 )
82.3( +1.1 )
✓
✓
48.9( +6.6 )
83.1( +1.9 )
✓
✓
✓
49.7 ( +7.4 )
83.4 ( +2.2 )
Table 3: Ablation on the components of the layout refinement cost.
Method
CountAcc (%)
TC (%)
Baseline
42.3
81.2
Random
44.1( +1.8 )
82.6( +1.4 )
All average
43.0( +0.7 )
82.4( +1.2 )
Top-3
48.2( +5.9 )
82.5( +1.3 )
Top-2
49.4( +7.1 )
83.3( +2.1 )
Top-1
49.7 ( +7.4 )
83.4 ( +2.2 )
Table 4: Ablation on the self-attention head selection strategy.
Text-to-image diffusion models achieve high visual fidelity but surprisingly exhibit systematic failures in numerical control when prompts specify explicit object counts. To address this limitation, we introduce ATHENA, a model-agnostic, test-time adaptive steering framework that improves object count fidelity without modifying model architectures or requiring retraining. ATHENA leverages intermediate representations during sampling to estimate object counts and applies count-aware noise corrections early in the denoising process, steering the generation trajectory before structural errors become difficult to revise. We present three progressively more advanced variants of ATHENA that trade additional computation for improved numerical accuracy, ranging from static prompt-based steering to dynamically adjusted count-aware control. Experiments on established benchmarks and a new visually and semantically complex dataset show that ATHENA consistently improves count fidelity, particularly at higher target counts, while maintaining favorable accuracy-runtime trade-offs across multiple diffusion backbones.
Mohammad Shahab Sepehri, Asal Mehradfar, Berk Tinaz +2
Department of Electrical and Computer Engineering, University of Southern California, Los Angeles, CA, USA
Text-to-video diffusion models generate realistic videos, but often fail on prompts requiring fine-grained compositional understanding, such as relations between entities, attributes, actions, and motion directions. We hypothesize that these failures need not be addressed by retraining the generator, but can instead be mitigated by steering the denoising process using the model's own internal grounding signals. We propose CVG, an inference-time guidance method for improving compositional faithfulness in frozen text-to-video models. Our key observation is that cross-attention maps already encode how prompt concepts are grounded across space and time. We train a lightweight compositional classifier on these attention features and use its gradients during early denoising steps to steer the latent trajectory toward the desired composition. Built on a frozen VLM backbone, the classifier transfers across semantically related composition labels rather than relying only on narrow category-specific features. CVG improves compositional generation without modifying the model architecture, fine-tuning the generator, or requiring layouts, boxes, or other user-supplied controls. Experiments on compositional text-to-video benchmarks show improved prompt faithfulness while preserving the visual quality of the underlying generator.
Ariel Shaulov, Eitan Shaar, Amit Edenzon +2
1Tel-Aviv University · 2Independent Researcher · 3Bar Ilan University +1
Text-to-video diffusion models generate temporally coherent content from natural language, yet when a prompt describes an early scene that persists while a new event emerges on top of it---such as "a tall sandcastle standing on a beach where a wave rushes in and washes it away"---generation frequently fails to realize the late-segment event in the corresponding frames. We identify this failure as Temporal Prior Suppression (TPS): the dominant prior of the early segment captures the cross-attention trajectory across the temporal axis and suppresses the guidance signal needed for late-segment realization, a competing tendency existing guidance mechanisms do not model. We introduce Temporal Prior Decoupling (TPD), a training-free framework that restores suppressed late-segment signals during diffusion sampling. TPD constructs a temporal counterfactual by conditioning on the early segment alone, and defines the discrepancy between the full-prompt and counterfactual trajectories as a suppressed signal direction. Rather than removing this direction as in prior subtractive projection methods, TPD restores it through a frame-selective lower-bound constraint resolved jointly over diffusion timestep and video frame, realizing the suppressed event in the late frames without disrupting early-segment coherence: where prior work enforces upper-bound feasibility to remove unwanted semantics, TPD enforces lower-bound feasibility to guarantee suppressed-signal contribution. TPD runs entirely within standard diffusion sampling without retraining, and is defined purely in classifier-free guidance space, making it backbone-agnostic by construction. Experiments show that TPD significantly improves late-concept realization while preserving temporal coherence and visual fidelity, and that the targeted suppression recurs across distinct text-to-video backbones.
Taewon Kang, Matthias Zwicker
University of Maryland at College Park, United States