At SUFE, 2023. With thanks to Yixuan and Taiyun.
My research lies at the intersection of Statistics and AI. I develop statistical foundations and methodologies to make modern AI more capable and efficient. My interests include:
- Statistical Machine Learning
- Representation Learning
- Generative AI
- Foundation Models
I lead SAGA Lab at CUHK-Shenzhen, where we combine theoretical analysis, mechanistic investigation, and empirical research to develop better generative models for vision and language. A current focus is continuous diffusion language modeling. I also co-initiated Polyadic Research Group, a collaborative research collective advancing this direction through model development and open, reproducible research.
Join us. I am recruiting PhD and MPhil students and research assistants. See SAGA Lab openings for research opportunities and how to apply.
Before joining CUHK-Shenzhen, I was a Postdoctoral Fellow at the National University of Singapore, working with Prof. Kenji Kawaguchi, and a Research Scientist in the AI Theory Group of Noah’s Ark Lab. I received my Ph.D. in Statistics from Purdue University under the supervision of Prof. Guang Cheng, after earning an M.S. in Statistics from the University of Chicago and a B.S. in Mathematics from Tsinghua University.
Selected papers
A selection across my research interests. View all papers →
- arXivTransformers Are Born Biased: Structural Inductive Biases at Random Initialization and Their Practical ConsequencesarXiv preprint, 2026The inspired methodologies are published in ICLR 2026 and ICML 2026.
- ICML
Oral Any-Order GPT as Masked Diffusion Model: Decoupling Formulation and ArchitectureInternational Conference on Machine Learning (selected for Oral presentation), 2026Abridged in the ICML 2025 Workshop on Efficient Systems for Foundation Models (ES-FoMo III) (Selected for Oral presentation). - ICLRPrefixMemory-Tuning: Modernizing Prefix-Tuning by Decoupling the Prefix from AttentionInternational Conference on Learning Representations, 2026Abridged in the ICML 2025 Workshop on Test-Time Adaptation: Putting Updates to the Test (PUT).
- NeurIPSReward-Instruct: A Reward-Centric Approach to Fast Photo-Realistic Image GenerationAdvances in Neural Information Processing Systems, 2025
- JMLRMinimax Optimal Deep Neural Network Classifiers Under Smooth Decision BoundaryJournal of Machine Learning Research, 26 (136), 2025
- JMLRRandom Smoothing Regularization in Kernel Gradient Descent LearningJournal of Machine Learning Research, 25 (284), 2024
- NeurIPS
Spotlight Complexity Matters: Rethinking the Latent Space for Generative ModelingAdvances in Neural Information Processing Systems (selected for Spotlight presentation), 2023 - NeurIPSDiff-Instruct: A Universal Approach for Transferring Knowledge From Pre-trained Diffusion ModelsAdvances in Neural Information Processing Systems, 2023
- ICLRYour Contrastive Learning Is Secretly Doing Stochastic Neighbor EmbeddingInternational Conference on Learning Representations, 2023
- NeurIPS
Spotlight Understanding Square Loss in Training Overparametrized Neural Network ClassifiersAdvances in Neural Information Processing Systems (selected for Spotlight presentation), 2022 - AISTATSRegularization Matters: A Nonparametric Perspective on Overparametrized Neural NetworkIn International Conference on Artificial Intelligence and Statistics, 2021