Publications

Full list also available on Google Scholar

2026

Smoothing Dark Areas in Molecular Latent Diffusion
Xi Wang, Jiahan Li, Yuxuan Xia, Yingcheng Wu, Shaoyi Zheng, Shengjie Wang arXiv preprint
InfoFlow KV: Information-Flow-Aware KV Recomputation for Long Context
Shaoyi Zheng*, Xin Teng*, Canyu Zhang*, Danyang Zhuo, Tianyi Zhou, Shengjie Wang ICML 2026
Hilbert Attention for Image Generation with Diffusion Models
Shaoyi Zheng, Wenbo Lu, Yuxuan Xia, Haomin Liu, Shengjie Wang ICML 2026

2025

ToMA: Token Merge with Attention for Diffusion Models
Shaoyi Zheng*, Wenbo Lu*, Yuxuan Xia, Shengjie Wang ICML 2025
Submodular Context Partitioning and Compression for In-Context Learning
Shaoyi Zheng, Canyu Zhang, Tianyi Zhou, Shengjie Wang arXiv preprint