Saved in:
Bibliographic Details
Main Authors: Zheng, Léon, Hirtz, Thomas, Janati, Yazid, Moulines, Eric
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2602.07102
Tags: Add Tag
No Tags, Be the first to tag this record!
Table of 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.