← Search

JaeWoong Choi

15 accepted papers

2026

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

ICML 2026poster

Diffusion models often yield highly curved trajectories and noisy score targets due to an uninformative, memoryless forward process that induces independent data-noise coupling. We propose Adjoint Schrödinger Bridge Matching (ASBM), a generative modeling framework that recovers optimal trajectories …

Cited by 0SourceScholar
2025

APT: Adaptive Personalized Training for Diffusion Models with Limited Data

CVPR 2025poster

Personalizing diffusion models using limited data presents significant challenges, including overfitting, loss of prior knowledge, and degradation of text alignment. Overfitting leads to shifts in the noise prediction distribution, disrupting the denoising trajectory and causing the model to lose se…

Cited by 0SourcePDFScholar
2025

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

ICML 2025poster

We address the convergence problem in learning the Optimal Transport (OT) map, where the OT Map refers to a map from one distribution to another while minimizing the transport cost. Semi-dual Neural OT, a widely used approach for learning OT Maps with neural networks, often generates spurious soluti…

Cited by 0SourcePDFScholar
2025

Robust Barycenter Estimation using Semi-Unbalanced Neural Optimal Transport

ICLR 2025poster

Aggregating data from multiple sources can be formalized as an *Optimal Transport* (OT) barycenter problem, which seeks to compute the average of probability distributions with respect to OT discrepancies. However, in real-world scenarios, the presence of outliers and noise in the data measures can…

2025

Unpaired Point Cloud Completion via Unbalanced Optimal Transport

ICML 2025poster

Unpaired point cloud completion is crucial for real-world applications, where ground-truth data for complete point clouds are often unavailable. By learning a completion map from unpaired incomplete and complete point cloud data, this task avoids the reliance on paired datasets. In this paper, we pr…

Cited by 0SourcePDFScholar
2024

Scalable Wasserstein Gradient Flow for Generative Modeling through Unbalanced Optimal Transport

ICML 2024poster

Wasserstein gradient flow (WGF) describes the gradient dynamics of probability density within the Wasserstein space. WGF provides a promising approach for conducting optimization over the probability distributions. Numerically approximating the continuous WGF requires the time discretization method.…

Cited by 10SourcePDFScholar
2023

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

ICLR 2023poster

The ideally disentangled latent space in GAN involves the global representation of latent space using semantic attribute coordinates. In other words, in this disentangled space, there exists the global semantic basis as a vector space where each basis component describes one attribute of generated…

Cited by 3SourcePDFScholar
2023

Generative Modeling through the Semi-dual Formulation of Unbalanced Optimal Transport

NeurIPS 2023poster

Optimal Transport (OT) problem investigates a transport map that bridges two distributions while minimizing a given cost function. In this regard, OT between tractable prior distribution and data has been utilized for generative modeling tasks. However, OT-based methods are susceptible to outliers a…

2023

MAGANet: Achieving Combinatorial Generalization by Modeling a Group Action

ICML 2023poster

Combinatorial generalization refers to the ability to collect and assemble various attributes from diverse data to generate novel unexperienced data. This ability is considered a necessary passing point for achieving human-level intelligence. To achieve this ability, previous unsupervised approaches…

Cited by 7SourcePDFScholar
2023

Understanding the Latent Space of Diffusion Models through the Lens of Riemannian Geometry

NeurIPS 2023poster

Despite the success of diffusion models (DMs), we still lack a thorough understanding of their latent space. To understand the latent space $\mathbf{x}_t \in \mathcal{X}$, we analyze them from a geometrical perspective. Our approach involves deriving the local latent basis within $\mathcal{X}$ by le…

2022

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

ICLR 2022poster

The discovery of the disentanglement properties of the latent space in GANs motivated a lot of research to find the semantically meaningful directions on it. In this paper, we suggest that the disentanglement property is closely related to the geometry of the latent space. In this regard, we propose…

2022

Style-Guided and Disentangled Representation for Robust Image-to-Image Translation

AAAI 2022technical

Recently, various image-to-image translation (I2I) methods have improved mode diversity and visual quality in terms of neural networks or regularization terms. However, conventional I2I methods relies on a static decision boundary and the encoded representations in those methods are entangled with e…