Paper Overview
- Field: Computer Vision (CV)
- Authors: Zhengyang Sun, Yu Chen, Xin Zhou
- Published: 2025-04-10
- arXiv: 2504.07941
- Problem: Text-to-video diffusion models frequently miscount objects specified in prompts.
- Approach: Training-free framework combining discriminative attention-head selection, countable latent layout estimation, conservative layout refinement, and cross-attention modulation.
- Results: Counting accuracy gains of 7.4% (Wan2.1-1.3B), 4.9% (5B), and 5.5% (14B) on CountBench, with improved CLIP alignment and preserved temporal consistency.
Original Abstract
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. The authors 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, providing a practical path toward count-accurate text-to-video diffusion.