Papers & Preprints

C: Conference, W: Workshop, J: Journal, P: Preprint / * Equal contribution, † Corresponding author 

Names in bold = Group members


[P6] Unlearning for One-Step Generative Models via Unbalanced Optimal Transport

Hyundo Choi*, Junhyeong An*, Jinseong Park†, Jaewoong Choi

Submitted.


[P5] Conditional Unbalanced Optimal Transport Maps: An Outlier-Robust Framework for Conditional Generative Modeling

Jiwoo Yoon*, Kyumin Choi*, Jaewoong Choi

Submitted.


[P4] Rate-Optimal Noise Annealing in Semi-Dual Neural Optimal Transport: Tangential Identifiability, Off-Manifold Ambiguity, and Guaranteed Recovery

Raymond Chu, Jaewoong Choi, Dohyun Kwon†

Submitted.


[P3] Neural Optimal Transport in Hilbert Spaces: Characterizing Spurious Solutions and Gaussian Smoothing

Jae-Hwan Choi, Jiwoo Yoon, Dohyun Kwon†, Jaewoong Choi

Submitted.


[P2] Reweighted Flow Matching via Unbalanced OT for Label-free Long-tailed Generation

Hyunsoo Song, Minjung Gim, Jaewoong Choi

Submitted.

Preliminary version: NeurIPS 2025 Workshop on Reliable ML from Unreliable Data


[P1] Scalable Simulation-free Entropic Unbalanced Optimal Transport

Jaemoo Choi, Jaewoong Choi

Submitted.



[C13] Efficient Generative Modeling beyond Memoryless Diffusion via Adjoint Schrödinger Bridge Matching

Jeongwoo Shin, Jinhwan Sul, Joonseok Lee†, Jaewoong Choi, Jaemoo Choi

Proceedings of the 43rd International Conference on Machine Learning (ICML), 2026


[C12] UOTIP: Unbalanced Optimal Transport Map for Unpaired Inverse Problems

Donggyu Lee*, Taekyung Lee*, Jaewoong Choi

Proceedings of the 43rd International Conference on Machine Learning (ICML), 2026

Preliminary version: NeurIPS 2025 Workshop on Structured Probabilistic Inference & Generative Modeling


[C11] Overcoming Spurious Solutions in Semi-Dual Neural Optimal Transport: A Smoothing Approach for Learning the Optimal Transport Plan

Jaemoo Choi, Jaewoong Choi†, Dohyun Kwon†

Proceedings of the 42nd International Conference on Machine Learning (ICML), 2025


[C10] Unpaired Point Cloud Completion via Unbalanced Optimal Transport

Taekyung Lee, Jaemoo Choi, Jaewoong Choi†, Myungjoo Kang†

Proceedings of the 42nd International Conference on Machine Learning (ICML), 2025

Preliminary version: ICLR 2025 Workshop.


[C9] Improving Neural Optimal Transport via Displacement Interpolation

Jaemoo Choi, Yongxin Chen, Jaewoong Choi

Proceedings of the International Conference on Learning Representations (ICLR), 2025


[C8] Robust Barycenter Estimation using Semi-Unbalanced Neural Optimal Transport

Milena Gazdieva*, Jaemoo Choi*, Alexander Kolesov, Jaewoong Choi, Petr Mokrov, Alexander Korotin

Proceedings of the International Conference on Learning Representations (ICLR), 2025


[J2] Analyzing the Latent Space of GAN through Local Dimension Estimation for Disentanglement Evaluation

Jaewoong Choi, Geonho Hwang, Hyunsoo Cho, Myungjoo Kang†

Pattern Recognition, 2025


[C7] Scalable Wasserstein Gradient Flow for Generative Modeling through Unbalanced Optimal Transport

Jaemoo Choi*, Jaewoong Choi*, Myungjoo Kang†

Proceedings of the 41st International Conference on Machine Learning (ICML), 2024


[C6] Analyzing and Improving Optimal-Transport-based Adversarial Networks

Jaemoo Choi*, Jaewoong Choi*, Myungjoo Kang†

Proceedings of the International Conference on Learning Representations (ICLR), 2024


[C5] 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


[C4] Understanding the Latent Space of Diffusion Models through the Lens of Riemannian Geometry

Yong-Hyun Park*, Mingi Kwon*, Jaewoong Choi, Junghyo Jo, Youngjung Uh

Advances in Neural Information Processing Systems (NeurIPS), 2023


[C3] MAGANet: Achieving Combinatorial Generalization by Modeling a Group Action

Geonho Hwang, Jaewoong Choi, Hyunsoo Cho, Myungjoo Kang†

Proceedings of the 40th International Conference on Machine Learning (ICML), 2023


[C2] Finding the Global Semantic Representation in GAN through Fréchet Mean

Jaewoong Choi, Geonho Hwang, Hyunsoo Cho, Myungjoo Kang

Proceedings of the International Conference on Learning Representations (ICLR), 2023


[J1] Disentangling the correlated continuous and discrete generative factors of data

Jaewoong Choi, Geonho Hwang, Myungjoo Kang

Pattern Recognition, 2023


[C1] Do Not Escape From the Manifold: Discovering the Local Coordinates on the Latent Space of GANs

Jaewoong Choi*, Junho Lee*, Changyeon Yoon, Jung Ho Park, Geonho Hwang, Myungjoo Kang

Proceedings of the International Conference on Learning Representations (ICLR), 2022


[W1] Attention routing between capsules

Jaewoong Choi, Hyun Seo, Suii Im, Myungjoo Kang

Proceedings of the IEEE/CVF International Conference on Computer Vision Workshops (ICCVW), 2019