activity
20242026
most citedPrincipled Probabilistic Imaging using Diffusion Models as Plug-and-Play Priors

2 citations · 3 across the 11 of their papers we have counts for

collaborators

11 papers

cs.CV2026

Vera: A Layered Diffusion Model for Content-Preserving Video Editing

Hongkai Zheng, Ta-Ying Cheng, Benjamin Klein +2

Video diffusion models have enabled remarkable progress in video generation and editing. However, content preservation remains a core challenge: existing methods regenerate every p…

stat.ML2026

Flow Annealing Posterior Sampling for Function-Space Regression and Inverse Problems

Yaozhong Shi, Zachary E. Ross, Yisong Yue

Principled regression for stochastic processes is a long-standing challenge with deep connections to scientific inverse problems. We introduce Flow Annealing Posterior Sampling (FL…

cs.CV2026

End-to-End Autoregressive Image Generation with 1D Semantic Tokenizer

Wenda Chu, Bingliang Zhang, Jiaqi Han +4

Autoregressive image modeling relies on visual tokenizers to compress images into compact latent representations. We design an end-to-end training pipeline that jointly optimizes r…

cs.LG2026

Beyond Pairwise Preferences: Listwise Reward-Aware Alignment for Diffusion Models

Austin Wang, Jiaqi Han, Stefano Ermon +1

Preference optimization has emerged as an efficient alternative to online reinforcement learning from human feedback (RLHF) for aligning text-to-image diffusion models. However, ex…

cs.LG2025

Blade: A Derivative-free Bayesian Inversion Method using Diffusion Priors

Hongkai Zheng, Austin Wang, Zihui Wu +3

Derivative-free Bayesian inversion arises in science and engineering applications, particularly when forward model is costly or infeasible to differentiate through. Existing deriva…

cs.LG2025

Discrete Diffusion Trajectory Alignment via Stepwise Decomposition

Jiaqi Han, Austin Wang, Minkai Xu +6

Discrete diffusion models have demonstrated great promise in modeling various sequence data, ranging from human language to biological sequences. Inspired by the success of RL in l…

cs.CV2025

STeP: A Framework for Solving Scientific Video Inverse Problems with Spatiotemporal Diffusion Priors

Bingliang Zhang, Zihui Wu, Berthy T. Feng +3

Reconstructing spatially and temporally coherent videos from time-varying measurements is a fundamental challenge in many scientific domains. A major difficulty arises from the spa…

cs.LG2025★ 1 cited

InverseBench: Benchmarking Plug-and-Play Diffusion Priors for Inverse Problems in Physical Sciences

Hongkai Zheng, Wenda Chu, Bingliang Zhang +9

Plug-and-play diffusion priors (PnPDP) have emerged as a promising research direction for solving inverse problems. However, current studies primarily focus on natural image restor…

cs.LG2025

Split Gibbs Discrete Diffusion Posterior Sampling

Wenda Chu, Zihui Wu, Yifan Chen +2

We study the problem of posterior sampling in discrete-state spaces using discrete diffusion models. While posterior sampling methods for continuous diffusion models have achieved…

cs.LG2024

Ensemble Kalman Diffusion Guidance: A Derivative-free Method for Inverse Problems

Hongkai Zheng, Wenda Chu, Austin Wang +3

When solving inverse problems, one increasingly popular approach is to use pre-trained diffusion models as plug-and-play priors. This framework can accommodate different forward mo…

eess.IV2024★ 2 cited

Principled Probabilistic Imaging using Diffusion Models as Plug-and-Play Priors

Zihui Wu, Yu Sun, Yifan Chen +3

Diffusion models (DMs) have recently shown outstanding capabilities in modeling complex image distributions, making them expressive image priors for solving Bayesian inverse proble…