FaiREE: fair classification with finite-sample and distribution-free guarantee. - Puheng Li; James Zou 001; Linjun Zhang (ICLR, 2023)

On the Generalization Properties of Diffusion Models. - Puheng Li; Zhong Li 004; Huishuai Zhang; Jiang Bian 002 (NeurIPS, 2023)

Exploring Neural Network Landscapes: Star-Shaped and Geodesic Connectivity. - Zhanran Lin; Puheng Li; Lei Wu (CoRR, 2024)

Analyzing the Role of Permutation Invariance in Linear Mode Connectivity. - Keyao Zhan; Puheng Li; Lei Wu (AISTATS, 2025)

CHORDS: Diffusion Sampling Accelerator with Multi-core Hierarchical ODE Solvers. - Jiaqi Han; Haotian Ye; Puheng Li; Minkai Xu; James Zou 001; Stefano Ermon (CoRR, 2025)