Education Experience
|
University of California, Berkeley, Berkeley, United States
|
08/2024 - 05/2025
|
|
EECS Visiting Student
|
|
GPA: 4.0 / 4.0
|
|
|
South China University of Technology (SCUT), Guangzhou, China
|
09/2021 - 07/2025
|
|
B.Eng in Artificial Intelligence (AI)
|
|
GPA: 3.89 / 4.00 | Rank: 3 / 80
|
|
|
|
AI for Scientific Simulation and Discovery Lab, Westlake University
|
2025/08 - Present
|
Research Intern
Developing discrete diffusion models with improved efficiency and interpretability.
Advised by Prof. Tailin Wu.
|
|
|
Berkeley AI Research Lab (BAIR), UC Berkeley
|
2024/09 - 2025/08
|
Undergraduate Research Assistant
Developed foundation models for scientific computing and SciML agents for automatic code generation.
Advised by Prof. Michael Mahoney and
Dr. Amir Gholami.
|
|
|
AI for Scientific Simulation and Discovery Lab, Westlake University
|
2023/07 - 2024/06
|
Research Intern
AI for PDE and physical system simulation and control.
Advised by Prof. Tailin Wu.
|
|
|
Cyber-Med Laboratory, South China University of Technology
|
2022/04 - 2023/06
|
Student Research Project Leader
Biometric recognition in human-computer interaction.
Advised by
Prof. Zhanpeng Jin and
Prof. Yang Gao.
|
Contests
|
Mathematical Contest in Modeling (MCM/ICM)
|
Meritorious Winner
|
|
International Genetically Engineered Machine Competition (iGEM)
|
Silver Award
|
|
Asia and Pacific Mathematical Contest in Modeling (APMCM)
|
First Prize
|
|
Awards
(Selected)
|
UC Berkeley BGA Scholarship (Top 2.5%)
|
UC Berkeley
|
|
Best Technical Solution Award
|
UC Berkeley RDI
|
|
National Undergraduate Innovation Grant
|
SCUT
|
|
Provincial Key Undergraduate Innovation Grant
|
SCUT
|
|
Outstanding Student Trainee Award
|
Tencent (China)
|
|
First-Class Academic Scholarship (Top 5%)
|
SCUT
|
|
|
Publications
* indicates equal contribution
|
|
SciML Agents: Write the Solver, Not the Solution
Saarth Gaonkar *,
Xiang Zheng *,
Haocheng Xi,
Rishabh Tiwari,
Kurt Keutzer,
Dmitriy Morozov,
Michael Mahoney,
Amir Gholami
We explore LLMs as SciML agents that generate executable and numerically valid ODE solvers directly from natural-language problem descriptions. We introduce two new datasets, including a diagnostic adversarial benchmark and a large-scale benchmark of 1,000 diverse ODE tasks, and show that modern LLMs, with guided prompting and proper fine-tuning, can reliably select appropriate solvers and produce correct numerical code.
Math-AI Workshop, NeurIPS 2025
|
|
Wavelet Diffusion Neural Operator
Peiyan Hu *,
Rui Wang *,
Xiang Zheng,
Tao Zhang,
Haodong Feng,
Ruiqi Feng,
Long Wei,
Yue Wang,
Zhi-Ming Ma,
Tailin Wu
We propose the Wavelet Diffusion Neural Operator (WDNO), a wavelet-domain diffusion framework for PDE simulation and control that enables robust multi-resolution learning under abrupt dynamics. WDNO outperforms strong baselines on Burgers’, compressible Navier-Stokes, 2D incompressible flows, and the ERA5 climate dataset, achieving a 78% reduction in smoke leakage in challenging 2D control tasks.
ICLR 2025
|
|
CL-DiffPhyCon: Closed-loop Diffusion Control of Complex Physical Systems
Long Wei *,
Haodong Feng *,
Yuchen Yang,
Ruiqi Feng ,
Peiyan Hu ,
Xiang Zheng,
Tao Zhang,
Dixia Fan,
Tailin Wu
We propose CL-DiffPhyCon, an efficient closed-loop diffusion-based control framework that generates real-time control signals via asynchronous denoising across heterogeneous physical time steps. CL-DiffPhyCon achieves superior control performance with significantly improved sampling efficiency on 1D Burgers’ equation and 2D incompressible fluid control tasks.
ICLR 2025
|
|
Manuscripts
* indicates equal contribution
|
|
Scientific Data Augmentation for In-Context Learning of Differential Equations
Amin Totounferoush *,
Xiang Zheng *,
Arnur Nigmetov ,
Shashank Subramanian,
Haocheng Xi,
Kurt Keutzer,
Steffen Staab,
Dmitriy Morozov,
Michael Mahoney,
Amir Gholami
We study in-context learning (ICL) for solving dynamical systems and propose scientific data augmentation strategies based on fast approximate solvers to improve accuracy and generalization for both forward and inverse problems. Our methods yield up to 6× accuracy gains with ICON-LM and 78% improvement with GPT-4, significantly enhancing both in- and out-of-distribution performance.
|
|
Raise2Auth: A Dual-Factor, Adaptive Gesture Authentication System for Enhanced Mobile Security
Yang Gao,
Yuqin Yang,
Xiang Zheng,
Yaoshi Chen,
Yanzi Zhou,
Zhanpeng Jin
We propose Raise2Auth, a dual-factor, gesture-based mobile authentication system that leverages users’ natural grasping and lifting motions captured by smartphone IMU sensors for robust and seamless identity verification. Experiments with 25 participants achieve over 97.6% authentication accuracy and a false acceptance rate below 0.1% under both imitation and synthetic attacks.
|
|