← Search

Kiyoung Om

3 accepted papers

2026

Diffusion Alignment as Variataional Expectation-Maximization

ICLR 2026poster

Diffusion alignment aims to optimize diffusion models for the downstream objective. While existing methods based on reinforcement learning or direct backpropagation achieve considerable success in maximizing rewards, they often suffer from reward over-optimization and mode collapse. We introduce Dif…

Cited by 0SourcecodeScholar
2026

Diffusion Fine-Tuning via Reparameterized Policy Gradient of the Soft Q-Function

ICLR 2026poster

Diffusion models excel at generating high-likelihood samples but often require alignment with downstream objectives. Existing fine-tuning methods for diffusion models significantly suffer from reward over-optimization, resulting in high-reward but unnatural samples and degraded diversity. To mitigat…

Cited by 0SourceScholar
2025

Posterior Inference with Diffusion Models for High-dimensional Black-box Optimization

ICML 2025poster

Optimizing high-dimensional and complex black-box functions is crucial in numerous scientific applications. While Bayesian optimization (BO) is a powerful method for sample-efficient optimization, it struggles with the curse of dimensionality and scaling to thousands of evaluations. Recently, lever…