Selected Publications

2026

• Scaling reinforcement learning for diffusion models via velocity matching
  J. Choi, W. Guo, Y. Zhu, A. Vahdat, M. Tao, J. Berner, and Y. Chen

• Enhancing reasoning for diffusion llms via distribution matching policy optimization
  Y. Zhu, W. Guo, J. Choi, P. Molodyk, B. Yuan, M. Tao, and Y. Chen
  ICML (Spotlight)

• Rethinking the design space of reinforcement learning for diffusion models: on the importance of likelihood estimation beyond loss design
  J. Choi, Y. Zhu, W. Guo, P. Molodyk, B. Yuan, J. Bai, Y. Xin, M. Tao, and Y. Chen
  ICML

• CDGS: Compositional diffusion with guided search for long-horizon planning
  U. Mishra, D. He, Y. Chen, and D. Xu
  ICLR (Oral)

2025

• Safety verification of nonlinear stochastic systems via probabilistic tube
  Z. Liu, S. Jafarpour, and Y. Chen
  IEEE Transactions on Automatic Control

• A new proof of sub-Gaussian norm concentration inequality
  Z. Liu, S. Power, and Y. Chen

• MDNS: Masked diffusion neural sampler via stochastic optimal control
  Y. Zhu, W. Guo, J. Choi, G. Liu, Y. Chen, and M. Tao
  NeurIPS

• ASBS: Adjoint Schrodinger Bridge Sampler
  G. Liu, J. Choi, Y. Chen, B. Miler, and R. Chen
  NeurIPS (Oral)

2024

• Probabilistic reachability analysis of stochastic control systems
  S. Jafarpour, Z. Liu, and Y. Chen
  IEEE Transactions on Automatic Control

• Provable benefit of annealed Langevin Monte Carlo for non-log-concave sampling
  W. Guo, M. Tao, and Y. Chen
  ICLR

2023

• RES: Improved order analysis and design of exponential integrator for diffusion models sampling
  Q. Zhang, J. Song, and Y. Chen

• Improved dimension dependence of a proximal algorithm for sampling
  J. Fan, B. Yuan, and Y. Chen
  COLT

• Generative skill chaining: Long-horizon skill planning with diffusion models
  U. Mishra, S. Xue, Y. Chen, and D. Xu
  CoRL

2022

• DEIS: Fast sampling of diffusion models with exponential integrator
  Q. Zhang, and Y. Chen
  ICLR

• gDDIM: Generalized denoising diffusion implicit models
  Q. Zhang, M. Tao, and Y. Chen
  ICLR (Spotlight)

• Improved analysis for a proximal algorithm for sampling
  Y. Chen, S. Chewi, A. Salim, and A. Wibisono
  COLT

• A Gaussian variational inference approach to motion planning
  H. Yu, and Y. Chen
  IEEE Robotics and Automation Letters

2021

• Path Integral Sampler: a stochastic control approach for sampling
  Q. Zhang, and Y. Chen
  ICLR

• Diffusion Normalizing Flow
  Q. Zhang, and Y. Chen
  NeurIPS

• Variational Wasserstein gradient flow
  J. Fan, Q. Zhang, A. Taghvaei, and Y. Chen
  ICML

• Density control of interacting agent systems
  Y. Chen
  IEEE Transactions on Automatic Control

2020

• Scalable computations of Wasserstein barycenter via input convex neural networks
  J. Fan, A. Taghvaei, and Y. Chen
  ICML (Long talk)

• Multi-marginal optimal transport and probabilistic graphical models
  I. Haasler, R. Singh, Q. Zhang, J. Karlsson, and Y. Chen
  IEEE Transactions on Information Theory

• Energy harvesting from anisotropic fluctuations
  O. M. Miangolarra, A. Taghvaei, R. Fu, Y. Chen, and T. T. Georgiou
  Physical Review E

• The most likely evolution of diffusing and vanishing particles: Schrodinger bridges with unbalanced marginals
  Y. Chen, T. T. Georgiou, and M. Pavon
  SIAM Journal on Control and Optimization (SIGEST Paper)

2019

• Multimarginal optimal transport with a tree-structured cost and the Schroedinger bridge problem
  I. Haasler, A. Ringh, Y. Chen and J. Karlsson
  SIAM Journal on Control and Optimization (Best Paper)

2018

• Optimal transport for Gaussian mixture models
  Y. Chen, A. Tannenbaum, and T. T. Georgiou
  IEEE Access

2017

• Measure-valued spline curves: an optimal transport viewpoint
  Y. Chen, G. Conforti, and T. T. Georgiou
  SIAM Journal of Mathematical Analysis

• Stochastic control and nonequilibrium thermodynamics: Fundamental limits
  Y. Chen, A. Tannenbaum, and T. T. Georgiou
  IEEE Transactions on Automatic Control

2016

• Matrix optimal mass transport: a quantum mechanical approach
  Y. Chen, T. T. Georgiou, and A. Tannenbaum
  IEEE Transactions on Automatic Control

• Optimal steering of a linear stochastic system to a final probability distribution, Part III
  Y. Chen, T. T. Georgiou, and M. Pavon
  IEEE Transactions on Automatic Control

2015

• Entropic and displacement Interpolation: A computational approach using the Hilbert metric
  Y. Chen, T. T. Georgiou, and M. Pavon
  SIAM Journal on Applied Mathematics

• Optimal transport over a linear dynamical system
  Y. Chen, T. T. Georgiou, and M. Pavon
  IEEE Transactions on Automatic Control

2014

• Optimal steering of a linear stochastic system to a final probability distribution, Part I
  Y. Chen, T. T. Georgiou, and M. Pavon
  IEEE Transactions on Automatic Control (Best Paper)

• Optimal steering of a linear stochastic system to a final probability distribution, Part II
  Y. Chen, T. T. Georgiou, and M. Pavon
  IEEE Transactions on Automatic Control

• On the relation between optimal transport and Schroedinger bridges: A stochastic control viewpoint
  Y. Chen, T. T. Georgiou, and M. Pavon
  Journal of Optimization Theory and Applications