Hello! I'm currently a first-year PhD student at New York University. I obtained my MSc in Statistics from Renmin University of China, under the guidance of Prof. Hanfang Yang and Prof. Hongteng Xu. I was also a visiting research intern at Carnegie Mellon University, co-hosted by Prof. Shixiang (Woody) Zhu and Prof. Holly Wiberg. My research interests lie in machine learning for sequential data modeling and sequential decision making. Specifically, I focus on reinforcement learning, stochastic process, control theory, and their applications in cyber-physical energy systems.
I am excited to begin my PhD journey at NYU in the fall of 2025!
I am eager to explore interesting research ideas with fellow scholars and look forward to engaging in in-depth collaboration,
particularly in the areas of machine learning and control, as well as data-driven optimization for sustainable power systems.
Most recent publications on
CV.
‡ indicates equal contribution.
A Bayesian Mixture Model of Temporal Point Processes with Determinantal Point Process Prior
Yiwei Dong, Shaoxin Ye, Yuwen Cao, Qiyu Han, Hongteng Xu, Hanfang Yang
IEEE Transactions on Signal Processing, vol 73, 2216-2226, 2025. A short version is accepted by NeurIPS 2024 workshop on Bayesian Decision-making and Uncertainty.
Pedestrian Volume Prediction Using a Diffusion Convolutional Gated Recurrent Unit Model
Yiwei Dong, Tingjin Chu, Lele Zhang, Hadi Ghaderi, Hanfang Yang
Journal of Agricultural, Biological, and Environmental Statistics, 2025.
Extrapolated Random Tree for Regression
Yuchao Cai, Yuheng Ma, Yiwei Dong, Hanfang Yang
Proceedings of the 40th International Conference on Machine Learning (ICML), PMLR 202:3442-3468, 2023.
Meta-Adaptive Stock Movement Prediction with Two-Stage Representation Learning
Donglin Zhan‡, Yusheng Dai‡, Yiwei Dong‡, Jinghai He, Zhenyi Wang, James Anderson
Proceedings of the 2024 SIAM International Conference on Data Mining (SDM). Society for Industrial and Applied Mathematics, 2024.
A Bayesian Mixture Model of Temporal Point Processes with Determinantal Point Process Prior
Yiwei Dong, Shaoxin Ye, Yuwen Cao, Qiyu Han, Hongteng Xu, Hanfang Yang
IEEE Transactions on Signal Processing, vol 73, 2216-2226, 2025. A short version is accepted by NeurIPS 2024 workshop on Bayesian Decision-making and Uncertainty.
Pedestrian Volume Prediction Using a Diffusion Convolutional Gated Recurrent Unit Model
Yiwei Dong, Tingjin Chu, Lele Zhang, Hadi Ghaderi, Hanfang Yang
Journal of Agricultural, Biological, and Environmental Statistics, 2025.
Extrapolated Random Tree for Regression
Yuchao Cai, Yuheng Ma, Yiwei Dong, Hanfang Yang
Proceedings of the 40th International Conference on Machine Learning (ICML), PMLR 202:3442-3468, 2023.
Meta-Adaptive Stock Movement Prediction with Two-Stage Representation Learning
Donglin Zhan‡, Yusheng Dai‡, Yiwei Dong‡, Jinghai He, Zhenyi Wang, James Anderson
Proceedings of the 2024 SIAM International Conference on Data Mining (SDM). Society for Industrial and Applied Mathematics, 2024.
Full Resume in PDF.
I enjoy citywalk, musicals, and photography :D
See you in New York City, in China, in Antarctica, in the world!