| 2026 | AAAI | SIDE: Surrogate Conditional Data Extraction from Diffusion Models. | Yunhao Chen, Shujie Wang, Difan Zou, Xingjun Ma |
| 2026 | AAAI | Learning Diffusion Policy from Primitive Skills for Robot Manipulation. | Zhihao Gu, Ming Yang, Difan Zou, Dong Xu |
| 2026 | ACL | Breaking Contextual Inertia: Reinforcement Learning with Single-Turn Anchors for Stable Multi-Turn Interaction. | Xingwu Chen, Zhanqiu Zhang, Steven Y. Guo, Difan Zou |
| 2026 | COLT | Almost Linear Convergence under Minimal Score Assumptions: Quantized Transition Diffusion. | Xunpeng Huang, Yingyu Lin, Nikki Lijing Kuang, Hanze Dong, Difan Zou, Yian Ma, Tong Zhang |
| 2025 | ACL | SWE-Fixer: Training Open-Source LLMs for Effective and Efficient GitHub Issue Resolution. | Chengxing Xie, Bowen Li, Chang Gao, He Du, Wai Lam, Difan Zou, Kai Chen |
| 2025 | CVPR | Parallelized Autoregressive Visual Generation. | Yuqing Wang, Shuhuai Ren, Zhijie Lin, Yujin Han, Haoyuan Guo, Zhenheng Yang, Difan Zou, Jiashi Feng, Xihui Liu |
| 2025 | EMNLP | Model Unlearning via Sparse Autoencoder Subspace Guided Projections. | Xu Wang, Zihao Li, Benyou Wang, Yan Hu, Difan Zou |
| 2025 | ICLR | On the Feature Learning in Diffusion Models. | Andi Han, Wei Huang, Yuan Cao, Difan Zou |
| 2025 | ICLR | Beyond Surface Structure: A Causal Assessment of LLMs' Comprehension ability. | Yujin Han, Lei Xu, Sirui Chen, Difan Zou, Chaochao Lu |
| 2025 | ICLR | HyPoGen: Optimization-Biased Hypernetworks for Generalizable Policy Generation. | Hanxiang Ren, Li Sun, Xulong Wang, Pei Zhou, Zewen Wu, Siyan Dong, Difan Zou, Youyi Zheng, Yanchao Yang |
| 2025 | ICLR | How Does Critical Batch Size Scale in Pre-training? | Hanlin Zhang, Depen Morwani, Nikhil Vyas, Jingfeng Wu, Difan Zou, Udaya Ghai, Dean P. Foster, Sham M. Kakade |
| 2025 | ICML | Masked Autoencoders Are Effective Tokenizers for Diffusion Models. | Hao Chen, Yujin Han, Fangyi Chen, Xiang Li, Yidong Wang, Jindong Wang, Ze Wang, Zicheng Liu, Difan Zou, Bhiksha Raj |
| 2025 | ICML | Can Diffusion Models Learn Hidden Inter-Feature Rules Behind Images? | Yujin Han, Andi Han, Wei Huang, Chaochao Lu, Difan Zou |
| 2025 | ICML | Towards Understanding Fine-Tuning Mechanisms of LLMs via Circuit Analysis. | Xu Wang, Yan Hu, Wenyu Du, Reynold Cheng, Benyou Wang, Difan Zou |
| 2024 | COLT | Faster Sampling without Isoperimetry via Diffusion-based Monte Carlo. | Xunpeng Huang, Difan Zou, Hanze Dong, Yi-An Ma, Tong Zhang |
| 2024 | GLOBECOM | Optimized Transmit Beamformers for Dual-Function RadCom System. | Junhui Qian, Zhuoran Sun, Jie Wang, Le Zheng, Difan Zou, Xianxiang Yu, Jing Yang |
| 2024 | ICLR | Benign Oscillation of Stochastic Gradient Descent with Large Learning Rate. | Miao Lu, Beining Wu, Xiaodong Yang, Difan Zou |
| 2024 | ICLR | PRES: Toward Scalable Memory-Based Dynamic Graph Neural Networks. | Junwei Su, Difan Zou, Chuan Wu |
| 2024 | ICLR | How Many Pretraining Tasks Are Needed for In-Context Learning of Linear Regression? | Jingfeng Wu, Difan Zou, Zixiang Chen, Vladimir Braverman, Quanquan Gu, Peter L. Bartlett |
| 2024 | ICML | What Can Transformer Learn with Varying Depth? Case Studies on Sequence Learning Tasks. | Xingwu Chen, Difan Zou |
| 2024 | ICML | Improving Group Robustness on Spurious Correlation Requires Preciser Group Inference. | Yujin Han, Difan Zou |
| 2024 | ICML | Faster Sampling via Stochastic Gradient Proximal Sampler. | Xunpeng Huang, Difan Zou, Hanze Dong, Yian Ma, Tong Zhang |
| 2024 | ICML | Benign Overfitting in Two-Layer ReLU Convolutional Neural Networks for XOR Data. | Xuran Meng, Difan Zou, Yuan Cao |
| 2023 | COLT | The Implicit Bias of Batch Normalization in Linear Models and Two-layer Linear Convolutional Neural Networks. | Yuan Cao, Difan Zou, Yuanzhi Li, Quanquan Gu |
| 2023 | ICLR | Understanding the Generalization of Adam in Learning Neural Networks with Proper Regularization. | Difan Zou, Yuan Cao, Yuanzhi Li, Quanquan Gu |
| 2023 | ICML | Towards Robust Graph Incremental Learning on Evolving Graphs. | Junwei Su, Difan Zou, Zijun Zhang, Chuan Wu |
| 2023 | ICML | Finite-Sample Analysis of Learning High-Dimensional Single ReLU Neuron. | Jingfeng Wu, Difan Zou, Zixiang Chen, Vladimir Braverman, Quanquan Gu, Sham M. Kakade |
| 2023 | ICML | The Benefits of Mixup for Feature Learning. | Difan Zou, Yuan Cao, Yuanzhi Li, Quanquan Gu |
| 2022 | AISTATS | Self-training Converts Weak Learners to Strong Learners in Mixture Models. | Spencer Frei, Difan Zou, Zixiang Chen, Quanquan Gu |
| 2022 | ICML | Last Iterate Risk Bounds of SGD with Decaying Stepsize for Overparameterized Linear Regression. | Jingfeng Wu, Difan Zou, Vladimir Braverman, Quanquan Gu, Sham M. Kakade |
| 2021 | COLT | Benign Overfitting of Constant-Stepsize SGD for Linear Regression. | Difan Zou, Jingfeng Wu, Vladimir Braverman, Quanquan Gu, Sham M. Kakade |
| 2021 | ICLR | How Much Over-parameterization Is Sufficient to Learn Deep ReLU Networks? | Zixiang Chen, Yuan Cao, Difan Zou, Quanquan Gu |
| 2021 | ICLR | Direction Matters: On the Implicit Bias of Stochastic Gradient Descent with Moderate Learning Rate. | Jingfeng Wu, Difan Zou, Vladimir Braverman, Quanquan Gu |
| 2021 | ICML | Provable Robustness of Adversarial Training for Learning Halfspaces with Noise. | Difan Zou, Spencer Frei, Quanquan Gu |
| 2021 | ICML | On the Convergence of Hamiltonian Monte Carlo with Stochastic Gradients. | Difan Zou, Quanquan Gu |
| 2021 | UAI | Faster Convergence of Stochastic Gradient Langevin Dynamics for Non-Log-Concave Sampling. | Difan Zou, Pan Xu, Quanquan Gu |
| 2020 | ICLR | Improving Adversarial Robustness Requires Revisiting Misclassified Examples. | Yisen Wang, Difan Zou, Jinfeng Yi, James Bailey, Xingjun Ma, Quanquan Gu |
| 2020 | ICLR | On the Global Convergence of Training Deep Linear ResNets. | Difan Zou, Philip M. Long, Quanquan Gu |
| 2019 | AISTATS | Sampling from Non-Log-Concave Distributions via Variance-Reduced Gradient Langevin Dynamics. | Difan Zou, Pan Xu, Quanquan Gu |
| 2018 | ICML | Stochastic Variance-Reduced Hamilton Monte Carlo Methods. | Difan Zou, Pan Xu, Quanquan Gu |
| 2018 | UAI | Subsampled Stochastic Variance-Reduced Gradient Langevin Dynamics. | Difan Zou, Pan Xu, Quanquan Gu |
| 2017 | GLOBECOM | Characterization of a Practical Photon Counting Receiver in Optical Scattering Communication. | Difan Zou, Chen Gong, Kun Wang, Zhengyuan Xu |
| 2014 | ACSSC | Improving the NLOS optical scattering channel via beam reshaping. | Difan Zou, Shang-Bin Li, Zhengyuan Xu |