Hi, I’m a second year graduate student at ShanghaiTech University. My current research focus in Deep Generative Models, such as Diffusion Bridge and Flow Matching.

I am a National Master of Sport in Chess, certified by the Chinese Chess Association.

♛ My Chess Game

Move:

蓝色白方下棋 | 红色黑方下棋

🔥 News

2025.11: Our paper Sample from What You See: Visuomotor Policy Learning via Diffusion Bridge with Observation-Embedded Stochastic Differential Equation is submitting.
2025.11: Our paper A Unified and Fast-Sampling Diffusion Bridge Framework via Stochastic Optimal Control is submitting.
2025.09: Our paper Diffusion Bridge or Flow Matching? A Unifying Framework and Comparative Analysis is submitting.
2025.05: Our paper UniDB: A Unified Diffusion Bridge Framework via Stochastic Optimal Control accepted by ICML 2025 as a Spotlight!🎉🎉
2025.04: Our paper UniDB won Outstanding Paper Award of ICLR 2025 DeLTa Workshop!🎉🎉
2025.04: Our paper AffordDP: Generalizable Diffusion Policy with Transferable Affordance accepted by CVPR 2025!🎉🎉

📝 Publications

Arxiv

Sample from What You See: Visuomotor Policy Learning via Diffusion Bridge with Observation-Embedded Stochastic Differential Equation

Zhaoyang Liu, Mokai Pan, Zhongyi Wang, Kaizhen Zhu, Haotao Lu, Jingya Wang, Ye Shi*

[paper]

Arxiv

Diffusion Bridge or Flow Matching? A Unifying Framework and Comparative Analysis

Kaizhen Zhu#, Mokai Pan#, Zhechuan Yu, Jingyi Yu, Jingya Wang, Ye Shi*

[paper] [code]

Arxiv

A Unified and Fast-Sampling Diffusion Bridge Framework via Stochastic Optimal Control

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

[paper] [code]

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]

CVPR 2025

AffordDP: Generalizable Diffusion Policy with Transferable Affordance, CVPR 2025.

Shijie Wu#, Yihang Zhu#, Yunao Huang, Kaizhen Zhu, Jiayuan Gu, Jingyi Yu, Ye Shi*, Jingya Wang*

[paper] [project] [code]

🎖 Honors and Awards

2023.6 First Prize, Zhejiang Provincial Mathematics Competition 2023.
2022.11 National First Prize, China Undergraduate Mathematical Contest in Modeling (CUMCM) 2022.
2022.8 Second Prize, National Undergraduate Electronic Design Contest (NUEDC) 2022.

📖 Educations

ShanghaiTech University

2024.09 - now

School of Information Science and Technology

Major: M.S. in Computer Science

MoE Key Laboratory of Intelligent Perception and Human Machine Collaboration, VDI Center, YesAI Lab

Hangzhou Dianzi University

2020.09 - 2024.06

School of Electronic Information

Major: B.E. in Electronic Science and Technology