ML Anthology
Authors
Search
About
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