A convergent Plug-and-Play Majorization-Minimization algorithm for Poisson inverse problems
Organizations: CREATIS · INSA-Lyon, CREATIS UMR 5220 U1294, Villeurbanne, F-69621, France · ICJ, MMCS · INSA-Lyon, CNRS UMR 5208, Institut Camille Jordan, Villeurbanne, F-69621, France.
Abstract
In this paper, we present a novel variational plug-and-play algorithm for Poisson inverse problems. Our approach minimizes an explicit functional which is the sum of a Kullback-Leibler data fidelity term and a regularization term based on a pre-trained neural network. By combining classical likelihood maximization methods with recent advances in gradient-based denoisers, we allow the use of pre-trained Gaussian denoisers without sacrificing convergence guarantees. The algorithm is formulated in the majorization-minimization framework, which guarantees convergence to a stationary point. Numerical experiments confirm state-of-the-art performance in deconvolution and tomography under moderate noise, and demonstrate clear superiority in high-noise conditions, making this method particularly valuable for nuclear medicine applications.
Figures & tables
| Kernels | Gaussian | Levin1 | Levin4 | Uniform | Average | |||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Methods | PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS | |
| 60 | RL-TV [ 11 ] | 24.78 | 0.610 | 0.425 | 23.33 | 0.520 | 0.493 | 24.00 | 0.544 | 0.440 | 23.80 | 0.568 | 0.474 | 23.98 | 0.561 | 0.458 |
| DPS [ 36 ] | 20.41 | 0.469 | 0.549 | 21.98 | 0.545 | 0.382 | 22.85 | 0.580 | 0.380 | 18.56 | 0.404 | 0.567 | 20.95 | 0.499 | 0.469 | |
| Prox-DiffPIR [ 37 ] | 26.04 | 0.688 | 0.337 | 25.31 | 0.632 | 0.307 | 26.31 | 0.676 | 0.255 | 24.97 | 0 .630 | 0.391 | 25.66 | 0.656 | 0.322 | |
| PhD-Net [ 38 ] | 26.00 | 0.685 | 0.420 | 24.56 | 0.608 | 0.424 | 25.81 | 0.665 | 0.424 | 24.70 | 0.618 | 0.503 | 25.26 | 0.644 | 0.455 | |
| DPIR [ 17 ] | 2 6.46 | 0 .701 | 0.403 | 2 6.00 | 0.663 | 0.420 | 2 7.03 | 0.711 | 0.350 | 2 4.78 | 0.618 | 0.515 | 2 6.07 | 0 .673 | 0.422 | |
| 192 angles | 128 angles | 64 angles | ||||||||
| Methods | PSNR | SSIM | MAE (HU) | PSNR | SSIM | MAE (HU) | PSNR | SSIM | MAE (HU) | |
| 0.005 | FBP | 17.39 0.72 | 0.119 0.023 | 290 31.2 | 16.60 0.71 | 0.103 0.021 | 318 34.0 | 15.07 0.66 | 0.077 0.017 | 378 38.9 |
| MLEM [ 3 ] | 28.72 1.06 | 0.772 0.028 | 62.7 10.2 | 28.67 1.06 | 0.765 0.028 | 63.1 10.3 | 28.27 1.04 | 0.735 0.032 | 66.7 10.7 | |
| PDNet [ 50 ] | 3 7.28 0.97 | 0 .897 0.018 | 2 4.7 3.2 | 3 7.05 0.99 | 0 .893 0.018 | 2 5.2 3.2 | 3 4.58 1.08 | 0 .853 0.022 | 3 3.4 4.3 | |
| FBP-SwinIR [ 51 ] | 37.66 1.01 | 0.903 0.018 | 23.50 3.1 | 37.31 1.01 | 0.898 0.018 | 24.4 3.2 | 31.82 1.53 | 0.796 0.040 | 45.3 8.0 | |
| PnP-MM | 36.40 1.09 | 0.892 0.017 | 26.53 3.3 | 36.44 0.94 | 0.888 0.016 | 26.9 3.1 | 35.64 1.19 | 0.873 0.020 | 28.6 4.0 | |
| Noise level ( ) | 1 | 5 | 10 | 20 | 30 | 40 | 50 |
|---|---|---|---|---|---|---|---|
| NRMSE (%) |
| 192 angles | 128 angles | 64 angles | ||||||||
| Method | Parameter | |||||||||
| MLEM | Iterations | 40 | 35 | 20 | 40 | 35 | 20 | 20 | 20 | 20 |
| PnP-MM | 0.12 | 0.20 | 0.20 | 0.10 | 0.15 | 0.15 | 0.10 | 0.10 | 0.10 | |
| 4 | 4 | 8 | 5 | 5 | 8 | 5 | 8 | 8 | ||
| 1 | 1 | 1 | 1 | 1 | 1 | 1 | 1 | 1 | ||
| Iterations | 600 | 600 | 600 | 600 | 600 | 600 | 600 | 600 | 600 | |