Yuan Yuan (苑 苑)
I am currently a postdoctoral researcher at Courant Institute of Mathematical Sciences, New York University, working with Prof. Carlos Fernandez-Granda and Prof. Laure Zanna, and as a member of Multiscale Machine Learning In Coupled Earth System Modeling (M²LInES).
Previously, I completed my PhD at FIBLAB, Department of Electronic Engineering, Tsinghua University, advised by Prof. Depeng Jin and Prof. Yong Li. I received my bachelor degree from the Department of Electronic Engineering, Tsinghua University in 2020.
My research centers on developing scalable artificial intelligence methodologies for real-world spatiotemporal systems, with applications in climate dynamics, urban systems, and energy infrastructures. I aim to build robust foundation models that capture complex interdependencies across spatial and temporal scales, enabling intelligent reasoning and decision-making in dynamic and uncertain environments.
|
1126 Warren Weaver Hall,
New York, NY 10012
|
Research Interests
- Spatio-temporal Learning: AI foundation models for Earth systems, spatio-temporal foundation models, diffusion models for spatio-temporl data
- Urban Computing: Mobility foundation models, Urban spatiotemporal foundation models
|
News
- [New!] [2025.10] We are excited to release a global open data for human mobility,
WorldMove, by leveraging publicly available multi-source data.
The accompanying paper has been accepted by
Scientific Data.
- [New!] [2025.9] Our paper on urban foundation models,
UrbanDiT, has been accepted to NeurIPS 2025.
- [New!] [2025.8] Two papers on mobility foundation models,
MoveGCL and
UniMove, have been accepted to ACM SIGSPATIAL 2025.
- [New!][2025.6] I am honored to receive the Outstanding Doctoral Dissertation Award and the Outstanding Ph.D. Graduate Award from Tsinghua University.
- [New!] [2025.6] Our survey on world models has been published in
ACM Computing Surveys.
Check out our gularly updated paper list at
World-Model on GitHub.
Open to discussions and collaborations!
- [New!][2025.5] I successfully defended my Ph.D.! 😊 Deep thanks to Prof. Depeng Jin, Prof. Li Yong, and to everyone who has supported me throughout this journey.
- [New!] [2025.3] Our paper on learning the complexity of urban mobility, DeepMobility, has been published in PNAS Nexus.
- [New!] [2025.2] We are happy to release our new urban spatio-temporal foundation models —
UniST-v2 and
UniFlow.
UniST-v2 has been published in IEEE TKDE.
- [New!] [2025.1] Two papers on noise priors for diffusion models have been accepted by IJCAI 2025
(NPDiff)
and WWW 2025
(CoDiffMob).
- [New!] [2024.8] Our paper
UniST
is accepted by KDD 2024, which built a one-for-all foundation model for urban spatio-temporal prediction.
The code, pretrained weights and data (over 130$m$ points) are released
here.
- [New!] [2024.5] Our
paper
about spatio-temporal few-shot learning with diffusion model is accepted to ICLR 2024.
Feel free to try our
code on cross-city transfer tasks.
|
|
UniST: A Prompt-Empowered Universal Model for Urban Spatio-Temporal Prediction
Yuan Yuan, Jingtao Ding, Jie Feng, Depeng Jin, Yong Li
KDD, 2024
paper /
pdf /
code /
slides /
We build a universal model for general spatio-temporal prediction and show the benefits of a one-for-all solution in urban contexts.
|
|
Spatio-Temporal Few-Shot Learning via Diffusive Neural Network Generation
Yuan Yuan*, Chenyang Shao*, Jingtao Ding, Depeng Jin, Yong Li
ICLR, 2024
paper /
pdf /
code /
This diffusion-based framework performs generative pre-training on a collection of model parameters. By generating customized model parameters, we manage to address spatio-temporal few-shot learning.
|
|
Spatio-temporal Diffusion Point Processes
Yuan Yuan,Jingtao Ding, Chenyang Shao, Depeng Jin, Yong Li
KDD, 2023
paper /
pdf /
code /
We develop a diffusion model to learn spatio-temporal point processes.
|
|
Learning to Simulate Daily Activities via Modeling Dynamic Human Needs
Yuan Yuan, Huandong Wang, Jingtao Ding, Depeng Jin, Yong Li
The Web Conference, 2023
paper /
pdf /
code /
We introduce the modeling of dynamic human needs into activity simulation.
|
|
Activity Trajectory Generation via Modeling Spatiotemporal Dynamics
Yuan Yuan, Jingtao Ding, Huandong Wang, Depeng Jin, Yong Li
KDD, 2022
paper /
pdf /
code /
ActSTD captures spatiotemporal dynamics underlying activity trajectories by leveraging neural differential equations.
|
|