Publications

Full list also available on Google Scholar

2026 InfoFlow KV: Information-Flow-Aware KV Recomputation for Long Context Shaoyi Zheng*, Xin Teng*, Canyu Zhang*, Danyang Zhuo, Tianyi Zhou, Shengjie Wang ICML 2026
2025 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, PMLR 267:40930–40951
2025 Submodular Context Partitioning and Compression for In-Context Learning Shaoyi Zheng, Canyu Zhang, Tianyi Zhou, Shengjie Wang arXiv preprint