SP$^3$: Spherical Priors for Plug-and-Play Restoration
About
In this paper, we introduce SP$^3$, a novel Plug-and-Play algorithm that accelerates maximum a posteriori image restoration by replacing denoisers with Spherical Encoders (SE) as generative priors. SP$^3$ approximates the intractable proximal prior step by utilizing the SE tightly structured latent space as a robust projection onto the natural image manifold. Alternating this projection with a closed-form data-consistency step, via Half-Quadratic Splitting, achieves stable convergence without requiring gradient computation during inference. This unique formulation unlocks "anytime" restoration capabilities, producing sharp, plausible images from the first iteration. Evaluations across a variety of image restoration tasks demonstrate that SP$^3$ achieves perceptual quality comparable to state-of-the-art zero-shot diffusion and flow methods while being $3$-$630\times$ faster.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Inpainting | CelebA | PSNR27.23 | 49 | |
| Super-Resolution | CelebA (test) | PSNR29.46 | 41 | |
| Inpainting | AFHQ cats | KID3.71 | 30 | |
| Denoising | AFHQ Cat | PSNR22.84 | 18 | |
| Inference Speed | Average per-image runtime | Average Runtime (sec)0.016 | 15 | |
| Deblurring | CelebA | KID7.95 | 11 | |
| Paintbrush Inpainting | CelebA (test) | KID50 | 11 | |
| Denoising | CelebA | KID57.1 | 11 | |
| Deblurring | AFHQ cats | KID2.56 | 10 | |
| Superresolution | AFHQ cats | KID3.47 | 10 |