parameter estimation in bayesian blind deconvolution with ......this work was partially supported by...

5
PARAMETER ESTIMATION IN BAYESIAN BLIND DECONVOLUTION WITH SUPER GAUSSIAN IMAGE PRIORS Miguel Vega, Rafael Molina * Universidad de Granada, 18071 Granada Spain, e-mails: [email protected], [email protected] Aggelos K. Katsaggelos Department of Electrical Engineering and Computer Science, Northwestern University, Evanston, IL 60208 USA, e-mail: [email protected] ABSTRACT Super Gaussian (SG) distributions have proven to be very powerful prior models to induce sparsity in Bayesian Blind Deconvolution (BD) problems. Their conjugate based rep- resentations make them specially attractive when Variational Bayes (VB) inference is used since their variational parame- ters can be calculated in closed form with the sole knowledge of the energy function of the prior model. In this work we show how the introduction in the SG distribution of a global strength (not necessary scale) parameter can be used to im- prove the quality of the obtained restorations as well as to introduce additional information on the global weight of the prior. A model to estimate the new unknown parameter within the Bayesian framework is provided. Experimental results, on both synthetic and real images, demonstrate the effectiveness of the proposed approach. Index TermsBayesian methods, image processing, image restoration, Super Gaussian, blind deconvolution. 1. INTRODUCTION Blind image deconvolution is the problem of restoring an image x from its blurred and noisy version y when the blur H is unknown. Generally, the image y is modeled as y = Hx + n , (1) where n is the noise. Both y, x and n are lexicographically arranged N × 1 vectors, and H is an N × N matrix. In many cases the blur H is spatially-varying, but in this paper we as- sume that H is a spatially invariant two dimensional convolu- tion operator of unknown nucleus h. Since h, x and n are un- known, the problem is highly ill-posed and there are infinitely many solutions for x and h. * This work was partially supported by the “Ministerio de Ciencia e Inno- vacion” under contract TIN2010-15137, and “CEI BioTic” at the University of Granada under contract CEI2014-PTIC11. Blind image deconvolution is a widely investigated prob- lem in signal/image processing and computer vision [1], and recently attracted much attention mostly geared towards re- moving camera shake [2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]. Fer- gus et. al. [2] employed the variational Bayesian approach of Miskin and Mackay [13] with a mixture-of-Gaussians (MoG) image prior for modeling natural image statistics. After the success of this approach, subsequent methods proposed new image and blur modeling schemes and highly efficient infer- ence methods [5, 6, 7, 8]. In [14] a general VB blind deconvolution method, which can be used on a large family of sparsity promoting priors, was proposed. In this paper we extend the model in [14] by the introduction of a global scale parameter which is used to improve the quality of the restoration. The parameter is auto- matically estimated. The rest of this paper is organized as follows. Section 2 describes our Bayesian modelling of the blind deconvolution problem. The Bayesian inference and the proposed algorithm are described in Section 3. In Section 4 we present the exper- imental results. Section 5 concludes the paper. 2. BAYESIAN MODELS 2.1. Degradation model In this paper the deconvolution problem is formulated in the filter space [5, 8], by applying high-pass filters {F γ } L γ=1 (such as derivatives, wavelets, curvelets, etc.) to the blurred noisy image y to obtain L pseudo-observations y γ = F γ y = HF γ x + F γ n = Hx γ + n γ , (2) with x γ = F γ x, and n γ = F γ n. In (2) it is assumed that H and F γ have the same set of e-vectors, so that they commute. Assuming that n is uncorrelated white Gaussian noise of known inverse variance β, we define the distribution of the observed images y γ as p(y γ |x γ , h)= N (y γ |Hx γ -1 γ I) , (3) 1632

Upload: others

Post on 08-Aug-2021

4 views

Category:

Documents


0 download

TRANSCRIPT

Page 1: Parameter estimation in Bayesian Blind Deconvolution with ......This work was partially supported by the “Ministerio de Ciencia e Inno-vacion” under contract TIN2010-15137, and

PARAMETER ESTIMATION IN BAYESIAN BLIND DECONVOLUTIONWITH SUPER GAUSSIAN IMAGE PRIORS

Miguel Vega, Rafael Molina ∗

Universidad de Granada,18071 Granada Spain,

e-mails: [email protected], [email protected]

Aggelos K. Katsaggelos

Department of Electrical Engineeringand Computer Science,

Northwestern University,Evanston, IL 60208 USA,

e-mail: [email protected]

ABSTRACT

Super Gaussian (SG) distributions have proven to be verypowerful prior models to induce sparsity in Bayesian BlindDeconvolution (BD) problems. Their conjugate based rep-resentations make them specially attractive when VariationalBayes (VB) inference is used since their variational parame-ters can be calculated in closed form with the sole knowledgeof the energy function of the prior model. In this work weshow how the introduction in the SG distribution of a globalstrength (not necessary scale) parameter can be used to im-prove the quality of the obtained restorations as well as tointroduce additional information on the global weight of theprior. A model to estimate the new unknown parameter withinthe Bayesian framework is provided. Experimental results, onboth synthetic and real images, demonstrate the effectivenessof the proposed approach.

Index Terms— Bayesian methods, image processing,image restoration, Super Gaussian, blind deconvolution.

1. INTRODUCTION

Blind image deconvolution is the problem of restoring animage x from its blurred and noisy version y when the blurH is unknown. Generally, the image y is modeled as

y = Hx + n , (1)

where n is the noise. Both y, x and n are lexicographicallyarranged N × 1 vectors, and H is an N ×N matrix. In manycases the blur H is spatially-varying, but in this paper we as-sume that H is a spatially invariant two dimensional convolu-tion operator of unknown nucleus h. Since h, x and n are un-known, the problem is highly ill-posed and there are infinitelymany solutions for x and h.

∗This work was partially supported by the “Ministerio de Ciencia e Inno-vacion” under contract TIN2010-15137, and “CEI BioTic” at the Universityof Granada under contract CEI2014-PTIC11.

Blind image deconvolution is a widely investigated prob-lem in signal/image processing and computer vision [1], andrecently attracted much attention mostly geared towards re-moving camera shake [2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]. Fer-gus et. al. [2] employed the variational Bayesian approach ofMiskin and Mackay [13] with a mixture-of-Gaussians (MoG)image prior for modeling natural image statistics. After thesuccess of this approach, subsequent methods proposed newimage and blur modeling schemes and highly efficient infer-ence methods [5, 6, 7, 8].

In [14] a general VB blind deconvolution method, whichcan be used on a large family of sparsity promoting priors,was proposed. In this paper we extend the model in [14] bythe introduction of a global scale parameter which is used toimprove the quality of the restoration. The parameter is auto-matically estimated.

The rest of this paper is organized as follows. Section 2describes our Bayesian modelling of the blind deconvolutionproblem. The Bayesian inference and the proposed algorithmare described in Section 3. In Section 4 we present the exper-imental results. Section 5 concludes the paper.

2. BAYESIAN MODELS

2.1. Degradation model

In this paper the deconvolution problem is formulated in thefilter space [5, 8], by applying high-pass filters {Fγ}Lγ=1

(such as derivatives, wavelets, curvelets, etc.) to the blurrednoisy image y to obtain L pseudo-observations

yγ = Fγy = HFγx + Fγn = Hxγ + nγ , (2)

with xγ = Fγx, and nγ = Fγn. In (2) it is assumed that Hand Fγ have the same set of e-vectors, so that they commute.

Assuming that n is uncorrelated white Gaussian noise ofknown inverse variance β, we define the distribution of theobserved images yγ as

p(yγ |xγ ,h) = N (yγ |Hxγ , β−1γ I) , (3)

1632

Page 2: Parameter estimation in Bayesian Blind Deconvolution with ......This work was partially supported by the “Ministerio de Ciencia e Inno-vacion” under contract TIN2010-15137, and

Table 1. Different choices for the penalty functionLabel ρ(s) ρ′(s)/|s|`p , 0 < p ≤ 1 1

p |s|p |s|p−2

log log(ε+ |s|) (ε+ |s|)−1|s|−1

bup, p > 2 12−p (ε+ |s|)

2−p (ε+ |s|)−p

where we use the approximation 1βFγF

Tγ ≈ 1

βγI.

2.2. Prior models

Let us now proceed to model our prior knowledge on theimage. To promote sparsity on xγ (derivatives, wavelets,curvelets, ...) we could use on each xγ(i) a Super Gaussiandistribution,

p (xγ(i)|αγ) = Zγ(αγ) exp (−αγρ (xγ(i))) , (4)

for i = 1, . . . , N , where Zγ(αγ) is the partition function, andthe parameter αγ > 0 globally regulates the prior strength.Table 1 shows some penalty functions, corresponding to SGdistributions (see [14]). In Table 1, the acronym bup standsfor the, so-called in [14], bottom-up approach, for a givenvalue of p.

Notice that the global parameter αγ in (4), which is notconsidered in [14] but introduced in this work, plays a veryimportant role as we will see in the experimental section. No-tice also that, it represents much more than a rescaling ofxγ(i), see for instance the log prior in Table 1 where the in-troduction of the αγ represents powering in the filter space.

For p (xγ(i)|αγ) in (4) to be SG, ρ(·) has to be symmetricaround zero, and the function ρ(

√s) has to be increasing and

concave for s ∈ (0,∞) [15]. This condition is equivalentto ρ′(s)/s being decreasing on (0,∞). If this condition issatisfied, then ρ can be represented as (see [16, Ch. 12])

ρ (xγ(i)) = infηγ(i)>0

1

2ηγ(i)x

2γ(i) − ρ∗

(1

2ηγ(i)

)(5)

⇒ ρ (xγ(i)) ≤1

2ηγ(i)x

2γ(i)− ρ∗

(1

2ηγ(i)

)(6)

where inf denotes the infimum, ρ∗ (·) is the concave conju-gate of ρ(·) and ηγ = {ηγ(i)}Ni=1 are positive variationalparameters. These parameters have an intuitive meaning andextreme importance in the deconvolution performance, as willbe shown later. The relationship dual to (5) is given by [16]

ρ∗(1

2ηγ(i)

)= infxγ(i)

1

2ηγ(i)x

2γ(i)− ρ (xγ(i)) . (7)

The quadratic bound for ρ in (6) allows us to bound the prior

in (4) with a Gaussian form, specifically we can write

p (xγ(i)|αγ)

≥ Zγ(αγ) exp[−αγ(1

2ηγ(i)x

2γ(i)− ρ∗

(ηγ(i))

2

))]

= Zγ(αγ) exp[−αγL(xγ(i), ηγ(i))]. ∀ηγ(i) > 0 (8)

Combining Eqs. (3) & (8) we obtain the global variationallower bound

p(Θ,Y) ≥ p(h)

L∏γ=1

(p(αγ)p(yγ |xγ ,h)

×∏i

Zγ(αγ) exp[−αγL(xγ(i), ηγ(i))])

= P(Θ,Y,η) η > 0 (9)

where Θ = {h, α1,x1, . . . , αL,xL}, Y = {y1, . . . ,yL} andη = {ηγ1 , . . . ,ηγL}.

3. BAYESIAN INFERENCE

The inference scheme is based on the posterior distributionp(Θ|Y) = p(Θ,Y)

p(Y) . Since the posterior distribution cannotbe obtained in closed form, we base our estimation on VBinference [17], where the posterior p(Θ|Y) is approximatedby the distribution q(Θ) =

∏θ∈Θ q(θ), with

log q(θ) = E [log p(Θ,Y)]q(Θ\{θ}) + const , (10)

obtained by taking the expectation of the joint distributionwith respect to all unknowns except the one of interest,Θ\{θ}. We derive next the estimates of the different vari-ables.

Instead of using p(Θ,Y) we utilize its lower bound,which includes the variational parameter, and solve itera-tively

log q(θ) = E [log P(Θ,Y, η)]Θ\{θ} + const ,

η = argmaxη

E [log P(Θ,Y,η)]q(Θ) (11)

3.1. Estimation of Image and Blur

Assuming a flat prior p(h) we obtain for the blur

log q(h) =∑γ

E [log p(yγ |xγ ,h)]q(xγ) + const . (12)

Since the size of the blur is normally very small in compar-ison with the number of observations we treat the blur as adeterministic parameter which is only constrained to satisfyh ≥ 0,

∑i h(i) = 1. Then from (12) we obtain

h = argminh

∑γ

E[‖yγ −Hxγ‖22

]q(xγ)

= argminh

hTC−1h h− 2hT bh (13)

1633

Page 3: Parameter estimation in Bayesian Blind Deconvolution with ......This work was partially supported by the “Ministerio de Ciencia e Inno-vacion” under contract TIN2010-15137, and

subject to h ≥ 0,∑i h(i) = 1. For a kernel size of M ×M ,

bh is M2 × 1, the size of C−1h is M2 ×M2 and we have

C−1h (m,n) =

∑γ

N∑j=1

E [xγ(m+ j)] E [xγ(n+ j)]

+ Cxγ (m+ j, n+ j) (14)

bh(m) =∑γ

N∑j=1

E [xγ(m+ j)] yγ(j) . (15)

The estimation of the blur in (13) is thus a simple quadraticproblem which can be solved very efficiently.

For the filtered images we obtain

log q(xγ) = log p(yγ |xγ ,h)−1

2αγx

Tγ diag (ηγ)xγ + const

(16)

which is a multivariate Gaussian with precision matrix

C−1xγ = βγH

TH + αγdiag (ηγ) , (17)

and whose mean xγ , which we take as the estimate of xγ , isobtained by solving the following linear system

C−1xγ xγ = βγH

Tyγ . (18)

This system can be solved efficiently using Conjugate Gra-dient (CG) without finding Cxγ explicitly. However, this co-variance matrix is needed in (14). Since its computation isextremely expensive, we apply in (14) the Jacobi approxima-tion, and invert only the diagonal of Cxγ .

3.2. Estimation of the Variational Parameter ηγ

Using (10), we obtain for the variational parameters

ηγ(i) = argminηγ(i)

1

2ηγ(i)ν

2γ(i)− ρ∗

(ηγ(i)

2

)(19)

where νγ(i) =√

E[x2γ(i)

].

Since

ρ∗(ηγ(i)

2) = min

x

1

2ηγ(i)x

2 − ρ(x) (20)

whose minimum is achieved at x = νγ(i), we have, differen-tiating the right hand side of the above equation with respectto x,

ηγ(i) = ρ′(νγ(i))/νγ(i) (21)

3.3. Estimation of αγ

From (11) we obtain the following distribution for αγ

log q(αγ) = const + log p(αγ)

+

N∑i=1

logZγ(αγ) exp[−αγρ(νγ(i))] (22)

We utilize the mode of this distribution as the αγ estimate.A flat hyperprior (p(αγ) ∝ const) results in the followingequation for αγ

∂αγlogZγ(αγ) =

1

N

N∑i=1

ρ(νγ(i)) . (23)

The `p and bup penalty functions shown in Table 1 pro-duce proper priors, for which the partition function can be

evaluated. For the `p function Zγ(αγ) = 12Γ(1/p)α

1pγ p

1− 1p ,

and for the bup one Zγ(αγ) = 12Γ(1/(2−p))α

12−p (2 − p)

1−p2−p .

However, the log penalty function produces an improperprior. We tackle this problem examining, for αγ 6= 1, thebehavior of

Zγ(αγ ,K)−1 =

∫ K

−Ke−αγρ(s)ds . (24)

and keeping in ∂ logZγ(αγ ,K)/∂αγ the term that depends onαγ . This produces the estimate

1

αγ − 1=

1

N

N∑i=1

ρ(νγ(i)) . (25)

3.4. The proposed Algorithm

Algorithm 1 Blind Deconvolution using Sparse Image PriorsRequire: Degraded image y, noise parameter β, choice forρ function, and filters Fγ .repeat

for γ = 1 to L doInitialization: Set xγ = yγ , Cxγ = 0.

1. Compute νγ(i) =√E[x2γ(i)

], ηγ using (21), and

αγ solving (23).2. Estimate filtered image xγ by solving (18).

end for3. Estimate the blur kernel h using (13) .4. Approximate Cxγ (i, i) with 1/C−1

xγ (i, i).until convergence5. Compute the final image estimate x by solving

(βHTH +∑γ

αγFTγ diag (ηγ)Fγ) x = βHT y . (26)

The proposed Algorithm 1 has two parts. First, Algo-rithm 1 iteratively alternates between the estimates of the fil-tered images xγ and the blur h. At each iteration, the methodalso estimates the parameters νγ , αγ , and ηγ , and approxi-mates Cxγ (i, i). In the estimation of h, the pyramid coarse-to-fine approach suggested in [9] has been applied. In its sec-ond part, Algorithm 1 has to construct the image x from thexγ filtered images. As this requires a non trivial integration ofall xγ , we propose instead the estimate shown in (26), whichstill enforces sparsity in the filter domain through the use ofηγ , and requires only one more CG application.

1634

Page 4: Parameter estimation in Bayesian Blind Deconvolution with ......This work was partially supported by the “Ministerio de Ciencia e Inno-vacion” under contract TIN2010-15137, and

4. EXPERIMENTAL RESULTS

The effectiveness of the proposed method, for the ρ penaltyfunctions `p, log, and bup, in Table 1, has been quantitativelyassessed over the synthetic dataset in [8]. Results for the `ppenalty function with p = 0.8, called `0.8, and for the buppenalty with p = 3, called bu3, are reported. We use theSum Square Distances (SSD) between the deconvolved andoriginal images as figure of merit.

For the images in the dataset in [8], Figure 1 shows thecumulative SSD histograms of the blindly deconvolved im-ages obtained when using the penalty functions in Table 1 andLevin et al.’s MAPk approach [9], a MoG prior for kernel es-timation. This method was shown in [9] to outperform theMAP based approaches of [5] and [6] in this dataset. Figure 1also shows the SSD histograms for non-blind deconvolutions,and for the ratios between blind and non-blind SSD values.These SSD values are better than the reported in [14]. in Fig-ure 1 it can be observed that the log prior gives the best perfor-mance, and that both `p and bup outperform MoG. The figuresof merits for the `p prior for non-blind deconvolution are verygoods. However, their blind SSD values, and their ratios arepoorer. The results obtained with the bup prior are better forblind than for non-blind deconvolution, and this prior givesthe best performance in ratio terms.

Figure 2 shows the results obtained with the different pri-ors for one of the images in the dataset, and Figure 3 comparesthe proposed method with the methods in [6], and in [10], us-ing their implementations, and parameter settings, on a realimage.

The introduction of the αγ parameter in the general sparseprior in (4) improves the performance of the method proposedin [14]. The improvement is very minor the log prior, forwhich the estimated parameter values are very close to one.For the `p penalty αγ ≈ 102 has been obtained, and αγ ≈10−4 for bup.

5. CONCLUSIONS

In this paper we have introduced an additional parameter inthe Super Gaussian (SG) prior modeling of the image in thefilter space. We have shown how this parameter can be esti-mated using variational inference. Its effectiveness has beendemonstrated experimentally on real and synthetic images.

REFERENCES

[1] T.E. Bishop, S.D. Babacan, B. Amizic, T. Chan,R. Molina, and A.K. Katsaggelos, “Blind image de-convolution: problem formulation and existing ap-proaches,” in Blind image deconvolution: theory andapplications. CRC press, 2007.

[2] R. Fergus, B. Singh, A. Hertzmann, S.T. Roweis, andW.T. Freeman, “Removing camera shake from a single

photograph,” ACM Trans. Graph., vol. 25, pp. 787–794,July 2006.

[3] N. Joshi, C.L. Zitnick, R. Szeliski, and D.J. Kriegman,“Image deblurring and denoising using color priors,” inCVPR, june 2009.

[4] J. Jia, “Single image motion deblurring using trans-parency,” in CVPR, 2007.

[5] Q. Shan, J. Jia, and A. Agarwala, “High-quality motiondeblurring from a single image,” ACM Trans. Graph.(SIGGRAPH), 2008.

[6] S. Cho and S. Lee, “Fast motion deblurring,” ACMTrans. Graph. (SIGGRAPH ASIA), vol. 28, no. 5, 2009.

[7] L. Xu and J. Jia, “Two-phase kernel estimation for ro-bust motion deblurring,” in ECCV, 2010.

[8] A. Levin, Y. Weiss, F. Durand, and W. T. Freeman,“Understanding and evaluating blind deconvolution al-gorithms,” in CVPR, 2009, pp. 1964–1971.

[9] A. Levin, Y. Weiss, F. Durand, and W. T. Freeman, “Ef-ficient marginal likelihood optimization in blind decon-volution,” in CVPR, 2011, pp. 2657–2664.

[10] D. Krishnan, T. Tay, and R. Fergus, “Blind deconvo-lution using a normalized sparsity measure,” in CVPR,2011.

[11] X. Zhou, F. Zhou, and X. Bai, “Blind deconvolutionusing a nondimensional gaussianity measure,” in ICIP,2013.

[12] C. Wang, Y. Yue, F. Dong, Y. Tao, X. Ma, G. Clap-worthy, H. Lin, and X. Ye, “Nonedge-specific adap-tive scheme for highly robust blind motion deblurringof natural imagess,” IEEE Trans. Image Proc., vol. 22,pp. 884–897, Mar. 2013.

[13] J. W. Miskin and D. J. C. MacKay, “Ensemble learningfor blind image separation and deconvolution,” in Adv.in Independent Component Analysis. 2000, Springer.

[14] S. D. Babacan, R. Molina, M. N. Do, and A. K. Kat-saggelos, “Blind deconvolution with general sparseimage priors,” in ECCV, 2012.

[15] J. Palmer, K. Kreutz-Delgado, and S. Makeig, “Strongsub- and super-Gaussianity,” in LVA/ICA, 2010, vol.6365, pp. 303–310.

[16] R.T. Rockafellar, Convex analysis, Princeton Univer-sity Press, 1996.

[17] A. C. Likas and N. P. Galatsanos, “A variational ap-proach for Bayesian blind image deconvolution,” IEEETrans. on Signal Proc., vol. 52, pp. 2222–2233, 2004.

1635

Page 5: Parameter estimation in Bayesian Blind Deconvolution with ......This work was partially supported by the “Ministerio de Ciencia e Inno-vacion” under contract TIN2010-15137, and

20 40 60 800

0.1

0.2

0.3

0.4

0.5

0.6

0.7

0.8

0.9

1

SS

E B

lind

log

l0.8

MOG

bu3

5 10 15 20 250

0.1

0.2

0.3

0.4

0.5

0.6

0.7

0.8

0.9

1

SS

E N

o B

lind

1 2 3 40

0.1

0.2

0.3

0.4

0.5

0.6

0.7

0.8

0.9

1

Ratios S

SE

Blind / N

o B

lind

SSDs (blind) SSDs (non-blind) Ratios

Fig. 1. Cumulative histograms of SSDs on the dataset of [8] for blind and non-blind deconvolution, and their ratios (right).

Ground Truth Blurred Image

log SSD(58.8) R(2.2) l0.8 SSD(159) R(6.3)

[8] MoG SSD(64.9) R(2.5) bu3 SSD(56.5) R(1.9)

Fig. 2. Ground truth, blurred image and results obtainedwith our method using the different penalty functions and themethod in [8], for image no. 2, and blur no. 2 of the syntheticimage set in [8]. Estimated kernels are shown in insets.

Blurry Image Krishnan [10]

Ours (log) Cho [6]

Ours (l0.8) Ours (bu3)

Fig. 3. Blind Deconvolution results on a real image. Estimatedkernels are shown in insets.

1636