I’m a Map! Interpretable Motion-Attentive Maps: Spatio-Temporally Localizing Concepts in Video Diffusion Transformers
Interpretable Motion-Attentive Maps: Spatio-Temporally Localizing Concepts in Video Diffusion Transformers
IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) 2026
Youngjun Jun, Seil Kang, Woojung Han, Seong Jae Hwang
Yonsei University
Video Diffusion Transformers (DiTs) have been synthesizing high-quality video with high fidelity from given text descriptions involving motion. However, understanding how Video DiTs convert motion words into video remains insufficient. Furthermore, while prior studies on interpretable saliency maps primarily target objects, motion-related behavior in Video DiTs remains largely unexplored. In this paper, we investigate concrete motion features that specify when and which object moves for a given motion concept. First, to spatially localize, we introduce GramCol, which adaptively produces per-frame saliency maps for any text concept, including both motion and non-motion. Second, we propose a motion-feature selection algorithm to obtain an Interpretable Motion-Attentive Map (IMAP) that localizes motion spatially and temporally. Our method discovers concept saliency maps without the need for any gradient calculation or parameter update. Experimentally, our method shows outstanding localization capability on the motion localization task and zero-shot video semantic segmentation, providing interpretable and clearer saliency maps for both motion and non-motion concepts.
- Python > 3.10
- CUDA > 12.4
Please install the appropriate conda environment corresponding to the model you intend to use.
For CogVideoX:
conda env create -f EnvCreate.yml
conda activate imap
pip install torch==2.6.0 torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124
pip install flash-attn==2.7.4.post1 --no-build-isolation
pip install -r requirements-cogvideox.txtFor Wan2.1 or HunyuanVideo:
conda env create -f EnvCreate.yml
conda activate imap
pip install torch==2.6.0 torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124
pip install flash-attn==2.7.4.post1 --no-build-isolation
pip install -r requirements-wan-hunyuan.txtTo create an IMAP while generating a video, run the sampling script.
Note: The appearance of the visualized map can vary significantly depending on the visualization method used. We highly recommend saving and using the
.npyfiles for visualization.
Run the following command:
python main_Sampling.pyKey arguments and variables used in the paper can be found in the scripts directory.
To extract an IMAP from an existing real video, run the renoising script.
Note: The appearance of the visualized map can vary significantly depending on the visualization method used. We highly recommend saving and using the
.npyfiles for visualization.
Important - Input Resolution & Frames:
- CogVideoX and HunyuanVideo: Require 49-frame videos with a resolution of 480x720.
- Wan2.1: Requires 49-frame videos with a resolution of 480x832.
Run the following command:
python main_Renoising.pyKey arguments and variables used in the paper can be found in the scripts directory.
Our implementation partially builds on helblazer811/ConceptAttention.
