Cheng, Ziheng

14 publications

ICLR 2025 Convergence of Distributed Adaptive Optimization with Local Updates Ziheng Cheng, Margalit Glasgow
NeurIPS 2025 OVERT: A Benchmark for Over-Refusal Evaluation on Text-to-Image Models Ziheng Cheng, Yixiao Huang, Hui Xu, Somayeh Sojoudi, Xuandong Zhao, Dawn Song, Song Mei
NeurIPS 2025 Provable Sample-Efficient Transfer Learning Conditional Diffusion Models via Representation Learning Ziheng Cheng, Tianyu Xie, Shiyue Zhang, Cheng Zhang
NeurIPSW 2024 Convergence of Distributed Adaptive Optimization with Local Updates Ziheng Cheng, Margalit Glasgow
NeurIPSW 2024 Data-Efficient Training by Evolved Sampling Ziheng Cheng, Zhong Li, Jiang Bian
NeurIPS 2024 Functional Gradient Flows for Constrained Sampling Shiyue Zhang, Longlin Yu, Ziheng Cheng, Cheng Zhang
ICML 2024 Kernel Semi-Implicit Variational Inference Ziheng Cheng, Longlin Yu, Tianyu Xie, Shiyue Zhang, Cheng Zhang
ICLR 2024 Momentum Benefits Non-Iid Federated Learning Simply and Provably Ziheng Cheng, Xinmeng Huang, Pengfei Wu, Kun Yuan
ICML 2024 Reflected Flow Matching Tianyu Xie, Yu Zhu, Longlin Yu, Tong Yang, Ziheng Cheng, Shiyue Zhang, Xiangyu Zhang, Cheng Zhang
COLT 2024 The Limits and Potentials of Local SGD for Distributed Heterogeneous Learning with Intermittent Communication Kumar Kshitij Patel, Margalit Glasgow, Ali Zindari, Lingxiao Wang, Sebastian U Stich, Ziheng Cheng, Nirmit Joshi, Nathan Srebro
NeurIPS 2023 Particle-Based Variational Inference with Generalized Wasserstein Gradient Flow Ziheng Cheng, Shiyue Zhang, Longlin Yu, Cheng Zhang
CVPR 2021 Memory-Efficient Network for Large-Scale Video Compressive Sensing Ziheng Cheng, Bo Chen, Guanliang Liu, Hao Zhang, Ruiying Lu, Zhengjue Wang, Xin Yuan
CVPR 2021 MetaSCI: Scalable and Adaptive Reconstruction for Video Compressive Sensing Zhengjue Wang, Hao Zhang, Ziheng Cheng, Bo Chen, Xin Yuan
ECCV 2020 BIRNAT: Bidirectional Recurrent Neural Networks with Adversarial Training for Video Snapshot Compressive Imaging Ziheng Cheng, Ruiying Lu, Zhengjue Wang, Hao Zhang, Bo Chen, Ziyi Meng, Xin Yuan