mlsanityData to 7 Oct 2026

Hallucination-Aware Diffusion Sampling for Inverse Problems via Robust Prior Updates

arXiv 2606.02331 ↗First posted 1 Jun 2026cs.CV cs.LGcited by 0

On the map: Diffusion models → Inverse problems and image restoration

Abstract (arXiv)

Diffusion-based inverse problem solvers can produce realistic reconstructions, but realism alone does not ensure that the recovered details are supported by the measurement. We study this failure as measurement-conditioned hallucination: visually meaningful content that is either implausible or inconsistent with the measured instance. Our analysis separates Bayes-rule-based diffusion inverse solvers into a prior update and a measurement-conditioning step, showing that hallucinated content can enter through the prior-side proposal before the measurement correction is applied. Motivated by this view, we propose Robust Prior Update (RPU), a solver-level module that probes the local stability of the diffusion prior update, re-anchors the resulting displacement at the current iterate, and leaves the measurement update unchanged. We instantiate RPU in DPS and evaluate it on FFHQ and ImageNet inverse problems using automatic metrics and human faithfulness studies. On FFHQ, RPU improves PSNR and LPIPS over DPS across box inpainting, Gaussian deblurring, and motion deblurring. In human judgments, RPU receives 91.9% of blind non-tie majority preferences and 91.1% of ground-truth-assisted non-tie preferences on FFHQ box inpainting, while the ImageNet Gaussian reader study is tie-heavy but favors RPU among non-tie cases. These results support a targeted claim: robustifying the prior update can improve instance faithfulness in diffusion inverse solvers, especially when the prior shapes weakly constrained content.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Inverse problems and image restoration, one of 14 directions in Diffusion models. Builds on 28 of its 39 references that are in the map. First posted on arXiv 1 Jun 2026.

Builds on

28 of 39 references.

Closest ancestors · 7

Same direction, newest first

  1. 2024A Survey on Diffusion Models for Inverse Problems30 Sep
  2. 2023Denoising Diffusion Models for Plug-and-Play Image Restoration15 May
  3. 2023A Variational Perspective on Solving Inverse Problems with Diffusion Models7 May
  4. 2022Zero-Shot Image Restoration Using Denoising Diffusion Null-Space Model1 Dec
  5. 2022Diffusion Posterior Sampling for General Noisy Inverse Problems29 Sep
  6. 2022Improving Diffusion Models for Inverse Problems using Manifold Constraints2 Jun
  7. 2022Denoising Diffusion Restoration Models27 Jan

Other references · 21

  1. 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
  2. 2017Towards Deep Learning Models Resistant to Adversarial Attackscited by 6,326
  3. 2022Survey of Hallucination in Natural Language Generationcited by 2,056
  4. 2021Image Super-Resolution via Iterative Refinementcited by 1,506
  5. 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,404
  6. 2014Image Super-Resolution Using Deep Convolutional Networkscited by 1,218
  7. 2017Virtual Adversarial Training: A Regularization Method for Supervised and Semi-Supervised Learningcited by 918
  8. 2021Solving Inverse Problems in Medical Imaging with Score-Based Generative Modelscited by 514
  9. 2017Compressed Sensing using Generative Modelscited by 337
  10. 2021Score-based diffusion models for accelerated MRIcited by 308
  11. 2023Practical and Asymptotically Exact Conditional Sampling in Diffusion Modelscited by 218
  12. 2024Understanding Hallucinations in Diffusion Models through Mode Interpolationcited by 87
9 more
  1. 2020On hallucinations in tomographic image reconstructioncited by 66
  2. 2022Solving Audio Inverse Problems with a Diffusion Modelcited by 62
  3. 2020The troublesome kernel -- On hallucinations, no free lunches and the accuracy-stability trade-off in inverse problemscited by 40
  4. 2024Hallucination Index: An Image Quality Metric for Generative Reconstruction Modelscited by 25
  5. 2022Theoretical Perspectives on Deep Learning Methods in Inverse Problemscited by 22
  6. 2024Looks Too Good To Be True: An Information-Theoretic Analysis of Hallucinations in Generative Restoration Modelscited by 12
  7. 2025Tackling Hallucination from Conditional Models for Medical Image Reconstruction with DynamicDPScited by 10
  8. 2025RODS: Robust Optimization Inspired Diffusion Sampling for Detecting and Reducing Hallucination in Generative Modelscited by 3
  9. 2024Two-Stage Distributionally Robust Optimization: Intuitive Understanding and Algorithm Development from the Primal Perspectivecited by 2

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026HAD: Hallucination-Aware Diffusion Priors for 3D Reconstruction
  2. 2025DAPS++: Rethinking Diffusion Inverse Problems with Decoupled Posterior Annealing
  3. 2026Posterior Continuation with Noise-Conditioned Frequency Exposure for Diffusion Inverse Problems
  4. 2024Enhancing Diffusion Posterior Sampling for Inverse Problems by Integrating Crafted Measurements
  5. 2025MACS: Measurement-Aware Consistency Sampling for Inverse Problems
  6. 2023DiracDiffusion: Denoising and Incremental Reconstruction with Assured Data-Consistency
  7. 2024Bayesian Conditioned Diffusion Models for Inverse Problems
  8. 2025Diffusion models for inverse problems