← Search

HeeSun Bae

10 accepted papers

2026

Semantic-aware Wasserstein Policy Regularization for Large Language Model Alignment

ICLR 2026poster

Large language models (LLMs) are commonly aligned with human preferences using reinforcement learning from human feedback (RLHF). In this method, LLM policies are generally optimized through reward maximization with Kullback-Leibler (KL) divergence regularization of the reference policy. However, KL…

Cited by 0SourceScholar
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
2025

Preference Optimization by Estimating the Ratio of the Data Distribution

NeurIPS 2025poster

Direct preference optimization (DPO) is widely used as a simple and stable method for aligning large language models (LLMs) with human preferences. This paper investigates a generalized DPO loss that enables a policy model to match the target policy from a likelihood ratio estimation perspective. T…

Cited by 0SourcecodeScholar
2024

Dirichlet-based Per-Sample Weighting by Transition Matrix for Noisy Label Learning

ICLR 2024poster

For learning with noisy labels, the transition matrix, which explicitly models the relation between noisy label distribution and clean label distribution, has been utilized to achieve the statistical consistency of either the classifier or the risk. Previous researches have focused more on how to es…

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

Make Prompts Adaptable: Bayesian Modeling for Vision-Language Prompt Learning with Data-Dependent Prior

AAAI 2024technical

Recent vision-language pre-trained (VLP) models have become the backbone for many downstream tasks, but they are utilized as frozen model without learning. Prompt learning is a method to improve the pre-trained VLP model by adding a learnable context vector to the inputs of the text encoder. In a fe…

2024

Unknown Domain Inconsistency Minimization for Domain Generalization

ICLR 2024poster

The objective of domain generalization (DG) is to enhance the transferability of the model learned from a source domain to unobserved domains. To prevent overfitting to a specific domain, Sharpness-Aware Minimization (SAM) reduces source domain’s loss sharpness. Although SAM variants have delivered…

Cited by 2SourcePDFScholar
2023

Loss-Curvature Matching for Dataset Selection and Condensation

AISTATS 2023poster

Training neural networks on a large dataset requires substantial computational costs. Dataset reduction selects or synthesizes data instances based on the large dataset, while minimizing the degradation in generalization performance from the full dataset. Existing methods utilize the neural network…

2022

From Noisy Prediction to True Label: Noisy Prediction Calibration via Generative Model

ICML 2022spotlight

Noisy labels are inevitable yet problematic in machine learning society. It ruins the generalization of a classifier by making the classifier over-fitted to noisy labels. Existing methods on noisy label have focused on modifying the classifier during the training procedure. It has two potential prob…