joint image sharpening and denoising by 3d transform-domain collaborative...

8
JOINT IMAGE SHARPENING AND DENOISING BY 3D TRANSFORM-DOMAIN COLLABORATIVE FILTERING Kostadin Dabov, Alessandro Foi, Vladimir Katkovnik, and Karen Egiazarian Institute of Signal Processing, Tampere University of Technology, P.O. Box 553, Tampere 33101, Finland rstname.lastname@tut. ABSTRACT In order to simultaneously sharpen image details and at- tenuate noise, we propose to combine the recent block- matching and 3D ltering (BM3D) denoising approach, based on 3D transform-domain collaborative ltering, with alpha-rooting, a transform-domain sharpening technique. The BM3D exploits grouping of similar image blocks into 3D arrays (groups) on which collaborative ltering (by hard-thresholding) is applied. We propose two approaches of sharpening by alpha-rooting; the rst applies alpha- rooting individually on the 2D transform spectra of each grouped block; the second applies alpha-rooting on the 3D-transform spectra of each 3D array in order to sharpen ne image details shared by all grouped blocks and further enhance the interblock differences. The conducted experi- ments with the proposed method show that it can preserve and sharpen ne image details and effectively attenuate noise. 1. INTRODUCTION The problem of sharpening image details while attenuat- ing noise arises in practice as a preprocessing step prior to segmentation, binarization (by thresholding) in the con- text of image analysis, classication, pattern recognition. The procedure is also very useful in consumer applica- tions, where it is utilized to obtain more visually appealing digital photographs. Various methods for image sharp- ening exist. Traditional ones rely on linear ltering that boosts higher frequencies or on elementwise transforma- tions such as the histogram-based methods (equalization, matching, and shaping). Current advances [1, 2, 3] in the eld exploit transforms such as wavelet decomposi- tions and trigonometric transforms (DCT, DFT, etc.) to improve the efciency of the sharpening. The sharpen- ing is typically achieved by amplifying certain parts of the transform spectrum. Example of a well established such technique is the alpha-rooting [4, 5], where the amplica- tion is achieved by taking the α-root of the magnitude of the original coefcients, for some constant α> 1. A re- cent work [3] showed that histograms of the logarithm of a transform spectrum can also be utilized for sharpening using methods such as histogram matching and shaping. This work was supported by the Academy of Finland, project No. 213462 (Finnish Centre of Excellence program [2006 - 2011]). An inherent drawback of most sharpening methods is the amplication of the noise component that is inevitably present when dealing with practical applications. Recently, we proposed a highly effective image de- noising method [6], namely the block-matching and 3D ltering (BM3D), based on ltering of enhanced sparse non-local image representations. Using block-wise process- ing, we group blocks similar to the currently processed one into 3D arrays, termed groups. Subsequently, we apply what we call collaborative ltering on each of these groups. In [6] we showed that this ltering can be efciently realized by shrinkage (e.g., hard-thresholding) in 3D-transform domain. It is exactly in the 3D-transform domain where the enhanced sparse representation of the true signal is attained. Hence, the shrinkage of the trans- form coefcients allows for both good noise attenuation and faithful detail preservation. The result of the collab- orative ltering of a group is a set of grouped block esti- mates, which are then returned to their original location in the image domain. Since the block estimates from a given group can mutually overlap and also overlap with ones from different groups, we aggregate them by a weighted averaging in order to obtain a single estimate of each im- age pixel. In this paper we propose to combine the BM3D de- noising approach with alpha-rooting, in order to simulta- neously sharpen image details and attenuate noise. Hard- thresholding is applied on the 3D-transform spectrum of each group to attenuate the noise. Subsequently, sharp- ening is realized by modifying the thresholded transform coefcients of the grouped blocks. We show experimental results corresponding to two approaches, one with alpha- rooting performed on the 3D transform spectrum and the other with alpha-rooting applied separately on the 2D trans- form spectrum of each grouped block. The joint applica- tion of the collaborative ltering and alpha-rooting allows for both good noise suppression and effective preserva- tion and sharpening of even very ne image details. It is worth noting that the adopted 3D transform is a separa- ble composition of a 2D transform (on each block) and a 1D transform (in the temporal dimension, along which blocks are stacked), which makes possible the application of alpha-rooting on 2D transform spectra. In the following sections we present the developed method and show that

Upload: others

Post on 31-May-2020

3 views

Category:

Documents


0 download

TRANSCRIPT

Page 1: JOINT IMAGE SHARPENING AND DENOISING BY 3D TRANSFORM-DOMAIN COLLABORATIVE FILTERINGfoi/GCF-BM3D/BM3D-Sharp-SMMSP07.pdf · 2007-07-31 · JOINT IMAGE SHARPENING AND DENOISING BY 3D

JOINT IMAGE SHARPENING AND DENOISING BY 3D TRANSFORM-DOMAINCOLLABORATIVE FILTERING

Kostadin Dabov, Alessandro Foi, Vladimir Katkovnik, and Karen Egiazarian

Institute of Signal Processing, Tampere University of Technology,P.O. Box 553, Tampere 33101, Finland

Þrstname.lastname@tut.Þ

ABSTRACTIn order to simultaneously sharpen image details and at-tenuate noise, we propose to combine the recent block-matching and 3D Þltering (BM3D) denoising approach,based on 3D transform-domain collaborative Þltering, withalpha-rooting, a transform-domain sharpening technique.The BM3D exploits grouping of similar image blocks into3D arrays (groups) on which collaborative Þltering (byhard-thresholding) is applied. We propose two approachesof sharpening by alpha-rooting; the Þrst applies alpha-rooting individually on the 2D transform spectra of eachgrouped block; the second applies alpha-rooting on the3D-transform spectra of each 3D array in order to sharpenÞne image details shared by all grouped blocks and furtherenhance the interblock differences. The conducted experi-ments with the proposed method show that it can preserveand sharpen Þne image details and effectively attenuatenoise.

1. INTRODUCTION

The problem of sharpening image details while attenuat-ing noise arises in practice as a preprocessing step prior tosegmentation, binarization (by thresholding) in the con-text of image analysis, classiÞcation, pattern recognition.The procedure is also very useful in consumer applica-tions, where it is utilized to obtain more visually appealingdigital photographs. Various methods for image sharp-ening exist. Traditional ones rely on linear Þltering thatboosts higher frequencies or on elementwise transforma-tions such as the histogram-based methods (equalization,matching, and shaping). Current advances [1, 2, 3] inthe Þeld exploit transforms such as wavelet decomposi-tions and trigonometric transforms (DCT, DFT, etc.) toimprove the efÞciency of the sharpening. The sharpen-ing is typically achieved by amplifying certain parts of thetransform spectrum. Example of a well established suchtechnique is the alpha-rooting [4, 5], where the ampliÞca-tion is achieved by taking the α-root of the magnitude ofthe original coefÞcients, for some constant α > 1. A re-cent work [3] showed that histograms of the logarithm ofa transform spectrum can also be utilized for sharpeningusing methods such as histogram matching and shaping.

This work was supported by the Academy of Finland, project No.213462 (Finnish Centre of Excellence program [2006 - 2011]).

An inherent drawback of most sharpening methods isthe ampliÞcation of the noise component that is inevitablypresent when dealing with practical applications.Recently, we proposed a highly effective image de-

noising method [6], namely the block-matching and 3DÞltering (BM3D), based on Þltering of enhanced sparsenon-local image representations. Using block-wise process-ing, we group blocks similar to the currently processedone into 3D arrays, termed �groups�. Subsequently, weapply what we call �collaborative Þltering� on each ofthese groups. In [6] we showed that this Þltering can beefÞciently realized by shrinkage (e.g., hard-thresholding)in 3D-transform domain. It is exactly in the 3D-transformdomain where the enhanced sparse representation of thetrue signal is attained. Hence, the shrinkage of the trans-form coefÞcients allows for both good noise attenuationand faithful detail preservation. The result of the collab-orative Þltering of a group is a set of grouped block esti-mates, which are then returned to their original location inthe image domain. Since the block estimates from a givengroup can mutually overlap and also overlap with onesfrom different groups, we aggregate them by a weightedaveraging in order to obtain a single estimate of each im-age pixel.In this paper we propose to combine the BM3D de-

noising approach with alpha-rooting, in order to simulta-neously sharpen image details and attenuate noise. Hard-thresholding is applied on the 3D-transform spectrum ofeach group to attenuate the noise. Subsequently, sharp-ening is realized by modifying the thresholded transformcoefÞcients of the grouped blocks. We show experimentalresults corresponding to two approaches, one with alpha-rooting performed on the 3D transform spectrum and theother with alpha-rooting applied separately on the 2D trans-form spectrum of each grouped block. The joint applica-tion of the collaborative Þltering and alpha-rooting allowsfor both good noise suppression and effective preserva-tion and sharpening of even very Þne image details. It isworth noting that the adopted 3D transform is a separa-ble composition of a 2D transform (on each block) and a1D transform (in the �temporal� dimension, along whichblocks are stacked), which makes possible the applicationof alpha-rooting on 2D transform spectra. In the followingsections we present the developed method and show that

Page 2: JOINT IMAGE SHARPENING AND DENOISING BY 3D TRANSFORM-DOMAIN COLLABORATIVE FILTERINGfoi/GCF-BM3D/BM3D-Sharp-SMMSP07.pdf · 2007-07-31 · JOINT IMAGE SHARPENING AND DENOISING BY 3D

it can be very effective when applied on both artiÞciallydegraded images and real retinal medical images.

2. PROPOSEDMETHOD

We consider a noisy image z = y+η, where y is the noise-free image (with poor contrast) and η is i.i.d. Gaussiannoise with zero mean and variance σ2. Following arethe two variations of the proposed method, denominatedBM3D-SH3D and BM3D-SH2D. The former performs alpha-rooting on 3D transform spectra and the latter on 2D trans-form spectra. Their corresponding ßowcharts are shownin Figure 1.

1. Process overlapping blocks in a raster scan. Foreach such block, do the following:

(a) Use block-matching to Þnd the locations ofthe blocks in z that are similar to the currentlyprocessed one. Form a 3D array (group) bystacking the blocks located at the obtained lo-cations.

(b) Apply a 3D transform on the formed group.(c) Attenuate the noise by hard-thresholding the

3D transform spectrum.(d) Sharpening. We propose the following two al-

ternatives.

� BM3D-SH3D. Apply alpha-rooting on thehard-thresholded 3D transform spectrumand invert the 3D transform to produceÞltered grouped blocks.

� BM3D-SH2D. Invert the 1D transformsalong the temporal dimension of the form-ed 3D array, then perform alpha-rootingseparately on the 2D transform spectra ofeach grouped block and subsequently in-vert the 2D transforms to produce Þlteredgrouped blocks.

2. Return the Þltered blocks to their original locationsin the image domain and compute the resultant Þl-tered image by a weighted average of these Þlteredblocks.

Except for the alpha-rooting and a modiÞcation of the ag-gregation weights, both described below, the rest of thesteps of the algorithm are taken without modiÞcation fromthe Þrst step of BM3D [6] (the one that uses hard-thresho-lding). We refer the reader to [6] for details.Given a transform spectrum t of a signal, which con-

tains a DC coefÞcient termed t (0), the alpha-rooting isperformed as

tsh (i) =

sign [t (i)] |t (0)|¯̄̄t(i)t(0)

¯̄̄ 1α

, if t (0) 6= 0t (i) , otherwise,

(1)where tsh is the spectrum of the resultant signal and wherean exponent α > 1 results in sharpening of image details

[5]. This simple technique is applied for both the BM3D-SH3D and the BM3D-SH2D on 3D and 2D transformspectra, respectively. The alpha-rooting is motivated byfact that, except for the DC, the transform basis elementsextract differential information of the signal. Its effec-tiveness is based on amplifying these coefÞcients accord-ing to Equation (1). Performing hard-thresholding priorto alpha-rooting resolves problems [7] of this sharpen-ing technique regarding noise ampliÞcation (which causesspike artifacts).As shown in [6], the overlapping Þltered blocks can

be effectively aggregated using weights that are inverselyproportional to the total variance of the Þltered groups towhich the respective blocks belong. While for the denois-ing method described there, this total variance can be ap-proximated as σ2 times the number of retained (i.e. non-zero) coefÞcients after thresholding, in the case of jointdenoising and sharpening, due to the ampliÞcation causedby alpha-rooting, the variance of each sharpened coefÞ-cient in the group could be different from the original σ2.We derive a simple estimator for the total residual vari-

ance in each group. Let t be the hard-thresholded 3Dtransform spectrum of a group and tsh be the spectrum ofthe sharpened signal obtained by Equation 1. If t (0) 6= 0,we can rewrite tsh (i) as the product of two independentterms *0 = |t (0)|1− 1

α and ρi = sign [t (i)] |t (i)| 1α . As-suming that the moduli of t (0) and t (i) are much largerthan their standard deviation σ (which can be a reasonableassumption, since t (0) is a DC coefÞcient and t (i) sur-vived thresholding), the variances of the product of thesetwo terms can be roughly approximated (using the Þrstpartial derivatives) as follows

var {tsh (i)} ' (*00 (E {t (0)})E {ρi})2 var {t (0)}++(ρ0i (E {t (i)})E {*0})2 var {t (i)}

'µ1− 1

α

¶2|E {t (0)}|− 2

α E2 {ρi}σ2 +

+1

α2|E {t (i)}| 2α−2E2 {*0}σ2

'µ1− 1

α

¶2 ¯̄̄̄E {t (i)}E {t (0)}

¯̄̄̄ 2α

σ2 +

+1

α2|E {t (i)}| 2α−2 |E {t (0)}|2− 2

α σ2

'µ1− 1

α

¶2|t (0)|− 2

α |t (i)| 2α σ2 +

+1

α2|t (i)| 2α−2 |t (0)|2− 2

α σ2

= ωiσ2

Thus, the total variance of the Þltered (i.e., thresholdedand sharpened) group is approximated as

v = σ2 +X

t(i)6=0,i>0ωiσ

2. (2)

Consequently, the weights used for the aggregation aredeÞned as the reciprocal of the above v. Note that these

Page 3: JOINT IMAGE SHARPENING AND DENOISING BY 3D TRANSFORM-DOMAIN COLLABORATIVE FILTERINGfoi/GCF-BM3D/BM3D-Sharp-SMMSP07.pdf · 2007-07-31 · JOINT IMAGE SHARPENING AND DENOISING BY 3D

BM3D-SH3D WITH ALPHA-ROOTING PERFORMED ON 3D SPECTRA

BM3D-SH2D WITH ALPHA-ROOTING PERFORMED ON THE 2D SPECTRA OF GROUPED BLOCKS

Figure 1. Flowcharts of the two variations of the proposed method. Top: BM3D-SH3D, which applies alpha-rooting onthe 3D spectrum of each group; bottom: BM3D-SH2D, which applies alpha-rooting on the 2D spectrum of each groupedblock.

weights are used only for BM3D-SH3D and for the BM3D-SH2D, for simplicity, we resort to the same weights usedin BM3D [6]. (i.e. ωi = 1). Further, one can observe thatif α = 1, we obtain ωi = 1. Thus, (2) can be interpretedas a generalization of the weights used in [6].

3. RESULTS

We present experimental results from the two variationsof the proposed method. Their implementation along withthe original noise-free and the Þltered images are avail-able online1. We used the same algorithm parameters as inBM3D [6]; one can refer to the provided Matlab codes foradditional details. The exponent α sets the desired levelof sharpening. An illustration of applying the proposedmethod for a few values of α is given in Figure 2. If notspeciÞed otherwise, in our experiments we used α = 1.5.In Figures 4 and 5 one can see the results of applying

the proposed method on images degraded by noise withσ = 10, 20. It can be observed that Þne details are wellpreserved and sharpened while the noise is suppressed. Itis worth recalling that the BM3D is particularly effectivefor objects that are formed by blocks for which there canbe found plenty of similar blocks at different spatial lo-cations. Such objects are edges, repeating patterns, andtextures. Hence, these (e.g., the repeating circular objectsin Harbour and the edges of the building in Pentagon) arewell preserved and subsequently sharpened.

1Matlab code available at www.cs.tut.fi/~foi/GCF-BM3D.

Figure 3 shows an example of denoising and sharpen-ing a fragment from a real retinal image with added noiseof σ = 10, 20. The resultant Þltered images preserve andsharpen most of the image details while efÞciently sup-press noise. Moreover, the Þltered images clearly revealdetails which are hard to see even in the original image.

4. DISCUSSION AND CONCLUSIONS

We developed a method for joint denoising and sharpen-ing of grayscale images. It inherits the outstanding denois-ing potential of the BM3D [6] and in combination withalpha-rooting, it achieves perceptually appealing Þlteredimages where Þne details are effectively sharpened.Let us discuss the two variations of the proposedmethod.

The BM3D-SH3D differs fromBM3D-SH2D in that alpha-rooting is applied on the 3D transform spectrum whichconveys information of the temporal differences in a gr-oup in addition to the spatial one (we call �temporal� thedimension along which blocks are stacked together in agroup). In this 3D transform domain, the application ofEquation (1) can amplify also differential information alongthis temporal dimension (and not along the spatial ones).It is noteworthy that in the case of ideal grouping, i.e.having identical blocks in a group, the two approachesproduce identical Þltered blocks (and differ only in theweighted aggregation of these blocks). The visual resultsof both approaches (Figures 4, 5, 3, 7) are quite similarby subjective evaluation. Therefore, we abstain from con-cluding which of them is better in practice. Instead, we

Page 4: JOINT IMAGE SHARPENING AND DENOISING BY 3D TRANSFORM-DOMAIN COLLABORATIVE FILTERINGfoi/GCF-BM3D/BM3D-Sharp-SMMSP07.pdf · 2007-07-31 · JOINT IMAGE SHARPENING AND DENOISING BY 3D

Original House Noisy House, σ = 10 BM3D-SH2D, α = 1.2 BM3D-SH3D, α = 1.2

BM3D-SH2D, α = 1.4 BM3D-SH3D, α = 1.4 BM3D-SH2D, α = 1.6 BM3D-SH3D, α = 1.6

BM3D-SH2D, α = 1.8 BM3D-SH3D, α = 1.8 BM3D-SH2D, α = 2.0 BM3D-SH3D, α = 2.0

Figure 2. Result of Þltering with BM3D-SH2D and BM3D-SH3D for α = 1.2, 1.4, 1.6, 1.8, 2.0. Higher values of αcorrespond to stronger sharpening of the details. The processed image is House with noise of standard deviation σ = 10.

only point to some of the noticeable differences. In the en-larged fragment of Pentagon in Figure 5, one can observethat for σ = 10 the BM3D-SH3D suppresses slightly bet-ter the noise in the smooth areas, mainly due to the aggre-gation weights deÞned from the total variance estimate inEquation (2). In Figure 7, we see that the BM3D-SH2Dcan preserve more effectively the oscillating curve. Inaddition, Figure 6 presents a plot of line 365 of Penta-gon, which shows that BM3D-SH3D attains slightly bettersharpening results than the BM3D-SH2D.The proposed method can be extended for color data,

where the denoising is performed on all three luminance-chrominance color channels as in [8] and the alpha-rootingis performed on all channels or only on the luminance one,depending on the application.

5. REFERENCES

[1] D.-Y. Tsai, Y. L. M. Sekiya, S. Sakaguchi, and I. Ya-mada, �A method of medical image enhancement us-

Figure 7. First row: noise-free artiÞcial test images 24×96pixels; second row: corresponding noisy images withσ = 15; third row: results of applying BM3D-SH2D;fourth row: results of applying BM3D-SH3D. For the ex-periments in this Þgure, both algorithms used α = 2 andthe sliding step of the BM3D was set to unity.

ing wavelet analysis,� in Proc. IEEE Int. Conf. SignalProcess., vol. 1, August 2002, pp. 723�726.

Page 5: JOINT IMAGE SHARPENING AND DENOISING BY 3D TRANSFORM-DOMAIN COLLABORATIVE FILTERINGfoi/GCF-BM3D/BM3D-Sharp-SMMSP07.pdf · 2007-07-31 · JOINT IMAGE SHARPENING AND DENOISING BY 3D

Original σ = 10 σ = 20

Filtered by BM3D-SH2D:

Filtered by BM3D-SH3D:

Figure 3. Results of applying the proposed methods on a fragment of a retinal fundus image. The second and third columscorrespond to noise with σ = 10 and σ = 20, respectively.

[2] S. Hatami, R. Hosseini, M. Kamarei, and H. Ahmadi,�Wavelet based Þngerprint image enhancement,� inProc. IEEE Int. Symp. Circ. Syst. (ISCAS), Kobe,Japan, May 2005.

[3] S. Agaian, B. Silver, and K. A. Panetta, �TransformcoefÞcient histogram-based image enhancement al-

gorithms using contrast entropy,� IEEE Trans. ImageProcess., vol. 16, no. 3, pp. 741�758, March 2007.

[4] W. K. Pratt, Digital Image Processing. John Wileyand Sons, New York, 1978.

[5] S. Aghagolzadeh and O. K. Ersoy, �Transform image

Page 6: JOINT IMAGE SHARPENING AND DENOISING BY 3D TRANSFORM-DOMAIN COLLABORATIVE FILTERINGfoi/GCF-BM3D/BM3D-Sharp-SMMSP07.pdf · 2007-07-31 · JOINT IMAGE SHARPENING AND DENOISING BY 3D

Original σ = 10 σ = 20

Filtered by BM3D-SH2D:

Filtered by BM3D-SH3D:

Filtered by BM3D-SH2D:

Filtered by BM3D-SH3D:

Figure 4. Results of applying the proposed methods on the noisy Harbour image (upper half) and a fragment of it (lowerhalf). The second and third colums correspond to noise with σ = 10 and σ = 20, respectively.

Page 7: JOINT IMAGE SHARPENING AND DENOISING BY 3D TRANSFORM-DOMAIN COLLABORATIVE FILTERINGfoi/GCF-BM3D/BM3D-Sharp-SMMSP07.pdf · 2007-07-31 · JOINT IMAGE SHARPENING AND DENOISING BY 3D

Original σ = 10 σ = 20

Filtered by BM3D-SH2D:

Filtered by BM3D-SH3D:

Filtered by BM3D-SH2D:

Filtered by BM3D-SH3D:

Figure 5. Results of applying the proposed methods on the noisy Pentagon image (upper half) and a fragment of it (lowerhalf). The second and third colums correspond to noise with σ = 10 and σ = 20, respectively.

Page 8: JOINT IMAGE SHARPENING AND DENOISING BY 3D TRANSFORM-DOMAIN COLLABORATIVE FILTERINGfoi/GCF-BM3D/BM3D-Sharp-SMMSP07.pdf · 2007-07-31 · JOINT IMAGE SHARPENING AND DENOISING BY 3D

Figure 6. Line 365 of Pentagon image for: the noise-free image y, a noisy (σ = 10) image z, the result of BM3D denoisingwith no sharpening, the result of joint denoising and sharpening with BM3D-SH2D, and the result of joint denoising andsharpening with BM3D-SH3D.

enhancement,� Opt. Eng., vol. 31, no. 3, pp. 614�626,March 1992.

[6] K. Dabov, A. Foi, V. Katkovnik, and K. Egiazarian,�Image denoising by sparse 3D transform-domaincollaborative Þltering,� IEEE Trans. Image Process.,vol. 16, no. 8, August 2007.

[7] J. McClellan, �Artifacts in alpha-rooting of images,�in Proc. IEEE Int. Conf. Acoust. Speech SignalProcess. (ICASSP), vol. 5, April 1980, pp. 449�452.

[8] K. Dabov, A. Foi, V. Katkovnik, and K. Egiazar-ian, �Color image denoising via sparse 3D collabo-rative Þltering with grouping constraint in luminance-chrominance space,� in Proc. IEEE Int. Conf. ImageProcess., San Antonio, Texas, September 2007.