A Neural-Network-Based Convex Regularizer for Inverse Problems
About
The emergence of deep-learning-based methods to solve image-reconstruction problems has enabled a significant increase in reconstruction quality. Unfortunately, these new methods often lack reliability and explainability, and there is a growing interest to address these shortcomings while retaining the boost in performance. In this work, we tackle this issue by revisiting regularizers that are the sum of convex-ridge functions. The gradient of such regularizers is parameterized by a neural network that has a single hidden layer with increasing and learnable activation functions. This neural network is trained within a few minutes as a multistep Gaussian denoiser. The numerical experiments for denoising, CT, and MRI reconstruction show improvements over methods that offer similar reliability guarantees.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Denoising | BSD68 | PSNR31.2 | 404 | |
| Image Denoising | Set14 | PSNR31.74 | 67 | |
| Deblurring | BSD68 | PSNR30.38 | 24 | |
| Image Denoising | McMaster | PSNR33.24 | 18 | |
| CT Reconstruction | LoDoPaB (test) | PSNR35.21 | 15 | |
| MRI Reconstruction | fastMRI PD | PSNR35.1 | 10 | |
| MRI Reconstruction | fastMRI PDFS | PSNR34.18 | 10 | |
| CS-MRI Reconstruction | CS-MRI (test) | Average Latency (s)6.89 | 4 | |
| CT Reconstruction | CT (test) | Average Duration (s)13.9 | 4 | |
| Image Deblurring | Deblurring (test) | Average Duration (s)6.45 | 4 |