Hi, I’m a 4th-year CS undergraduate at School of Information and Technology at ShanghaiTech University. My research interests includes computer vision and machine learning. My current research focus is on Deep Generative Models such as Diffusion Bridge Models and Flow Matching.

I have published more than 1 papers at the top international AI conferences with total google scholar citations 0+.

Please find my CV here for more about my research projects and background. And please feel free to contact me for any questions or opportunities.

🔥 News

• 2025.5: Our paper UniDB: A Unified Diffusion Bridge Framework via Stochastic Optimal Control accepted by ICML 2025 as a Spotlight!🎉🎉
• 2025.4: Our paper UniDB won Outstanding Paper Award of ICLR 2025 DeLTa Workshop!🎉🎉

📝 Publications

ICML 2025 Spotlight

UniDB: A Unified Diffusion Bridge Framework via Stochastic Optimal Control, ICML 2025 Spotlight

Kaizhen Zhu#, Mokai Pan#, Yuexin Ma, Yanwei Fu, Jingyi Yu, Jingya Wang, Ye Shi*

[paper] [project] [code]

🎖 Honors and Awards

• 2021.11 Second Prize in “Chinese Mathematics Competitions”.

📖 Educations

• 2021.09 - 2025.06 (now), a CS undergraduate at School of Information and Technology at ShanghaiTech University.