Fast and Robust Likelihood-Guided Diffusion Posterior Sampling with Amortized Variational Inference

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zheng, Léon, Hirtz, Thomas, Janati, Yazid, Moulines, Eric
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910014528028672
author Zheng, Léon
Hirtz, Thomas
Janati, Yazid
Moulines, Eric
author_facet Zheng, Léon
Hirtz, Thomas
Janati, Yazid
Moulines, Eric
contents Zero-shot diffusion posterior sampling offers a flexible framework for inverse problems by accommodating arbitrary degradation operators at test time, but incurs high computational cost due to repeated likelihood-guided updates. In contrast, previous amortized diffusion approaches enable fast inference by replacing likelihood-based sampling with implicit inference models, but at the expense of robustness to unseen degradations. We introduce an amortization strategy for diffusion posterior sampling that preserves explicit likelihood guidance by amortizing the inner optimization problems arising in variational diffusion posterior sampling. This accelerates inference for in-distribution degradations while maintaining robustness to previously unseen operators, thereby improving the trade-off between efficiency and flexibility in diffusion-based inverse problems.
format Preprint
id arxiv_https___arxiv_org_abs_2602_07102
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Fast and Robust Likelihood-Guided Diffusion Posterior Sampling with Amortized Variational Inference
Zheng, Léon
Hirtz, Thomas
Janati, Yazid
Moulines, Eric
Machine Learning
Artificial Intelligence
Zero-shot diffusion posterior sampling offers a flexible framework for inverse problems by accommodating arbitrary degradation operators at test time, but incurs high computational cost due to repeated likelihood-guided updates. In contrast, previous amortized diffusion approaches enable fast inference by replacing likelihood-based sampling with implicit inference models, but at the expense of robustness to unseen degradations. We introduce an amortization strategy for diffusion posterior sampling that preserves explicit likelihood guidance by amortizing the inner optimization problems arising in variational diffusion posterior sampling. This accelerates inference for in-distribution degradations while maintaining robustness to previously unseen operators, thereby improving the trade-off between efficiency and flexibility in diffusion-based inverse problems.
title Fast and Robust Likelihood-Guided Diffusion Posterior Sampling with Amortized Variational Inference
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2602.07102