1. Introduction
Image restoration is an important task in many areas of applied sciences since digital images are frequently degraded by blur and noise during the acquisition process. Image restoration can be mathematically formulated as the linear inverse problem [
1]
where
and
, respectively, are vectorized forms of the observed
image and the exact
image to be restored,
is the linear operator modeling the imaging system, and
represents Gaussian white noise with mean zero and standard deviation
. The image restoration problem (
1) is inherently ill-posed and regularization strategies, based on the prior information on the unknown image, are usually employed in order to effectively restore the image
from
.
In a variational framework, image restoration can be reformulated as a constrained optimization problem of the form
whose objective function contains a
-based term, imposing consistency of the model with the data, and a regularization term
, forcing the solution to satisfy some a priori properties. Here and henceforth, the symbol
denotes the Euclidean norm. The constraint imposes some characteristics on the solution which are often given by the physics underlying the data acquisition process. Since image pixels are known to be nonnegative, a typical choice for
is the positive orthant.
The quality of the restored images strongly depends on the choice of the regularization term which, in a very general framework, can be expressed as
where the positive scalars
are regularization parameters and the
are regularization functions for
. The multi-penalty approach (
3) allows to impose several regularity properties on the desired solution, however a crucial issue with its realization is the need to define reliable strategies for the choice of the regularization parameters
,
.
Therefore, in the literature, the most common and famous regularization approach is single-penalty regularization, also known as Tikhonov-like regularization, which corresponds to the choice
:
In image restoration, smooth functions based on the
-norm or convex nonsmooth functions like the Total Variation, the
norm or the Total Generalized Variation are usually used for
in (
4) [
2,
3]. Even in the case
, the development of suitable parameter choice criteria is still an open question. The recent literature has demonstrated a growing interest in multi-penalty regularization, with a significant number of researchers focusing on scenarios involving two penalty terms. Notably, the widely-used elastic regression in Statistics serves as an example of a multi-penalty regularization technique, integrating the
and
penalties from the Lasso and Ridge methods. However, the majority of the literature primarily addresses the development of suitable rules for parameter selection. Lu, Pereverzev et al. [
4,
5] have extensively investigated two
-based terms, introducing a refined discrepancy principle to compute dual regularization parameters, along with its numerical implementation. The issue of parameter selection is further discussed in [
6], where a generalized multi-parameter version of the L-curve criterion is proposed, and in [
7], which suggests a methodology based on the GCV method. Reichel and Gazzola [
8] propose regularization terms of the form
where
are suitable regularization matrices. They present a method to determine the regularization parameters utilizing the discrepancy principle, with a special emphasis on the case
. Fornasier et al. [
9] proposed a modified discrepancy principle for multi-penalty regularization and provided theoretical background for this a posteriori rule. Works such as [
10,
11,
12,
13,
14] also explore multi-penalty regularization for unmixing problems, employing two penalty terms based on
and
norms,
and
. The latter specifically concentrates on the
and
norms. The study [
15] assesses two-penalty regularization, incorporating
and
penalty terms to tackle nonlinear ill-posed problems and analyzes its regularizing characteristics. In [
16], an automated spatially adaptive regularization model combining harmonic and Total Variation (TV) terms is introduced. This model is dependent on two regularization parameters and two edge information matrices. Despite the dynamic update of the edge information matrix during iterations, the model necessitates fixed values for the regularization parameters. Calatroni et al. [
17] present a space-variant generalized Gaussian regularization approach for image restoration, emphasizing its applicative potential. In [
18], a multipenalty point-wise approach based on the Uniform Penalty principle is considered and analyzed for general linear inverse problems, introducing two iterative methods, UpenMM and GUpenMM, and analyzing their convergence.
Here, we extend the methodology developed in [
18] to image restoration problems and we perform a comparative analysis with state-of-the-art regularization methods for this application. We propose to find an estimate
of
satisfying
where
is a positive scalar and
is the discrete Laplacian operator. This model, named MULTI, is specifically tailored for the image restoration problem. Observe that MULTI incorporates a pixel-wise regularization term and includes a rule for choosing the parameters. We formulate an iterative algorithm for computing the solution
of (
6), where
. Once the regularization parameters are set in every inner iteration, the constrained minimization subproblem is efficiently solved by a customized version of the Newton Projection (NP) method. Here, the Hessian matrix is approximated by a Block Circulant with Circulant Blocks (BCCB) matrix, which is easily invertible in the Fourier space. This modified version of NP was designed in [
19] for single-penalty image restoration under Poisson noise and it is adapted here to the context of multi-penalty regularization. Consequently, the convergence of the modified NP method can be established.
The principal contributions of this work are summarized as follows:
We propose a variational pixel-wise regularization model tailored for image restoration and derived from the theoretical model developed in [
18].
We devise an algorithm capable of effectively and efficiently solving the proposed model.
Through numerical experiments, we demonstrate that the proposed approach can proficiently eliminate noise and blur in smooth areas of an image while preserving its edges.
The structure of this paper is as follows:
Section 2 introduces the proposed algorithm. The results of numerical experiments are presented in
Section 3, and finally, the conclusions are drawn in
Section 4.
2. Materials and Methods
In this section, we present the iterative algorithm that generates the sequence
converging to the solution
in (
6).
Starting from an initial guess
taken as the observed image
, the correspondent initial guess of the regularization parameters is computed as:
where
and
is a neighborhood of size
, (with
R odd and
) of the
-th pixel with coordinates
.
The successive terms
are obtained by the update formulas reported in steps 3–5 of Algorithm 1. The iterations are stopped when the relative distance between two successive regularization vectors is smaller than a fixed tolerance
.
Algorithm 1 Input: , , , Output: |
- 1:
Set . - 2:
repeat - 3:
- 4:
- 5:
- 6:
- 7:
until - 8:
,
|
Algorithm 1 is well defined, and we experimentally observe a converging behaviour. Its formal convergence proof is obtained in [
18] (theorem 3.4) for the case
, since in this case, Algorithm 1 corresponds to UPenMM. Otherwise, to preserve convergence, we should introduce a correction as proposed in the Generalized Uniform Penalty method (GUPenMM) [
18]. However, even without this correction, we obtained good-quality results and we prefer here to investigate Algorithm 1 because, in the case of large scale image restoration problems, it is much more convenient from a computational point of view. Moreover, we verified that the results obtained with such a correction are qualitatively comparable with those given by Algorithm 1, as the human eye cannot distinguish differences smaller than a few gray levels.
At each inner iteration, the constrained minimization subproblem (step 3 in Algorithm 1) is solved efficiently by a tailored version of the NP method where the Hessian matrix is approximated by a BCCB matrix easily invertible in the Fourier space.
Let us denote by
the function to be minimized at step 3 in Algorithm 1:
and by
its gradient
where the iteration index
k has been omitted for easier notation. Moreover, let
denote the reduced gradient:
where
is the set of indices [
20]:
with
and
is a small positive parameter.
The Hessian matrix
has the form
where
is the diagonal matrix with diagonal elements
.
A general iteration of the proposed NP-like method has the form:
where
is the search direction,
is the steplength and
denotes the projection on the positive orthant.
At each iteration
ℓ, the computation of
requires the solution of the linear system
where
is the following approximation to
Under periodic boundary conditions,
is a BCCB matrix and system (
11) can be efficiently solved in the Fourier space by using Fast Fourier Transforms. Therefore, despite its simplicity, the BCCB approximation
is efficient, since it allows to solve the linear system in
operations, and effective, as is shown by the numerical results. Finally, given the solution
of (
11), the search direction
is obtained as
The step length
is computed with the variation of the Armijo rule discussed in [
20] as the first number of the sequence
,
, such that
where
and
.
We observe that the approximated Hessian
is constant for each inner iteration
ℓ and it is positive definite, then it satisfies
Following that, the results given in [
19] for single-penalty image restoration under Poisson noise can be applied here to prove the convergence of the NL-like iterations to critical points.
The stopping criteria for the NP-like method are based on the relative distance between two successive iterates and the relative projected gradient norm. In addition, a maximum number of NP iterations have been fixed.
3. Numerical Experiments
All the experiments were performed under Windows 10 and MATLAB R2021a running on a desktop (Intel(R) Core(TM)
[email protected] GHz). Quantitatively, we evaluated the quality of image restoration by the relative error (RE), improved signal to noise ratio (ISNR), and mean structural similarity index (MSSIM) measures. The MSSIM is defined by Wang et al. [
21] and ISNR is calculated as:
where
is the restored image,
is the reference image, and
is the blurred, noisy image.
Four reference images were used in the experiments:
galaxy,
mri,
leopard, and
elaine, shown in
Figure 1,
Figure 2,
Figure 3 and
Figure 4. The first three images have size
, while the
elaine image is
. In order to define the test problems, each reference image was convolved with two PSFs corresponding to a Gaussian blur with variance 2, generated by the
psfGauss function from the MATLAB toolbox Restore-Tool [
1], and an out-of-focus blur with radius 5, obtained with the function
fspecial from the MATLAB Image Processing Toolbox. The resulting blurred image was then corrupted by Gaussian noise with different values of the noise level
. The values
were used.
We compared the proposed pixel-wise multi-penalty regularization model (MULTI) with some commonly used state-of-the-art methods based on a variational approach. In particular, we considered the Tikhonov method (TIKH) [
22], the Total Variation (TV) [
2], and Total Generalized Variation (TGV) [
3] regularization with nonnegative constraints. Tikhonov and TV regularization are quite classic regularization terms. It is well known that Tikhonov regularization tends to make images overly smooth and it fails to preserve sharp edges. On the contrary, TV regularization better preserves sharp edges but often produces staircase effects. TGV has been recently proposed to overcome the drawbacks of Tikhonov and TV regularization such as blurring and the staircasing effect. Therefore, we compared MULTI with TGV in order to demonstrate the capacity of MULTI to preserve sharp features as well as smooth transition variations.
In our numerical experiments, the regularization parameter values for TIKH, TV, and TGV were chosen heuristically by minimizing the relative error values. The Alternating Direction Method of Multipliers (ADMM) was used for the solution of the TV-based minimization problem, while for TIKH, we used the Scaled Gradient Projection (SGP) method with Barzilai and Borwein rules for the step length selection [
23]. Regarding the TGV regularization, the RESPOND method [
24] was used. We remark that RESPOND has been originally proposed for the restoration of images corrupted by Poisson noise by using Directional Total Generalized Variation regularization. It has been adapted here to deal with TGV-based restoration of images under Gaussian noise. The MATLAB implementation for Poisson noise is available on GitHub at the url
https://github.com/diserafi/respond (accessed on 18 September 20).
The tolerance , in the outer loop of MULTI in Algorithm 1 step 7, was , while the maximum number of iterations was 20. Regarding the NP method, a tolerance of was used and the maximum number of iterations was 1000.
The size of the neighborhood
in (
8) was
pixels for all tests except for
galaxy, where a
neighborhood was used.
The values of the parameter
in (
6), used in the various tests, are in the range
. In order to compare all the algorithms at their best performance, the values
used in each test are reported in
Table 1, where we observe that the value of
is proportional to the noise level. The parameter
represents a threshold and, in general, should have a small value when compared to the non-null values of
. We note that at the cost of adjusting a single parameter
, it is possible to achieve point-wise optimal regularization.
Table 2,
Table 3,
Table 4 and
Table 5 report the numerical results for all the test problems. The last column of the Tables shows the used values of the regularization parameter for TIKH, TV, and TGV while, for MULTI, it reports the norm of the regularization parameters vector
computed by Algorithm 1. Column 7 shows the number of RESPOND, ADMM, and SGP iterations for TGV, TV, and TIKH, respectively. For the MULTI algorithm, Column 7 shows the number of outer iterations and NP iterations in parenthesis.
In
Table 2,
Table 3,
Table 4 and
Table 5, we note that for the case of Gaussian blur, MULTI consistently achieves the best results, as highlighted in bold. This is evident from its higher MSSIM and ISNR values and lower RE values. However, for the Out-of-focus case, there are three instances (representing
) where TGV exhibits superior error parameters. Furthermore, our observations indicate that TGV consistently outperforms both TIKH and TV in terms of accuracy. Therefore, in
Figure 1,
Figure 2,
Figure 3 and
Figure 4 we only represent the images obtained by MULTI and TGV in the out-of-focus case, with
, as this is a very challenging case.
The strength of MULTI is evident by observing some details of the reconstructed images. In
Figure 5,
Figure 6,
Figure 7 and
Figure 8 we show some cropped details of the original images and compare it with MULTI and TGV reconstructions.
Figure 5 shows a detail of the
galaxy with a few stars over a dark background. In this case the image sparsity is better preserved by MULTI.
Figure 6 shows the galaxy centre: it is a smooth area which is well recovered by MULTI while TGV shows staircasing. In
Figure 7 we observe that the leopard’s whiskers and fur spots are better reproduced by MULTI. Moreover, from the images provided in
Figure 8, it can be observed that MULTI method better preserves the local characteristics of the image, avoiding flattening the smooth areas and optimally preserving the sharp contours. We observe that a smooth area such as the cheek is better represented by MULTI, avoiding the staircase effect. Moreover, an area with strong contours, such as the teeth and the eyes, is better depicted. In summary, these examples show the good capacity of MULTI to preserve the different image structures, narrow peaks, and smooth areas by using local regularization parameters that are inversely proportional to the local curvature approximated by the discrete Laplacian.
The regularization parameters computed by MULTI are represented in
Figure 9; the adherence of the regularization parameters’ values to the image content is clear, showing larger values in areas where the image is flat and smaller values where there are pronounced gradients (edges). The range of the parameters automatically adjusts according to the different test types.
Finally, we show in
Figure 10 an example of the algorithm behavior, reporting the history of the regularization parameter norm, relative error, and residual norm (top row), in the case of
leopard test with out-of-focus blur and noise level
. In the bottom row we show the decreasing behavior of the objective function and projected gradient norm. The relative error flattens after a few iterations, and the same behavior can be observed in all the other tests. Therefore, we used a large tolerance value (
) in the outer loop of Algorithm 1, making it possible to obtain good regularization parameters and accurate restorations in a few outer iterations. We observe that even in the most difficult case,
Table 4 row 12, the total computation time is 285 s, proving the algorithm efficiency.