Noise-Adaptive Diffusion Sampling for Inverse Problems Without Task-Specific Tuning

arXiv:2604.16919v1 Announce Type: cross Abstract: Diffusion models (DMs) have recently shown remarkable performance on inverse problems (IPs). Optimization-based methods can fast solve IPs using DMs as powerful regularizers, but they are susceptible to local minima and noise overfitting. Although DMs can provide strong priors for Bayesian approaches, enforcing measurement consistency during the denoising process leads to manifold infeasibility issues. We propose Noise-space Hamiltonian Monte Carlo (N-HMC), a posterior sampling method that treats reverse diffusion as a deterministic mapping from initial noise to clean images. N-HMC enables comprehensive exploration of the solution space, avoiding local optima. By moving inference entirely into the initial-noise space, N-HMC keeps proposals on the learned data manifold. We provide a comprehensive theoretical analysis of our approach and extend the framework to a noise-adaptive variant (NA-NHMC) that effectively handles IPs with unknown noise type and level. Extensive experiments across four linear and three nonlinear inverse problems demonstrate that NA-NHMC achieves superior reconstruction quality with robust performance across different hyperparameters and initializations, significantly outperforming recent state-of-the-art methods. The code is available at https://github.com/NA-HMC/NA-HMC.

Related

• DiffusionPrint: Learning Generative Fingerprints for Diffusion-Based Inpainting Localization
• Remote Sensing Image Super-Resolution for Imbalanced Textures: A Texture-Aware Diffusion Framework
• Structure-Adaptive Sparse Diffusion in Voxel Space for 3D Medical Image Enhancement
• Domain-Specific Latent Representations Improve the Fidelity of Diffusion-Based Medical Image Super-Resolution
• Degradation-Aware and Structure-Preserving Diffusion for Real-World Image Super-Resolution

Source: arXiv cs.CV | 2026-04-21