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.

profile photo
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.

Selected Publications

Google scholar for full list.

project image

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.

project image

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.

project image

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.

project image

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.

project image

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.


Design and source code from Jon Barron's website