Zekun Cai

self_pic.jpg

I am a postdoctoral researcher at the University of Tokyo. I obtained my Ph.D. in 2024 from the Center for Spatial Information Science at the University of Tokyo, where I was supervised by Prof. Ryosuke Shibasaki and Prof. Xuan Song. Additionally, I conducted a research visit at Emory University from 2023. Prior to that, I received my M.S. from the University of Tokyo in 2021 and my B.S. from the University of Electronic Science and Technology of China in 2018.

During my Ph.D., I was fortunate to be co-advised by Prof. Renhe Jiang at UTokyo and collaborated with Prof. Liang Zhao at Emory University. My work centered on developing principled approaches for temporal domain generalization under continuously evolving data distributions. In parallel with my academic research, I also collaborated with industry partners such as Tencent and Yahoo! JAPAN, focusing on analyzing, modeling, and predicting trillion-scale human behavioral data in real-world environments.

I have published over 20 peer-reviewed papers in venues including NeurIPS, KDD, TKDE, TMC, and CIKM, and I regularly serve as a PC member for conferences such as NeurIPS, KDD, IJCAI, and CIKM.

Research Interests

My recent research focuses on developing mathematically and physically grounded approaches to enhance the adaptability and generalization of AI systems in complex environments. Key areas include:

  • Continuous Domain Generalization: Synchronizing evolution of data and neural networks through Structured Representations, Manifold Geometry, Group Equivariant, Neural Differential Equations, etc.
  • Spatiotemporal Learning at Scale: Graph Neural Network, Spatiotemporal Foundation Models, Dynamic Representation Learning, Real-world Applications, etc.

News

Sep 26, 2024 Our paper on Continuous Temporal Domain Generalization was accepted to NeurIPS 2024. Looking forward to see you in Vancouver.
Sep 22, 2024 I graduated with a Ph.D. from The University of Tokyo.
Feb 25, 2024 I will conduct a research visit at Emory University, collaborating with Prof. Liang Zhao on temporal domain generalization.
Aug 25, 2023 Our paper on urban human dynamics was accepted by IEEE Transactions on Mobile Computing (TMC). It introduces a new framework for large-scale citywide crowd transition prediction.
Aug 05, 2023 Our work MemDA: Forecasting Urban Time Series with Memory-based Drift Adaptation was accepted to CIKM 2023. See you in Birmingham!

Selected Publications

  1. Arxiv
    cdg.png
    Continuous Domain Generalization
    Zekun Cai, Yiheng Yao, Guangji Bai, Renhe Jiang, Xuan Song, Ryosuke Shibasaki, and Liang Zhao
    arXiv preprint, 2025
  2. NeurIPS
    koodos.gif
    Continuous Temporal Domain Generalization
    Zekun Cai, Guangji Bai, Renhe Jiang, Xuan Song, and Liang Zhao
    In Advances in Neural Information Processing Systems, Vancouver, Canada, 2024
  3. CIKM
    memda.jpg
    MemDA: Forecasting Urban Time Series with Memory-based Drift Adaptation
    Zekun Cai, Renhe Jiang, Xinyu Yang, Zhaonan Wang, Diansheng Guo, Hill Hiroki Kobayashi, Xuan Song, and Ryosuke Shibasaki
    In Proceedings of the 32nd ACM International Conference on Information and Knowledge Management, Birmingham, UK, 2023
  4. IEEE TMC
    citywideCTP.jpg
    Forecasting Citywide Crowd Transition Process via Convolutional Recurrent Neural Networks
    Zekun Cai, Renhe Jiang, Xinlei Lian, Chuang Yang, Zhaonan Wang, Zipei Fan, Kota Tsubouchi, Hill Hiroki Kobayashi, Xuan Song, and Ryosuke Shibasaki
    IEEE Transactions on Mobile Computing, 2023
  5. IEEE TKDE
    deepcrowd.png
    DeepCrowd: A Deep Model for Large-Scale Citywide Crowd Density and Flow Prediction
    Renhe Jiang*, Zekun Cai*, Zhaonan Wang, Chuang Yang, Zipei Fan, Quanjun Chen, Kota Tsubouchi, Xuan Song, and Ryosuke Shibasaki
    IEEE Transactions on Knowledge and Data Engineering, 2021
  6. CIKM
    dltraff.png
    DL-Traff: Survey and Benchmark of Deep Learning Models for Urban Traffic Prediction
    Renhe Jiang, Du Yin, Zhaonan Wang, Yizhuo Wang, Jiewen Deng, Hangchen Liu, Zekun Cai, Jinliang Deng, Xuan Song, and Ryosuke Shibasaki
    In Proceedings of the 30th ACM International Conference on Information and Knowledge Management), Virtual Event, Queensland, Australia, 2021
  7. KDD
    deepurbanevent.png
    Deepurbanevent: A System for Predicting Citywide Crowd Dynamics at Big Events
    Renhe Jiang, Xuan Song, Dou Huang, Xiaoya Song, Tianqi Xia, Zekun Cai, Zhaonan Wang, Kyoung-Sook Kim, and Ryosuke Shibasaki
    In Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, Anchorage, Alaska, 2019