Signal Denoising Using the Minimum-Probability-of-Error Criterion
Jishnu Sadasivan, Subhadip Mukherjee, and Chandra Sekhar Seelamantula

TL;DR
This paper introduces a novel signal denoising method based on the minimum probability of error (MPE) criterion, which outperforms traditional MSE-based methods, especially at low SNR levels, by minimizing the likelihood of large estimation errors.
Contribution
The paper proposes a new transform-domain shrinkage approach using the MPE criterion, extending it with subband grouping, and demonstrates superior performance over existing methods in low SNR scenarios.
Findings
Outperforms MSE-based denoising methods at low SNRs
Achieves consistent SNR gains below 5 dB in tests
Effective across various noise distributions using Gaussian mixture models
Abstract
We address the problem of signal denoising via transform-domain shrinkage based on a novel criterion called the minimum probability of error (MPE), which measures the probability that the estimated parameter lies outside an -neighborhood of the actual value. However, the MPE, similar to the mean-squared error (MSE), depends on the ground-truth parameter, and has to be estimated from the noisy observations. We consider linear shrinkage-based denoising functions, wherein the optimum shrinkage parameter is obtained by minimizing an estimate of the MPE. When the probability of error is integrated over , it leads to the expected distortion. The proposed MPE and distortion formulations are applicable to various noise distributions by invoking a Gaussian mixture model approximation. Within the realm of MPE, we also develop an extension of…
| Input SNR | Output SNR (dB) | ||||
|---|---|---|---|---|---|
| (dB) | MPE | SURE | WF | ||
| 3.5 | 2.5 | 1.5 | |||
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Signal Denoising Using the Minimum-Probability-of-Error Criterion
Jishnu Sadasivan, Subhadip Mukherjee, and Chandra Sekhar Seelamantula, *Senior member, IEEE * J. Sadasivan is with the Department of Electrical Communication Engineering, Indian Institute of Science, Bangalore, India. Phone: +91 80 2293 2276. Fax: +91 80 2360 0563. E-mail: [email protected]. S. Mukherjee and C. S. Seelamantula are with the Department of Electrical Engineering, Indian Institute of Science, Bangalore, India. Phone: +91 80 2293 2695. Fax: +91 80 2360 0444. E-mails: {subhadip, chandra.sekhar}@ee.iisc.ernet.in.
Abstract
We address the problem of signal denoising via transform-domain shrinkage based on a novel risk criterion called the minimum probability of error (MPE), which measures the probability that the estimated parameter lies outside an -neighborhood of the actual value. However, the MPE, similar to the mean-squared error (MSE), depends on the ground-truth parameter, and has to be estimated from the noisy observations. We consider linear shrinkage-based denoising functions, wherein the optimum shrinkage parameter is obtained by minimizing an estimate of the MPE. When the probability of error is integrated over , it leads to the expected distortion. The proposed MPE and distortion formulations are applicable to various noise distributions by invoking a Gaussian mixture model approximation. Within the realm of MPE, we also develop an extension of the transform-domain shrinkage by grouping transform coefficients, resulting in subband shrinkage. The denoising performance obtained within the proposed framework is shown to be better than that obtained using the minimum MSE-based approaches formulated within Stein’s unbiased risk estimation (SURE) framework, especially in the low measurement signal-to-noise ratio (SNR) regime. Performance comparison with three state-of-the-art denoising algorithms, carried out on electrocardiogram signals and two test signals taken from the Wavelab toolbox, exhibits that the MPE framework results in consistent SNR gains for input SNRs below dB.
Index Terms:
Minimum probability of error, shrinkage estimator, risk estimation, transform-domain shrinkage, subband shrinkage, expected distortion, Gaussian mixture model.
I Introduction
Signal denoising algorithms are often developed with the objective of minimizing the mean-squared error (MSE) between an estimate and the ground-truth, which may be deterministic or stochastic with a known prior. The latter formalism leads to Bayesian estimators. Within the deterministic signal estimation paradigm, which is also the formalism considered in this paper, one typically desires that the estimator has minimum variance and is unbiased (MVU) [1, 2]. An MVU estimator may not always exist, and if it does, it can be obtained using the theory of sufficient statistics. Eldar and Kay [2] showed that, when it comes to minimizing the MSE, biased estimates may outperform the MVU estimate. For example, one could shrink the MVU estimate and optimize for the shrinkage parameter so that the MSE is minimum.
In this paper, we consider the problem of estimating a deterministic signal corrupted by additive white noise. The noise distribution is assumed to be known, but not restricted to be Gaussian. We propose a new distortion metric based on the probability of error and develop estimators using a transform-domain shrinkage approach. Before proceeding with the developments, we review some important literature related to the problem at hand.
I-A Prior Art
The MSE is by far the most widely used metric for obtaining the optimum shrinkage parameter. Since the MSE is a function of the parameter to be estimated, direct minimization might result in an unrealizable estimate, in the sense that it might depend on the unknown parameter. However, in some cases, it is possible to find the optimum shrinkage parameter, for example, using a min-max approach [2], where the parameter is constrained to a known set. An optimum shrinkage estimator, when the variance of the unbiased estimate (or MVU) is a scaled version of the square of the parameter, with a known scaling, is proposed in [2].
Optimum shrinkage estimators have also been computed based on risk estimation, where an unbiased estimate of the MSE that depends only on the noisy observations is obtained and subsequently minimized over the shrinkage parameter. Under the assumption of Gaussian noise, an unbiased estimate of the MSE, namely Stein’s unbiased risk estimator (SURE), was developed based on Stein’s lemma [3], and has been successfully employed in numerous denoising applications. In his seminal work [3], Stein proved that the shrinkage estimator of the mean of a multivariate Gaussian distribution, obtained from its independent and identically distributed (i.i.d.) samples by minimizing SURE, dominates the classical least-squares estimate when the number of samples exceeds three [4].
A risk minimization approach for denoising using a linear expansion of elementary thresholding functions has been addressed in [5, 6, 7, 8, 9], wherein the combining weights are chosen optimally to minimize the SURE objective. SURE-optimized wavelet-domain thresholding techniques have been developed in [10, 11, 12]. Atto et al. [13, 14] have investigated the problem of signal denoising based on optimally selecting the parameters of a wavelet-domain smooth sigmoidal shrinkage function by minimizing the SURE criterion. The use of SURE objective is not restricted to denoising; it has found applications in image deconvolution as well [15].
Ramani et al. [16] developed a Monte-Carlo technique to select the parameters of a generic denoising operator based on SURE. An image denoising algorithm based on non-local means (NLM) is proposed in [17], where parameters of NLM are optimized using SURE. Notable denoising algorithms that aim to optimize the SURE objective include wavelet-domain multivariate shrinkage [18], local affine transform for image denoising [19], optimal basis selection for denoising [20], raised-cosine-based fast bilateral filtering for image denoising [21], SURE-optimized Savitzky-Golay filter [22], etc..
The original formulation of SURE, which assumed independent Gaussian noise was extended to certain distributions in continuous and discrete exponential families in [23] and [24], respectively, with the assumption of independence left unchanged. Eldar generalized SURE (GSURE) for distributions belonging to the non-i.i.d. multivariate exponential family [25]. Giryes et al. [26] used a projected version of GSURE for selecting parameters in the context of solving inverse problems. An unbiased estimate of the Itakura-Saito (IS) distortion and corresponding pointwise shrinkage was developed in [27] and [28], and successfully applied to speech denoising. A detailed discussion of Gaussian parameter estimation using shrinkage estimators, together with a performance comparison of SURE with the maximum-likelihood (ML) and soft-thresholding-based estimators, can be found in [29] (Chapter ). It is shown in [29] that the soft-thresholding-based estimator dominates the James-Stein shrinkage estimator in terms of MSE if the parameter vector to be estimated is sparse. On the other hand, shrinkage estimator dominates if all coordinates of the parameter to be estimated are nearly equal.
I-B This Paper
We address the problem of signal denoising based on the minimum probability of error (MPE), which we first considered in [30]. The MPE quantifies the probability of the estimate lying outside an -neighborhood of the true value. Since the MPE risk depends on the ground truth, we consider a surrogate, which may be biased, and optimize it to obtain the shrinkage parameter (Section II). The optimization is carried out in the discrete cosine transform (DCT) domain, either in a pointwise fashion or on a subband basis. We derive the MPE risk for Gaussian, Laplacian, and Student’s- noise distributions (Sections II-A and II-B). In practical applications, where the noise distribution may be multimodal and not known explicitly, we propose to use a Gaussian mixture model (GMM) approximation [31, 32] (Section II-A3). We show the performance of the MPE-based denoising technique on the Piece-Regular signals taken from the Wavelab toolbox in Gaussian, Student’s-, and Laplacian noise contaminations (Section III). Proceeding further, we also consider the probability of error accumulated over (Section IV), which results in the expected distortion between the parameter and its estimate. The estimators for the expected distortion are also derived by invoking the GMM approximation (Section IV-A). We also assess the denoising performance of the shrinkage estimator obtained by minimizing the distortion for different input SNRs and for different number of noisy realizations (Section V).
To further boost the denoising performance of the distortion-based estimator, we develop an iterative algorithm to successively refine the cost function and the resulting estimate, starting with the noisy signal as the initialization (Section V). The iterations lead to an improvement of – dB in output signal-to-noise ratio (SNR) (Section V-A).
Performance comparison of the MPE and distortion-based estimators is carried out on the Piece-Regular and the HeaviSine signals from the Wavelab toolbox [33], and electrocardiogram (ECG) signals from the PhysioBank database [42], with three benchmark techniques: (i) wavelet-domain soft-thresholding [34], (ii) SURE-based orthonormal wavelet thresholding using a linear expansion of thresholds (SURE-LET) [5], and (iii) SURE-based smooth sigmoid shrinkage (SS) in wavelet domain[13]; all assuming Gaussian noise contamination (Section VI) for fair comparison.
II The MPE Risk
Consider the observation model in , where and denote the noisy and clean signals, respectively. The noise vector is assumed to have i.i.d. entries with zero mean and variance . The goal is to estimate from by minimizing a suitable risk function. The signal model is considered in an appropriate transform domain, where the signal admits a parsimonious representation, but noise does not. We consider two types of shrinkage estimators: (i) pointwise, where a shrinkage factor is applied to to obtain an estimate ; and (ii) subband-based, wherein a single shrinkage factor is applied to a group of coefficients {, } in subband . Shrinkage estimators may also be interpreted as premultiplication of with a diagonal matrix.
II-A MPE Risk for Pointwise Shrinkage
Assuming that the estimate of does not depend on , for , we drop the index for brevity of notation. The MPE risk is defined as
[TABLE]
where is a predefined tolerance parameter. The risk quantifies the estimation error using the probability measure and takes into account the noise distribution in its entirety. On the contrary, the MSE relies only on the first- and second-order statistics of noise for linear shrinkage estimators. Substituting , the risk evaluates to
[TABLE]
where is the cumulative distribution function (c.d.f.) of the additive noise. Since depends on , which is the parameter to be estimated, it is impractical to optimize it directly over . To circumvent the problem, we minimize an estimate of , which is obtained by replacing with an estimate , which, for example, may be obtained using any baseline denoising algorithm, or can even be taken as (which is also the ML estimate of ). In the first instance, the proposed technique becomes an add-on to an existing denoising algorithm, and in the second, it is a denoising scheme in itself. Such an estimate takes the form
[TABLE]
and correspondingly, the optimal shrinkage parameter is obtained as . A grid search is performed to optimize over , and the clean signal is obtained as . We next derive explicit formulae for the risk function for Gaussian, Laplacian, and Student’s- noise distributions.
(i) Gaussian distribution: In this case, the noisy observation also follows a Gaussian distribution, and therefore, is distributed as . The MPE risk estimate is given as
[TABLE]
where .
(ii) Student’s- distribution: Consider the case where the noise follow a Student’s- distribution with parameter and the probability density function (p.d.f.) of noise is given by
[TABLE]
The variance of is . The expression for is the one given in (3) with
[TABLE]
where is the hypergeometric function defined as
[TABLE]
and denotes the Pochhammer symbol:
[TABLE]
(iii) Laplacian distribution: Considering the noise to be i.i.d. Laplacian with zero-mean and parameter (variance ), with the p.d.f. , the MPE risk can be obtained by using the following expression for in (3):
[TABLE]
II-A1 Closeness of to
To measure the closeness of to , consider the example of estimating a scalar from a noisy observation . The MPE risk estimate is obtained by setting . In Figures 1(a), 1(b), and 1(c), we show the variation of the actual risk and its estimate with , averaged over independent trials, for Gaussian, Student’s-, and Laplacian noise distributions, respectively. The noise has zero mean, and the variance is taken as for Gaussian and Laplacian models, whereas for Student’s- model, the variance is . The value of is set equal to while computing the MPE risk. We observe that is a good approximation to , particularly in the vicinity of the minima. The deviation of the shrinkage parameter , obtained by minimizing , with respect to its true value resulted from the minimization of , is shown in Figure 1(d) for three noise models under consideration. The central red lines in Figure 1(d) indicate the medians, whereas the black lines on the top and bottom denote the and the percentile points, respectively. We observe that is well concentrated around , especially for Gaussian and Laplacian noise, barring a small number of outliers.
II-A2 Perturbation Probability of the location of minimum
The location of the minimum of the MPE risk determines the shrinkage parameter. Therefore, one must ensure that it does not deviate too much from its actual value, with high probability, when is replaced by in the original risk . Let denote the argument that minimizes the true risk . Consider the probability of deviation, given by
[TABLE]
for some . Using a first-order Taylor series approximation of about , and substituting , we obtain , where ′ denotes the derivative. The deviation probability in (7) simplifies to For additive Gaussian noise with zero mean and variance , placing the Chernoff bound on leads to
[TABLE]
To ensure that is less than , for a given , it suffices to have
[TABLE]
which translates to a lower-bound on the input SNR. Since there is no closed-form expression available for in the context of MPE risk, we empirically obtain the range of input SNR values , for which (8) is satisfied.
Analogously, to satisfy an upper bound on the deviation probability of the minimum in the case of SURE, for a given deviation , one must ensure that
[TABLE]
The proof of (9) is given in Appendix A.
The minimum input SNR required to ensure for both SURE- and MPE-based shrinkage estimators is shown in Figure 2, for different values of and . The MPE-risk estimate is obtained by replacing with and setting . We observe that reducing the amount of deviation for a given probability , or vice versa, leads to a higher input SNR requirement for both SURE and MPE. We also observe from Figure 2 that, for given and , SURE requires a higher input SNR than MPE to keep the -deviation probability under . Also, for a given input SNR, the -deviation probability of the estimated shrinkage parameter from the optimum is smaller for MPE than SURE, thereby indicating that the MPE-based shrinkage is comparatively more reliable than the SURE-based one at lower input SNRs.
II-A3 Unknown noise distributions
In practical applications, the distribution of noise may not be known in a parametric form and may also be multimodal. At best, one would have access to realizations of the noise, from which the distribution has to be estimated. In such cases, approximation of the noise p.d.f. using a GMM is a viable alternative [31], wherein one can estimate the parameters of GMM using the expectation-maximization algorithm [35]. Gaussian mixture modeling is attractive as it comes with certain guarantees. For example, it is known that a p.d.f. with a finite number of finite discontinuities can be approximated by a GMM to a desired accuracy except at the points of discontinuity [32, 31]. The GMM approximation can be used even for non-Gaussian, unimodal distributions. For the GMM-based noise p.d.f.
[TABLE]
the MPE risk turns out to be
[TABLE]
using (4). For illustration, consider the estimation of a scalar in the transform domain from its noisy observation . The additive noise is Laplacian distributed with zero mean and variance . The noise distribution is modeled using a GMM with components and the corresponding MPE risk estimate is obtained using (11) by setting . In Figure 3(a), we show a Laplacian p.d.f. and its GMM approximation. Figure 3(b) shows the GMM approximation to a multimodal distribution. Figure 4(a) shows the MPE risk based on the original Laplacian distribution as well as the GMM approximation, as a function of the shrinkage parameter . The close match between the two indicates that the GMM is a viable alternative when the noise distribution is unknown or follows a complicated model. In Figure 4(b), we plot the GMM-based MPE risk and its estimate averaged over independent trials. We observe that the locations of minima of the actual risk and its estimate match closely, thereby justifying the minimization of . The MPE risk and its estimate are shown in Figure 4(c) for the multimodal p.d.f. of Figure 3(b).
II-B MPE Risk for Subband Shrinkage
Let be the shrinkage factor applied to the set of coefficients {, } in subband . The estimate of the clean signal is obtained by , where and . For notational brevity, we drop the subscript , as we did for pointwise shrinkage, and express the estimator as , where boldface letters indicate vectors.
Analogous to pointwise shrinkage, the MPE risk for subband shrinkage is defined as which, for , becomes . For ,
[TABLE]
where , , , and is the c.d.f. of the non-central distribution, given by
[TABLE]
wherein denotes the central random variable having degrees of freedom.
Similar to pointwise shrinkage, we propose to obtain an estimate of for subband shrinkage estimators either by replacing with , or by an estimate produced by any standard denoising algorithm. The optimum subband shrinkage factor is obtained by minimizing
Figure 5 shows the subband MPE risk and its estimate versus , where the underlying clean signal is corrupted by Gaussian noise and the subband size is chosen to be . The clean signal is generated by drawing samples from , where and denote a -length vector of all ones and a identity matrix, respectively. The observation is obtained by adding zero-mean i.i.d. Gaussian noise to , with an input SNR of dB, where the input SNR is defined as . The MPE risk estimate is obtained by replacing with in (12), which does not significantly shift the location of the minimum (cf. Figure 5).
III Experimental Results for MPE-Based Denoising
The performance of the MPE-based pointwise and subband shrinkage estimator is validated on a synthesized harmonic signal (of length ) in Gaussian noise and the Piece-Regular signal (of length ) in Gaussian, Student’s-, and Laplacian noise. The Piece-Regular signal has both smooth and rapidly-varying regions, making it a suitable candidate for the assessment of denoising performance.
III-A Performance of Pointwise-Shrinkage Estimator
III-A1 Harmonic signal denoising
Consider the signal
[TABLE]
in additive white Gaussian noise, with zero mean and variance . Since the denoising is carried out in the DCT [39] domain, the Gaussian noise statistics remain unaltered. For the purpose of illustration, we assume that is known. In practice, may not be known a priori and could be replaced by the robust median estimate [37] or the trimmed estimate [38]. The clean signal is estimated using inverse DCT after applying the optimum shrinkage. The denoising performance of the MPE and SURE-based approaches is compared in Table I. In case of the Wiener filter, the power spectrum of the clean signal is estimated using the standard spectral subtraction technique [40, 41]. We observe that MPE-based shrinkage with is superior to SURE and Wiener filter (WF) by – dB. The comparison also shows that the performance of the MPE depends critically on .
III-A2 Piece-Regular signal denoising
We consider noisy copies of the Piece-Regular signal, taken from the Wavelab toolbox [33], under Gaussian, Student’s-, and Laplacian contaminations. The noise variance is assumed to be known. Notably, the Gaussian, GMM, and Student’s- distributions of noise are preserved by an orthonormal transform [36], unlike the Laplacian statistics. Therefore, the MPE estimate for Laplacian noise is computed based on a four-component GMM approximation in the DCT domain. The denoised output signal corresponding to Laplacian noise is shown in Figure 6 for illustration. The MPE estimates are better than SURE estimates. The SNR plots in Figure 7 indicate that the MPE outperforms SURE for the noise statistics under consideration and that the gains are particularly high in the input SNR range of to dB, and tend to reduce beyond dB.
III-A3 Effect of on the denoising performance of MPE
Obtaining a closed-form expression for the that maximizes the output SNR is not straightforward. We determine the optimum empirically by measuring the SNR gain as a function of (cf. Figure 8), for i.i.d. Gaussian noise. We observe that the output SNR exhibits a peak approximately at for the harmonic signal in (13) and at for the Piece-Regular signal. As a rule of thumb, we recommend to choose for pointwise shrinkage estimators.
III-B Performance of Subband MPE Shrinkage
To validate the performance of the MPE-based subband shrinkage estimator (cf. Section II-B), we consider denoising of the Piece-Regular signal in additive Gaussian noise. The clean signal and its noisy measurement are shown in Figure 9(a). Denoising is carried out by grouping adjacent DCT coefficients to form a subband. The denoised signals obtained using SURE and MPE are shown in Figures 9(b) and 9(c), respectively. The subband size is chosen to be and the parameter is set equal to , a value that was determined experimentally and found to be nearly optimal. We observe that the MPE gives 1 dB improvement in SNR than the SURE approach.
Variation of the output SNR is also studied as a function of (cf. Figure 10). We experimented with , , and corresponding to subband sizes , , and , respectively. For both SURE and MPE, as increases, the output SNR also increases and eventually saturates for . For input SNR below dB, MPE gives a comparatively higher SNR than SURE, and the margin diminishes with increase in input SNR or the subband size . The degradation in performance of SURE for low SNRs is due to the large error in estimating the MSE at such SNRs. The SURE-based estimate of MSE becomes increasingly reliable as increases, thereby leading to superior performance.
IV Accumulated Probability of Error: MPE Meets the Expected Distortion
The MPE is parametrized by , which has to be appropriately chosen in order to achieve optimal denoising performance. To suppress the direct dependence on , we consider the accumulated probability of error, namely as the risk to be minimized. For a nonnegative random variable , we know that Therefore, the accumulated probability of error is the expected distortion:
[TABLE]
For Gaussian noise distribution,
[TABLE]
Denoting and , the first integral in (15) is evaluated as
[TABLE]
The second term in (15) can be evaluated by replacing with in (16). Combining both integrals, we obtain the expression for the expected distortion:
[TABLE]
An estimate of the expected distortion is calculated by replacing with an estimate , which could also be , to begin with. In Figure 11(a), we show the variation of the original distortion and its estimate obtained by setting , as functions of , averaged over independent realizations of noise. The actual parameter value is . The figure shows that the minimum of the expected risk is close to that of its estimate.
In principle, one could iteratively minimize the distortion by starting with and successively refining it. Such an approach is given in Algorithm . An illustration of the denoising performance of the iterative algorithm is deferred to Section V.
IV-A Expected risk Using GMM Approximation
For the GMM p.d.f. in (10), the expected distortion evaluates to (cf. Appendix B for the derivation)
[TABLE]
where . The expected risk and its estimate for a multimodal (cf. Figure 3(b)) and Laplacian noise p.d.f.s are shown in Figures 11(b) and 11(c), respectively. We observe that, in both cases, the locations of the minima of the true risk and its estimate are in good agreement.
IV-B * Optimum Shrinkage Versus Posterior SNR*
We next study the behavior of for different input SNRs to compare the denoising capabilities of the MPE and the expected -distortion-based shrinkage estimators. The optimum pointwise shrinkage parameter for Gaussian noise statistics, obtained by minimizing SURE, MPE risk estimate, and the estimated risk, for different values of the a posteriori SNR is plotted in Figure 12(a). To illustrate the effect of , the variation of versus a posteriori SNR for MPE corresponding to Gaussian noise is shown in Figure 12(b), for different . We observe that the shrinkage profiles are characteristic of a reasonable denoising algorithm, as Figures 12(a) and 12(b) exhibit that the shrinkage parameters increase as the a posteriori SNR increases. Whereas in case of the MPE, the choice of is crucial, the expected distortion does not require tuning such a parameter. Moreover, the MPE attenuation profile for larger values of is reminiscent of a hard-thresholding function, whereas the expected distortion has an attenuation profile that resembles a soft-threshold.
V Performance of the Expected Distortion-Based Pointwise Shrinkage Estimator
In a practical denoising application, we have only one noisy realization from which the clean signal has to be estimated. However, it is instructive to consider the case of multiple realizations as it throws some light on the performance comparisons vis-à-vis other estimators such as the ML estimator. Consider the observation model in , , where one has access to noisy copies of the signal , and the noise vectors are drawn independently from the distribution. The ML estimator of the signal coefficient is given by , where is the component of . Dropping the subscript , as each coefficient is treated independently of the others, the shrinkage estimator takes the form . To study the behavior of the estimate with respect to , we consider two variants: (i) where is obtained by minimizing , referred to as the oracle-; and (ii) where is chosen to minimize , referred to as ML-. The output SNR as a function of for the Piece-Regular signal, corresponding to an input SNR of dB, is shown in Figure 13(a). For all three estimators, namely, oracle-, ML-, and the ML estimate, the output SNR increases with . However, for the oracle- and the ML- estimators, the output SNR stagnates as increases beyond . For , the oracle- and the ML- shrinkage estimators exhibit better performance compared with the ML estimator. As one would expect, the performance of the ML- estimator matches with that obtained using the oracle- as becomes large, because the ML estimate converges in probability to the true parameter. For , which is often the case in practice, the ML- estimate significantly dominates the ML estimator as seen in Figure 13(a). The SNR gain over the ML estimator could be further improved by using the iterative minimization algorithm introduced in Section IV (cf. Algorithm 1). The performance of the ML- and the ML estimators, for different values of and input SNR is shown in Figure 13(b). The figures show that for small values of SNR and , the ML- estimate outperforms the ML estimator. This is of significant importance in a practical setting where we have only one noisy realization ().
V-A Iterative Minimization of the Expected -Risk
When , the ML- estimator is obtained by minimizing , where is the noisy version of . We refer to this estimate as the non-iterative -based shrinkage estimator. Following Algorithm 1, one could iteratively refine the estimate, starting from . We compare the non-iterative -based estimator with its iterative counterpart, and present the results in Figures 14, 15, and 16, corresponding to Gaussian, multimodal (c.f. Figure 3(b)), and a GMM approximation to the Laplacian noise, respectively. The output SNR obtained using the oracle- estimator, calculated by minimizing , is also shown for benchmarking the performance.
We make the following observations from the Figures 14, 15, and 16: (i) the output SNR increases with iterations, albeit marginally after about 10 iterations; (ii) the iterative method consistently dominates the non-iterative one, with an overall SNR improvement of about to dB, for input SNR in the range dB to dB; and (iii) the SNR gain of the iterative technique also reduces for higher input SNR, similar to other denoising algorithms.
VI Performance Assessment of MPE and -Risk Minimization Algorithms Versus State-Of-The-Art Denoising Algorithms
We compare the MPE and the -based shrinkage estimators with three state-of-the-art denoising algorithms: (i) wavelet soft-thresholding111A Matlab implementation is included in the Wavelab toolbox available at:
http://statweb.stanford.edu/~wavelab/. [34]; (ii) the SURE-LET denoising algorithm222A MATLAB implementation of the SURE-LET algorithm is available at:
http://bigwww.epfl.ch/demo/suredenoising. [5]; and (iii) smooth sigmoid shrinkage (SS) [13] in the wavelet domain 333Pastor et al. kindly provided the MATLAB implementation of their denoising technique [13], which facilitated the comparisons reported in this paper.. In [34], a wavelet-based soft-thresholding scheme is used for denoising, with the threshold selected as for an length signal. The SURE-LET technique employs a linear expansion of thresholds (LET), which is a linear combination of elementary denoising functions and optimizes for the coefficients by minimizing the SURE criterion. In [13], a smooth sigmoid shrinkage is applied on the wavelet coefficients to achieve denoising, and the parameters of the sigmoid, which control the degree of attenuation, are obtained by minimizing the SURE objective. We consider ECG signals taken from the PhysioBank database, and the HeaviSine and Piece-Regular signals taken from Wavelab toolbox for performance evaluation.
The noise is assumed to follow a Gaussian distribution and the output SNR values are averaged over independent realizations. The noise variance is estimated using a median-based estimator [37], which is also used by Luisier et al.1 and Donoho2. In SURE-LET, SS, and wavelet thresholding techniques, denoising is performed using Symmlet-4, with three levels of decomposition, as these settings were found to be the best for the ECG signal (following [22]). In case of MPE and -based shrinkage estimators, denoising is performed in the DCT domain. We use the shorthand notations MPE and MPE-subband to denote the pointwise and subband shrinkage estimators, respectively. The corresponding SURE-based subband shrinkage estimator is denoted as SURE-subband. We set and for computing the subband shrinkage parameters. These parameters have not been specifically optimized; however, they were found to work well in practice. The output SNR as a function of the input SNR, obtained using various algorithms, is shown in Figure 17.
From the ECG signal denoising performance shown in Figure 17(a), we observe that the MPE estimate consistently dominates the soft-thresholding-based denoising for input SNRs ranging from dB to dB. The iterative -distortion-based shrinkage estimator (20 iterations) yields lower output SNR compared with the MPE-based estimate for input SNR values in the range to dB, but surpasses it for relatively higher values of input SNR ( to dB). The SURE-LET and the SS algorithms dominate both MPE and the -based shrinkage estimators, because they use more sophisticated denoising functions in the transform domain, thereby offering greater flexibility. For input SNR range of [math] dB to dB, the expected -distortion-based shrinkage estimator consistently outperforms the soft-thresholding-based techniques.
We have also found that it is possible to boost the denoising performance of an algorithm in the low-SNR regime by adding the MPE denoiser in tandem, that is, by replacing in the expression for the MPE risk estimate in (4) with the estimate obtained using a denoising technique, for example, the SURE-LET. We refer to this tandem approach as MPE-SURE-LET in Figure 17. This approach results in to dB gain in output SNR over SURE-LET for low and medium values of input SNR. We observe in Figure 17 that the MPE-subband estimator outperforms the competing algorithms (except for MPE-SURE-LET in Figure 17(a)), at low input SNR.
VII Conclusions
We have proposed a new framework for signal denoising based on a novel criterion, namely the probability of error. Our framework is applicable to scenarios where the noise samples are independent and additively distort the signal. Denoising is performed by transform-domain shrinkage and the optimum shrinkage parameter is obtained by minimizing an estimate of the MPE risk. We have considered both pointwise and subband shrinkage estimators within the MPE paradigm. The performance of the proposed MPE estimators depends on the choice of the error-tolerance parameter . In pointwise shrinkage, to deal with the issue of selecting an appropriate , we have proposed two approaches. In the first one, we experimentally determined an value that results in maximum SNR gain for a particular signal by evaluating the output SNR for different . In the second approach, we computed the accumulated probability of error, which is the expected distortion, and developed an iterative algorithm for minimization. We demonstrated that the shrinkage estimator obtained using the expected risk outperforms the classical ML estimator, when the number of observations is small or the input SNR is low. We also showed that the shrinkage estimator obtained by iteratively minimizing the risk dominates the non-iterative approach in terms of the output SNR.
Extensive performance comparison of the proposed MPE and the distortion-based approaches with state-of-the-art denoising algorithms is carried out on real ECG signals and Wavelab signals. Experimental results demonstrate that the shrinkage estimator based on the MPE-risk estimate outperforms the SURE-based estimator in terms of SNR gain, particularly in the regime of low SNR and smaller subband size. The proposed MPE-framework could be used as an add-on over an existing denoising technique, leading to an estimator that has a higher output SNR, particularly in the low input SNR regime.
For deriving the expression and validating the performance of the MPE-based subband shrinkage estimator, we considered denoising of signals corrupted with Gaussian noise. Experimentally, we have found that increase in the subband size leads to an increase in output SNR, and saturates beyond a point. We have also observed that, when the subband size or the input SNR is low, the MPE-based estimate has superior performance compared with the SURE-based estimator.
We demonstrated that the optimum shrinkage parameter obtained by minimizing estimates of the MPE/ distortions increases monotonically with the increase in a posteriori SNR. Such behavior of the shrinkage parameter is essential for denoising. A theoretical characterization of this behavior is needed and may lead to interesting inferences, which could potentially lead to a rigorous convergence proof for the iterative expected distortion minimization technique. Another important observation is that, for lower input SNRs, the proposed denoising framework yields a higher output SNR compared with the MSE-based techniques. The improvement in performance in terms of SNR of the denoised output may be attributed to the fact that the MPE framework incorporates knowledge of the distribution of the observations, which goes beyond the second-order statistics considered in de facto MSE-based optimization. We also believe that this is the first attempt at demonstrating competitive denoising performance with probability of error chosen as the distortion metric, in a non-Bayesian estimation framework.
Appendix A Perturbation of SURE-Based Pointwise Shrinkage
To analyze the perturbation in the location of the minimum of the SURE cost function, in comparison with the true MSE, one needs to evaluate
[TABLE]
where and . Let
[TABLE]
The Taylor-series expansion of about yields
[TABLE]
where is the derivative . Using the first-order Taylor series approximation , we obtain
[TABLE]
which, in turn, leads to an approximation of the perturbation probability :
[TABLE]
Invoking , and using the Chernoff bound[43], we obtain
[TABLE]
Consequently, to satisfy an upper bound on the deviation probability of the form , for a given , one must ensure that
[TABLE]
The condition in (19) translates to an equivalent condition on the minimum required SNR to achieve a certain .
Appendix B Expected Risk for GMM
For additive noise with the p.d.f. given in (10), we have
[TABLE]
using (11) and (14). Letting and , we get
[TABLE]
Subsequently, substituting (21) in (20) yields
[TABLE]
which is the expression for the expected distortion for noise following a GMM distribution.
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1[1] H. V. Poor, An Introduction to Signal Detection and Estimation , Springer, 1994.
- 2[2] S. Kay and Y. C. Eldar, “Rethinking biased estimation,” IEEE Signal Process. Mag. , vol. 25, no. 3, pp. 133–136, May 2008.
- 3[3] C. M. Stein, “Estimation of the mean of a multivariate normal distribution,” Ann. Stat. , vol. 9, no. 6, pp. 1135–1151, Nov. 1981.
- 4[4] W. James and C. Stein, “Estimation with quadratic loss,” Berkeley Symp. Math. Statist. and Prob. , Berkeley, CA, vol.1, pp. 361–379, 1961.
- 5[5] F. Luisier, T. Blu, and M. Unser, “A new SURE approach to image denoising: Interscale orthonormal wavelet thresholding,” IEEE Trans. Image Process. , vol. 16, no. 3, pp. 593–606, Mar. 2007.
- 6[6] F. Luisier, T. Blu, and M. Unser, “SURE-LET for orthonormal wavelet-domain video denoising,” IEEE Trans. Circ. Syst. Video Tech. , vol. 20, no. 6, pp. 913–919, Jun. 2010.
- 7[7] F. Luisier and T. Blu, “The SURE-LET multichannel image denoising: Interscale orthonormal wavelet thresholding,” IEEE Trans. Image Process. , vol. 17, no. 4, pp. 482 − - 492, Apr. 2008.
- 8[8] T. Blu and F. Luisier, “The SURE-LET approach to image denoising,” IEEE Trans. Image Process. , vol. 16, no. 11, pp. 2778–2786, Nov. 2007.
