Mao, Yuzhen

5 publications

TMLR 2026 ADiff4TPP: Asynchronous Diffusion Models for Temporal Point Processes Amartya Mukherjee, Ruizhi Deng, He Zhao, Yuzhen Mao, Leonid Sigal, Frederick Tung

ICLR 2026 IceCache: Memory-Efficient KV-Cache Management for Long-Sequence LLMs Yuzhen Mao, Qitong Wang, Martin Ester, Ke Li

NeurIPS 2025 Performative Risk Control: Calibrating Models for Reliable Deployment Under Performativity Victor Li, Baiting Chen, Yuzhen Mao, Qi Lei, Zhun Deng

ICML 2024 EvoluNet: Advancing Dynamic Non-IID Transfer Learning on Graphs Haohui Wang, Yuzhen Mao, Yujun Yan, Yaoqing Yang, Jianhui Sun, Kevin Choi, Balaji Veeramani, Alison Hu, Edward Bowen, Tyler Cody, Dawei Zhou

ICLR 2024 IceFormer: Accelerated Inference with Long-Sequence Transformers on CPUs Yuzhen Mao, Martin Ester, Ke Li