← Search

Byeonghu Na

20 accepted papers

2026

AC-Sampler: Accelerate and Correct Diffusion Sampling with Metropolis-Hastings Algorithm

ICLR 2026poster

Diffusion-based generative models have recently achieved state-of-the-art performance in high-fidelity image synthesis. These models learn a sequence of denoising transition kernels that gradually transform a simple prior distribution into a complex data distribution. However, requiring many transit…

Cited by 0SourcecodeScholar
2026

AMiD: Knowledge Distillation for LLMs with $\alpha$-mixture Assistant Distribution

ICLR 2026poster

Autoregressive large language models (LLMs) have achieved remarkable improvement across many tasks but incur high computational and memory costs. Knowledge distillation (KD) mitigates this issue by transferring knowledge from a large teacher to a smaller student through distributional alignment. Pre…

Cited by 0SourceScholar
2026

Distillation of Large Language Models via Concrete Score Matching

ICLR 2026poster

Large language models (LLMs) deliver remarkable performance but are costly to deploy, motivating knowledge distillation (KD) for efficient inference. Existing KD objectives typically match student and teacher probabilities via softmax, which blurs valuable logit information. While direct logit disti…

Cited by 0SourceScholar
2026

Lookahead Sample Reward Guidance for Test-Time Scaling of Diffusion Models

ICML 2026spotlight

Diffusion models have demonstrated strong generative performance; however, generated samples often fail to fully align with human intent. This paper studies a test-time scaling method that enables sampling from regions with higher human-aligned reward values. Existing gradient guidance methods appro…

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

Diffusion Bridge AutoEncoders for Unsupervised Representation Learning

ICLR 2025spotlight

Diffusion-based representation learning has achieved substantial attention due to its promising capabilities in latent representation and sample generation. Recent studies have employed an auxiliary encoder to identify a corresponding representation from data and to adjust the dimensionality of a la…

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
2025

Training-Free Safe Text Embedding Guidance for Text-to-Image Diffusion Models

NeurIPS 2025poster

Text-to-image models have recently made significant advances in generating realistic and semantically coherent images, driven by advanced diffusion models and large-scale web-crawled datasets. However, these datasets often contain inappropriate or biased content, raising concerns about the generatio…

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

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

Reward-based Input Construction for Cross-document Relation Extraction

ACL 2024long

Relation extraction (RE) is a fundamental task in natural language processing, aiming to identify relations between target entities in text. While many RE methods are designed for a single sentence or document, cross-document RE has emerged to address relations across multiple long documents. Given…

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…

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

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

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…

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

Multi-modal Text Recognition Networks: Interactive Enhancements between Visual and Semantic Features

ECCV 2022poster

"Linguistic knowledge has brought great benefits to scene text recognition by providing semantics to refine character sequences. However, since linguistic knowledge has been applied individually on the output sequence, previous methods have not fully utilized the semantics to understand visual clues…

2022

Unknown-Aware Domain Adversarial Learning for Open-Set Domain Adaptation

NeurIPS 2022accept

Open-Set Domain Adaptation (OSDA) assumes that a target domain contains unknown classes, which are not discovered in a source domain. Existing domain adversarial learning methods are not suitable for OSDA because distribution matching with $\textit{unknown}$ classes leads to negative transfer. Previ…