Table of Contents
Fetching ...

Block-coordinate Plug-And-Play Methods with Armijo-like line-search for Image Restoration

Federica Porta, Simone Rebegoldi, Andrea Sebastiani

TL;DR

The proposed block-coordinate Plug-and-Play methods are based on a block-coordinate forward-backward framework for solving non-convex and non-separable composite optimization problems, making them particularly suitable for large-scale and resource-constrained imaging applications.

Abstract

In this paper, we develop a class of block-coordinate Plug-and-Play (PnP) methods to address imaging inverse problems. The block-coordinate strategy is designed to reduce the high memory consumption arising in PnP methods that rely on Gradient Step denoisers, whose implementation typically requires storing large computational graphs. The proposed methods are based on a block-coordinate forward-backward framework for solving non-convex and non-separable composite optimization problems. Furthermore, such methods allow for the joint use of inertial acceleration, variable metric strategies, inexact proximal computations, and adaptive steplength selection via an appropriate line-search procedure. Under mild assumptions on the objective function, we establish a sublinear convergence rate and the stationarity of the limit points. Moreover, convergence of the entire sequence of the iterates is guaranteed under a Kurdyka-Łojasiewicz assumption. Numerical experiments on ill-posed imaging problems, including deblurring and super-resolution, demonstrate that the proposed PnP approach achieves state-of-the-art reconstruction quality while substantially reducing GPU memory requirements, making it particularly suitable for large-scale and resource-constrained imaging applications.

Block-coordinate Plug-And-Play Methods with Armijo-like line-search for Image Restoration

TL;DR

The proposed block-coordinate Plug-and-Play methods are based on a block-coordinate forward-backward framework for solving non-convex and non-separable composite optimization problems, making them particularly suitable for large-scale and resource-constrained imaging applications.

Abstract

In this paper, we develop a class of block-coordinate Plug-and-Play (PnP) methods to address imaging inverse problems. The block-coordinate strategy is designed to reduce the high memory consumption arising in PnP methods that rely on Gradient Step denoisers, whose implementation typically requires storing large computational graphs. The proposed methods are based on a block-coordinate forward-backward framework for solving non-convex and non-separable composite optimization problems. Furthermore, such methods allow for the joint use of inertial acceleration, variable metric strategies, inexact proximal computations, and adaptive steplength selection via an appropriate line-search procedure. Under mild assumptions on the objective function, we establish a sublinear convergence rate and the stationarity of the limit points. Moreover, convergence of the entire sequence of the iterates is guaranteed under a Kurdyka-Łojasiewicz assumption. Numerical experiments on ill-posed imaging problems, including deblurring and super-resolution, demonstrate that the proposed PnP approach achieves state-of-the-art reconstruction quality while substantially reducing GPU memory requirements, making it particularly suitable for large-scale and resource-constrained imaging applications.
Paper Structure (17 sections, 11 theorems, 124 equations, 5 figures, 2 tables, 1 algorithm)

This paper contains 17 sections, 11 theorems, 124 equations, 5 figures, 2 tables, 1 algorithm.

Key Result

Lemma 2.1

Bonettini-Prato-Rebegoldi-2024 Let $f:\mathbb{R}^n\rightarrow \mathbb{R}$ be continuously differentiable and $\phi:\mathbb{R}^n\rightarrow \mathbb{R} \cup \{\infty\}$ proper, lower semicontinuous and convex. Choose $0<\alpha\leq \alpha_{\max}$, $\beta>0$, $\mu>0$, $D\in\mathcal{S}_{++}(n)$ with $\fr

Figures (5)

  • Figure 1: Example of block GS denoiser action.
  • Figure 2: Results achieved solving the image deblurring problem on leaves by varying the number $N$ of blocks. For $N=1$ the results obtained by GS-PnP are also included.
  • Figure 3: Reconstructions provided by GS-PnP, Block-PHILA-v1 (second row) and Block-PHILA-v7 (third row) for the image deblurring problem.
  • Figure 4: Results achieved solving the image super resolution problem on starfish by varying the number $N$ of blocks. For $N=1$ the results obtained by GS-PnP are also included.
  • Figure 5: Reconstructions provided by GS-PnP, Block-PHILA-v1 (second row) and Block-PHILA-v7 (third row) for the super-resolution problem.

Theorems & Definitions (27)

  • Definition 2.1
  • Remark
  • Definition 2.2
  • Remark
  • Definition 2.3
  • Remark
  • Lemma 2.1
  • Definition 2.4
  • Lemma 4.1
  • proof
  • ...and 17 more