Total variation denoising

From HandWiki
Short description: Noise removal process during image processing
Example of application of the Rudin et al.[1] total variation denoising technique to an image corrupted by Gaussian noise. This example created using demo_tv.m by Guy Gilboa, see external links.

In signal processing, particularly image processing, total variation denoising, also known as total variation regularization or total variation filtering, is a noise removal process (filter). It is based on the principle that signals with excessive and possibly spurious detail have high total variation, that is, the integral of the image gradient magnitude is high. According to this principle, reducing the total variation of the signal—subject to it being a close match to the original signal—removes unwanted detail whilst preserving important details such as edges. The concept was pioneered by L. I. Rudin, S. Osher, and E. Fatemi in 1992 and so is today known as the ROF model.[1]

This noise removal technique has advantages over simple techniques such as linear smoothing or median filtering which reduce noise but at the same time smooth away edges to a greater or lesser degree. By contrast, total variation denoising is a remarkably effective edge-preserving filter, i.e., simultaneously preserving edges whilst smoothing away noise in flat regions, even at low signal-to-noise ratios.[2]

1D signal series

Application of 1D total-variation denoising to a signal obtained from a single-molecule experiment.[3] Gray is the original signal, black is the denoised signal.

For a digital signal [math]\displaystyle{ y_n }[/math], we can, for example, define the total variation as

[math]\displaystyle{ V(y) = \sum_n |y_{n+1} - y_n|. }[/math]

Given an input signal [math]\displaystyle{ x_n }[/math], the goal of total variation denoising is to find an approximation, call it [math]\displaystyle{ y_n }[/math], that has smaller total variation than [math]\displaystyle{ x_n }[/math] but is "close" to [math]\displaystyle{ x_n }[/math]. One measure of closeness is the sum of square errors:

[math]\displaystyle{ \operatorname E(x, y) = \frac{1}{n} \sum_n (x_n - y_n)^2. }[/math]

So the total-variation denoising problem amounts to minimizing the following discrete functional over the signal [math]\displaystyle{ y_n }[/math]:

[math]\displaystyle{ \operatorname E(x,y) + \lambda V(y). }[/math]

By differentiating this functional with respect to [math]\displaystyle{ y_n }[/math], we can derive a corresponding Euler–Lagrange equation, that can be numerically integrated with the original signal [math]\displaystyle{ x_n }[/math] as initial condition. This was the original approach.[1] Alternatively, since this is a convex functional, techniques from convex optimization can be used to minimize it and find the solution [math]\displaystyle{ y_n }[/math].[3]

Regularization properties

The regularization parameter [math]\displaystyle{ \lambda }[/math] plays a critical role in the denoising process. When [math]\displaystyle{ \lambda=0 }[/math], there is no smoothing and the result is the same as minimizing the sum of squares. As [math]\displaystyle{ \lambda \to \infty }[/math], however, the total variation term plays an increasingly strong role, which forces the result to have smaller total variation, at the expense of being less like the input (noisy) signal. Thus, the choice of regularization parameter is critical to achieving just the right amount of noise removal.

2D signal images

We now consider 2D signals y, such as images. The total-variation norm proposed by the 1992 article is

[math]\displaystyle{ V(y) = \sum_{i,j} \sqrt{|y_{i+1,j} - y_{i,j}|^2 + |y_{i,j+1} - y_{i,j}|^2} }[/math]

and is isotropic and not differentiable. A variation that is sometimes used, since it may sometimes be easier to minimize, is an anisotropic version

[math]\displaystyle{ V_\operatorname{aniso}(y) = \sum_{i,j} \sqrt{|y_{i+1,j} - y_{i,j}|^2} + \sqrt{|y_{i,j+1} - y_{i,j}|^2} = \sum_{i,j} |y_{i+1,j} - y_{i,j}| + |y_{i,j+1} - y_{i,j}|. }[/math]

The standard total-variation denoising problem is still of the form

[math]\displaystyle{ \min_y [ \operatorname E(x, y) + \lambda V(y)], }[/math]

where E is the 2D L2 norm. In contrast to the 1D case, solving this denoising is non-trivial. A recent algorithm that solves this is known as the primal dual method.[4]

Due in part to much research in compressed sensing in the mid-2000s, there are many algorithms, such as the split-Bregman method, that solve variants of this problem.

Rudin–Osher–Fatemi PDE

Suppose that we are given a noisy image [math]\displaystyle{ f }[/math] and wish to compute a denoised image [math]\displaystyle{ u }[/math] over a 2D space. ROF showed that the minimization problem we are looking to solve is:[5]

[math]\displaystyle{ \min_{u\in\operatorname{BV}(\Omega)} \; \|u\|_{\operatorname{TV}(\Omega)} + {\lambda \over 2} \int_\Omega(f-u)^2 \, dx }[/math]

where [math]\displaystyle{ \operatorname{BV}(\Omega) }[/math] is the set of functions with bounded variation over the domain [math]\displaystyle{ \Omega }[/math], [math]\displaystyle{ \operatorname{TV}(\Omega) }[/math] is the total variation over the domain, and [math]\displaystyle{ \lambda }[/math] is a penalty term. When [math]\displaystyle{ u }[/math] is smooth, the total variation is equivalent to the integral of the gradient magnitude:

[math]\displaystyle{ \|u\|_{\operatorname{TV}(\Omega)} = \int_\Omega\|\nabla u\| \, dx }[/math]

where [math]\displaystyle{ \|\cdot\| }[/math] is the Euclidean norm. Then the objective function of the minimization problem becomes:[math]\displaystyle{ \min_{u\in\operatorname{BV}(\Omega)} \; \int_\Omega\left[\|\nabla u\| + {\lambda \over 2}(f-u)^2 \right] \, dx }[/math]From this functional, the Euler-Lagrange equation for minimization – assuming no time-dependence – gives us the nonlinear elliptic partial differential equation:

[math]\displaystyle{ \begin{cases} \nabla\cdot\left({\nabla u\over{\|\nabla u\|}} \right ) + \lambda(f-u) = 0, \quad &u\in\Omega \\ {\partial u\over{\partial n}} = 0, \quad &u\in\partial\Omega \end{cases} }[/math]

For some numerical algorithms, it is preferable to instead solve the time-dependent version of the ROF equation:[math]\displaystyle{ {\partial u\over{\partial t}} = \nabla\cdot\left({\nabla u\over{\|\nabla u\|}} \right ) + \lambda(f-u) }[/math]

Applications

The Rudin–Osher–Fatemi model was a pivotal component in producing the first image of a black hole.[6]

See also

References

  1. 1.0 1.1 1.2 Rudin, L. I.; Osher, S.; Fatemi, E. (1992). "Nonlinear total variation based noise removal algorithms". Physica D 60 (1–4): 259–268. doi:10.1016/0167-2789(92)90242-f. Bibcode1992PhyD...60..259R. 
  2. Strong, D.; Chan, T. (2003). "Edge-preserving and scale-dependent properties of total variation regularization". Inverse Problems 19 (6): S165–S187. doi:10.1088/0266-5611/19/6/059. Bibcode2003InvPr..19S.165S. 
  3. 3.0 3.1 Little, M. A.; Jones, Nick S. (2010). "Sparse Bayesian Step-Filtering for High-Throughput Analysis of Molecular Machine Dynamics". 2010 IEEE International Conference on Acoustics, Speech and Signal Processing. http://www.maxlittle.net/publications/little_icassp2010.pdf. 
  4. Chambolle, A. (2004). "An algorithm for total variation minimization and applications". Journal of Mathematical Imaging and Vision 20: 89–97. doi:10.1023/B:JMIV.0000011325.36760.1e. 
  5. Getreuer, Pascal (2012). "Rudin–Osher–Fatemi Total Variation Denoising using Split Bregman". https://www.ipol.im/pub/art/2012/g-tvd/article_lr.pdf. 
  6. "Rudin–Osher–Fatemi Model Captures Infinity and Beyond" (in en). 2019-04-15. https://www.ipam.ucla.edu/news/rudin-osher-fatemi-model-captures-infinity-and-beyond/. 

External links