Shaoyi Zheng
Publications
CV
MISC
Publications
Full list also available on
Google Scholar
†
First-author or co-first-author contribution.
2026
De-Diffusing WAMs: Action-Sufficient Direct Prediction for Robot Control
†
arXiv preprint
Learning to Be Guided: Shaping Latent Geometry for Classifier-Free Guidance
†
arXiv preprint
Smoothing Dark Areas in Molecular Latent Diffusion
arXiv preprint
InfoFlow KV: Information-Flow-Aware KV Recomputation for Long Context
†
ICML 2026
Hilbert Attention for Image Generation with Diffusion Models
†
ICML 2026
2025
ToMA: Token Merge with Attention for Diffusion Models
†
ICML 2025
Submodular Context Partitioning and Compression for In-Context Learning
†
arXiv preprint