Publications

Accepted work, manuscripts under review, and ongoing research projects.

Publication and project status follows the latest CV. DraftKV is a project in progress; Draft Attention v2 and MosaicAttention are under review.

2026

  1. DraftKV: Efficient Long-Term Memory via Attention-Guided KV Retrieval for Video Generation
    Oct 2026
    Project in progress; First Author
  2. Draft Attention v2: Fast Video Diffusion with Low-Resolution-Guided Mixed-Precision Attention
    Rui Ding, Haopeng Li, Weize Ma, Yufa Zhou, Yitong Li, Xiaoling Zhou, Jiashuo Cao, Liyang Li, Hua Geng, Jiuxiang Gu, Jun Lin, Chunhua Shen, Enze Xie, and Xuan Shen
    Sep 2026
    Under review
  3. MosaicAttention: Hierarchical Block-Sparse Attention for Multi-View Geometry Transformers
    Zhizhen Pan, Rui Ding, Weize Ma, Xiaoling Zhou, Jun Lin, Chunhua Shen, and Xuan Shen
    Sep 2026
    Under review
  4. Mind Dreamer: Untethering Imagination via Active Causal Intervention on Latent Manifolds
    Shaojun Xu, Xiaoling Zhou, Yihan Lin, Yapeng Meng, Xinglong Ji, Luping Shi, and Rong Zhao
    In International Conference on Machine Learning (ICML), May 2026
    ICML 2026
    PDF