Yikun Bai

Physical AI Postdoc Fellow @ Purdue University. West Lafayette, IN, U.S.

Yikun_Bai.jpg
Math Department, Purdue University
West Lafayette, IN, U.S.

About Me

I am currently a Physical AI Postdoc Fellow at Purdue University, supervised by Dr. Guang Lin and Dr. Ruqi Zhang. Prior to Purdue, I was a postdoc in the Department of Computer Science at Vanderbilt University (2022-2025), supervised by Dr. Soheil Kolouri. I received my Ph.D. in Electrical and Computer Engineering from the University of Delaware in 2022, where I studied machine learning and statistics under the guidance of Dr. Dominique Guillot.

Current Research Interests

  • Discrete and Continuous Flow Matching / Diffusion Models: I study both discrete-time and continuous-time formulations of flow matching and diffusion models, with a focus on probability path design, transport dynamics, and scalable training/sampling algorithms for high-dimensional or non-Euclidean data.
  • Reinforcement Learning for Generative Models: I am interested in using RL methods to improve generative modeling pipelines, including reward-driven fine-tuning, policy optimization for generation quality/control, and learning strategies that connect exploration with robust sample synthesis.
  • Reinforcement Learning for Decision-Making Over Time (toward quant research): I work on RL for sequential and temporal decision problems under uncertainty, with long-term interests in applications aligned with quantitative researcher roles in hedge funds and related data-driven decision-making jobs.

News

"Glory to my Lord"