Bohan Chen 🌞
Bohan Chen

Postdoctoral Researcher

About Me

Bohan Chen is a postdoctoral researcher at Caltech working with Prof. Andrew Stuart. His research centers on Bayesian inverse problems and data assimilation, with an emphasis on scalable methods for learning and sampling complex posterior distributions. His current work combines conditional generative modeling, machine-learning-enhanced ensemble methods, and transformers and attention acting on probability measures. During his Ph.D. at UCLA, he worked with Prof. Andrea Bertozzi on graph-based and active-learning methods for imaging and label-efficient inference, using data geometry and adaptive information acquisition to address problems with limited observations.

Email: bhchen@caltech.edu

Phone: +1 424-402-7355

Interests
  • Bayesian Inverse Problems
  • Data Assimilation
  • Conditional Generative Modeling
  • Scientific Machine Learning
  • Attention on Probability Measures
  • Graph-Based Active Learning
Education
  • Ph.D. in Mathematics

    Department of Mathematics, University of California, Los Angeles

  • B.S. in Mathematics

    School of Mathematical Sciences, Peking University

📚 My Research
My research develops computational and learning-based methods for Bayesian inference from partial and noisy observations, with Bayesian inverse problems and data assimilation as two central settings. I am particularly interested in representing, learning, and sampling complex posterior distributions in high-dimensional systems. My current work combines conditional generative modeling and measure transport for posterior sampling, machine-learning-enhanced ensemble methods for nonlinear and non-Gaussian filtering, and transformers and attention as operators on probability measures. This measure-theoretic perspective connects practical algorithms with theoretical questions about continuum limits, permutation invariance, and generalization across ensemble sizes. During my Ph.D., I developed graph-based semi-supervised and active-learning methods for imaging and inverse problems. These methods use data geometry as a structural regularizer and adaptively select informative labels, providing a foundation for my broader interest in inference from limited observations.
Recent News
Featured Publications
Recent Publications
(2026). Variational Flow Maps: Make Some Noise for One-Step Conditional Generation. Proceedings of the 43rd International Conference on Machine Learning (ICML 2026), PMLR 306.
(2026). Learning Probabilistic Filters with Strictly Proper Scoring Rules. Submitted to the Journal of Machine Learning Research.
(2026). Flow Matching for Efficient and Scalable Data Assimilation. SIAM/ASA Journal on Uncertainty Quantification (accepted for publication).
(2026). Learning Enhanced Ensemble Filters. Journal of Computational Physics, 547, 114550.
(2024). GLL: A Differentiable Graph Learning Layer for Neural Networks. arXiv.
Recent & Upcoming Talks