Jaemoo Choi

Postdoc @ Georgia Tech

Profile

Hi! I’m a postdoctoral researcher at Georgia Tech, working at FLAIR Lab, with Prof. Yongxin Chen. I am very fortunate to be part of his group and to work with a supportive and inspiring community of researchers. I also have the pleasure of collaborating actively with Joonseok Lee and his research group. Prior to that, I received my Ph.D. in Mathematical Sciences and B.S. in Mathematics Education from Seoul National University.

My research focuses on the fundamental algorithms for generative AI (GenAI) and their applications across vision, language, and scientific domains (e.g., molecular generation). I am particularly interested in (discrete) diffusion models, flow-based methods, and (multi-modal) large language models (LLMs/MLLMs) and its applications, as well as their connections to control and dynamical systems.

Contact: jchoi843 [at] gatech [dot] edu
Follow: Google Scholar | jchoi843

Updates
[09/2025] Adjoint Schrödinger Bridge Sampler (ASBS) accepted to NeurIPS (Oral 0.3%).
[09/2025] Three papers, ASBS, NAAS, MDNS, accepted to NeurIPS.
[05/2025] Two papers, OTP, UOT-UPC, accepted to ICML.
[01/2025] Two papers, U-NOTB, DIOTM, accepted to ICLR.
[09/2024] I have joined FLAIR, Georgia Tech as a Postdoc.
[05/2024] Scalable Wasserstein Gradient Flow for Generative Modeling through Unbalanced Optimal Transport is accepted to ICML.
[01/2024] Analyzing and Improving Optimal-Transport-based Adversarial Networks is accepted to ICLR.
[09/2023] Unbalanced Optimal Transport Model (UOTM) is accepted to NeurIPS.
Selected Publications
(∗ Equal contribution, * Core contributors, † Equal advising. See Google Scholar for the full list.)
  1. Adjoint Schrödinger Bridge Sampler
    Guan-Horng Liu*, Jaemoo Choi*, Yongxin Chen, Benjamin K. Miller, Ricky T. Q. Chen*
    Advances in Neural Information Processing Systems (NeurIPS), 2025   [Oral 0.3%]
    arxiv
  2. Non-equilibrium Annealed Adjoint Sampler
    Jaemoo Choi*, Yongxin Chen, Molei Tao, Guan-Horng Liu*
    Advances in Neural Information Processing Systems (NeurIPS), 2025
    arxiv
  3. Overcoming Spurious Solutions in Semi-Dual Neural Optimal Transport: A Smoothing Approach for Learning the Optimal Transport Plan
    Jaemoo Choi, Jaewoong Choi†, Dohyun Kwon†
    International Conference on Machine Learning (ICML), 2025
    arxiv
  4. Robust barycenter estimation using semi-unbalanced neural optimal transport
    Milena Gazdieva∗, Jaemoo Choi∗, Alexander Kolesov, Jaewoong Choi, Petr Mokrov, Alexander Korotin
    International Conference on Learning Representations (ICLR), 2025
    arxiv | code
  5. Improving Neural Optimal Transport via Displacement Interpolation
    Jaemoo Choi, Yongxin Chen, Jaewoong Choi
    International Conference on Learning Representations (ICLR), 2025
    arxiv
  6. Scalable Wasserstein Gradient Flow for Generative Modeling through Unbalanced Optimal Transport
    Jaemoo Choi∗, Jaewoong Choi∗, Myungjoo Kang
    International Conference on Machine Learning (ICML), 2024
    arxiv | code
  7. Analyzing and Improving OT-based Adversarial Networks
    Jaemoo Choi∗, Jaewoong Choi∗, Myungjoo Kang
    International Conference on Learning Representations (ICLR), 2024
    arxiv | code
  8. Generative Modeling through the Semi-dual Formulation of Unbalanced Optimal Transport
    Jaemoo Choi∗, Jaewoong Choi∗, Myungjoo Kang
    Advances in Neural Information Processing Systems (NeurIPS), 2023
    arxiv | code
  9. Restoration based Generative Models
    Jaemoo Choi∗, Yesom Park∗, Myungjoo Kang
    International Conference on Machine Learning (ICML), 2023
    arxiv | code