← Search

Wanmo Kang

12 accepted papers

2025

Diffusion Adaptive Text Embedding for Text-to-Image Diffusion Models

NeurIPS 2025poster

Text-to-image diffusion models rely on text embeddings from a pre-trained text encoder, but these embeddings remain fixed across all diffusion timesteps, limiting their adaptability to the generative process. We propose Diffusion Adaptive Text Embedding (DATE), which dynamically updates text embeddi…

Cited by 0SourcecodeScholar
2024

Diffusion Rejection Sampling

ICML 2024poster

Recent advances in powerful pre-trained diffusion models encourage the development of methods to improve the sampling performance under well-trained diffusion models. This paper introduces Diffusion Rejection Sampling (DiffRS), which uses a rejection sampling scheme that aligns the sampling transiti…

2024

Label-Noise Robust Diffusion Models

ICLR 2024poster

Conditional diffusion models have shown remarkable performance in various generative tasks, but training them requires large-scale datasets that often contain noise in conditional inputs, a.k.a. noisy labels. This noise leads to condition mismatch and quality degradation of generated data. This pape…

2024

Training Unbiased Diffusion Models From Biased Dataset

ICLR 2024poster

With significant advancements in diffusion models, addressing the potential risks of dataset bias becomes increasingly important. Since generated outputs directly suffer from dataset bias, mitigating latent bias becomes a key factor in improving sample quality and proportion. This paper proposes tim…

2023

Refining Generative Process with Discriminator Guidance in Score-based Diffusion Models

ICML 2023oral

The proposed method, **Discriminator Guidance**, aims to improve sample generation of pre-trained diffusion models. The approach introduces a discriminator that gives explicit supervision to a denoising sample path whether it is realistic or not. Unlike GANs, our approach does not require joint trai…

2023

SAAL: Sharpness-Aware Active Learning

ICML 2023poster

While deep neural networks play significant roles in many research areas, they are also prone to overfitting problems under limited data instances. To overcome overfitting, this paper introduces the first active learning method to incorporate the sharpness of loss space into the acquisition function…

2022

Maximum Likelihood Training of Implicit Nonlinear Diffusion Model

NeurIPS 2022accept

Whereas diverse variations of diffusion models exist, extending the linear diffusion into a nonlinear diffusion process is investigated by very few works. The nonlinearity effect has been hardly understood, but intuitively, there would be promising diffusion patterns to efficiently train the generat…

2022

Soft Truncation: A Universal Training Technique of Score-based Diffusion Model for High Precision Score Estimation

ICML 2022spotlight

Recent advances in diffusion models bring state-of-the-art performance on image generation tasks. However, empirical results from previous research in diffusion models imply an inverse correlation between density estimation and sample generation performances. This paper investigates with sufficient…

2021

Counterfactual Fairness with Disentangled Causal Effect Variational Autoencoder

AAAI 2021technical

The problem of fair classification can be mollified if we develop a method to remove the embedded sensitive information from the classification features. This line of separating the sensitive information is developed through the causal inference, and the causal inference enables the counterfactual g…

Cited by 65SourcePDFScholar
2021

Improved Regret Bounds of Bilinear Bandits using Action Space Analysis

ICML 2021spotlight

We consider the bilinear bandit problem where the learner chooses a pair of arms, each from two different action spaces of dimension $d_1$ and $d_2$, respectively. The learner then receives a reward whose expectation is a bilinear function of the two chosen arms with an unknown matrix parameter $\Th…

Cited by 10SourcePDFScholar
2020

Mixout: Effective Regularization to Finetune Large-scale Pretrained Language Models

ICLR 2020poster

In natural language processing, it has been observed recently that generalization could be greatly improved by finetuning a large-scale language model pretrained on a large unlabeled corpus. Despite its recent success and wide adoption, finetuning a large pretrained language model on a downstream ta…

Cited by 242SourcecodeScholar