Research map / Image generation / Fast diffusion sampling
Self-diffusion for Solving Inverse Problems
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes self-diffusion, an iterative noising-and-denoising scheme that trains a randomly initialized CNN self-denoiser with a data-fidelity loss at each noise level, solving inverse problems without pretrained generative models.
Problem
Solving linear inverse problems (MRI reconstruction, low-level vision, radar angle estimation, compressed sensing) without pretrained generative priors or training data, where existing diffusion-based methods need a clean-data pretrained model.
Approach
Replace the pretrained denoiser in diffusion-based inverse solvers with a self-denoiser: a randomly initialized CNN re-trained at each noise level to minimize measurement misfit A·D(x_t + σ(t)ε) − y, with noise resampled across steps so the spectral bias of the network is modulated by a decreasing noise schedule.
Key results
- On 4x MRI with 20 ACS lines, SDI reaches 39.21 dB PSNR / 0.9640 SSIM, versus 38.35 dB for DIP and 32.18 dB for Aseq.
- On low-level vision (Table), SDI reaches 30.78 dB PSNR on inpainting and 27.76 dB on motion deblurring, above DIP (29.41 and 25.94).
- On CBSD68 denoising (sigma=25), SDI gets 28.10 dB PSNR, below FFDNet (31.22) and matching IR-SDE (28.09) on PSNR but worse on LPIPS (0.207 vs 0.101).
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| apply SDI instead of IMJENSE | fastMRI 2D reconstruction · fastMRI 2D, 4× undersampled | PSNR: 38.80 -> 39.21 | Helped |
| apply SDI instead of IMJENSE | fastMRI 2D reconstruction · fastMRI 2D, 6× undersampled | PSNR: 35.30 -> 36.84 | Helped |
| apply SDI instead of Aseq | fastMRI 2D reconstruction · fastMRI 2D, 4× undersampled without ACS | PSNR: 30.87 -> 38.20 | Helped |
| apply SDI instead of DIP | image restoration · ImageNet 2× SR | PSNR: 29.06 -> 30.30 | Helped |
| apply SDI instead of DIP | image restoration · ImageNet 4× SR | PSNR: 24.55 -> 25.82 | Helped |
| apply SDI instead of DIP | image restoration · ImageNet motion deblurring | PSNR: 25.94 -> 27.76 | Helped |
| apply SDI instead of DIP | image restoration · ImageNet inpainting | PSNR: 29.41 -> 30.78 | Helped |
| apply SDI instead of FFDNET | CBSD68 denoising · CBSD68, σ = 25 | PSNR: 31.22 -> 28.10 | Hurt |
| remove noise resampling (ablation) | SDI · fastMRI 2D, 4× undersampled | PSNR: 39.21 -> 39.09 | Hurt |
| remove noise resampling (ablation) | SDI · fastMRI 2D, 6× undersampled | PSNR: 36.84 -> 36.21 | Hurt |
| remove noise resampling (ablation) | SDI · fastMRI 2D, 4× undersampled without ACS | PSNR: 38.20 -> 37.12 | Hurt |
| remove noise resampling (ablation) | SDI · ImageNet 2× SR | PSNR: 26.97 -> 30.30 | Hurt |
Surprising findings
- On denoising, SDI's LPIPS (0.207) is much worse than IR-SDE (0.101) and its SSIM is lower than DIP's PSNR-level advantage would suggest, showing mixed perceptual quality.
- Against a pretrained CSGM on T2 MRI at 4x and 6x acceleration, CSGM is better, and SDI only wins in the no-ACS setting.
Limitations
- Per-instance optimization is slow: about 1.52 iterations/sec on a 5000x3000 image, requiring 20000 total iterations.
- Denoising results trail supervised FFDNet and generative IR-SDE, especially on LPIPS.
- Pretrained diffusion model CSGM outperforms SDI on T2 MRI with ACS lines, so benefits are setting-dependent.
Experimental setup
Datasets: fastMRI ImageNet CBSD68 RADIal
Benchmarks: fastMRI ImageNet 256x256 (inpainting, deblurring, SR) CBSD68 denoising RADIal radar
Metrics: PSNR SSIM NRMSE LPIPS
Baselines: Deep Image Prior Aseq IMJENSE CSGM ADMM-BP IR-SDE FFDNet DPS DDNM
Builds on: Deep Image Prior CSGM neural tangent kernel DDNM
Closest papers by meaning
- A Variational Perspective on Solving Inverse Problems with Diffusion Models (ICLR 2024 · similarity 0.91)
- Adapt and Diffuse: Sample-adaptive Reconstruction via Latent Diffusion Models (ICML 2024 · similarity 0.90)
- Improving Diffusion-based Inverse Algorithms under Few-Step Constraint via Linear Extrapolation (NeurIPS 2025 · similarity 0.90)
- InvFusion: Bridging Supervised and Zero-shot Diffusion for Inverse Problems (NeurIPS 2025 · similarity 0.90)
- Improving Diffusion Inverse Problem Solving with Decoupled Noise Annealing (CVPR 2025 · similarity 0.90)
- Ambient Diffusion Posterior Sampling: Solving Inverse Problems with Diffusion Models Trained on Corrupted Data (ICLR 2025 · similarity 0.90)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
