Particle-based Variational Inference with Generalized Wasserstein Gradient Flow. - Ziheng Cheng; Shiyue Zhang 002; Longlin Yu; Cheng Zhang (NeurIPS, 2023)

Reflected Flow Matching. - Tianyu Xie 001; Yu Zhu 004; Longlin Yu; Tong Yang; Ziheng Cheng; Shiyue Zhang 002; Xiangyu Zhang; Cheng Zhang (ICML, 2024)

Kernel Semi-Implicit Variational Inference. - Ziheng Cheng; Longlin Yu; Tianyu Xie 001; Shiyue Zhang 002; Cheng Zhang (ICML, 2024)

Functional Gradient Flows for Constrained Sampling. - Shiyue Zhang 002; Longlin Yu; Ziheng Cheng; Cheng Zhang (NeurIPS, 2024)

Semi-Implicit Functional Gradient Flow. - Shiyue Zhang 002; Ziheng Cheng; Cheng Zhang (CoRR, 2024)

Provable Sample-Efficient Transfer Learning Conditional Diffusion Models via Representation Learning. - Ziheng Cheng; Tianyu Xie 001; Shiyue Zhang 002; Cheng Zhang (CoRR, 2025)