markov Random Fields for Super-Resolution

From statwiki
Jump to: navigation, search

A Summary on
Markov Networks for Super-Resolution
by
W. T. Freeman and E. C. Pasztor

Introduction

There are some applications in computer vision in which the task is to infer the unobserved image called “scene” from the observed “image”. Typically, estimating the entire “scene” image at once is too complex. For example an ordinary 256*256 image has 65536 pixels and even a very simple linear model requires a 65536*65536 matrix to model operator and thus numerically is intractable. A common approach is to process the image regions locally and then generalize the interpretations across space. The interpretation of images can be done by modeling the relationship between local regions of “images” and “scenes”, and between neighboring local “scene” regions. The former allows us to estimate initial guess for “scene”, and the latter propagates the estimation. These problems are so-called low-level vision problems. Freeman first introduced a probabilistic approach <ref name="R1"> W. T. Freeman, E. C. Pasztor, and O. T. Carmichael, "Learning Low-Level Vision", International Journal of Computer Vision, 40(1), pp. 25-47, 2000.</ref> <ref name="R2"> W. Freeman and E. Pasztor, "Markov Networks for Super-Resolution", in Proc. of 34th Annual Conference on Information Sciences and Systems, Princeton University, March 2000.</ref> in which they try to exploit training method using “image”/ “scene” pairs and apply the Bayesian inference of graphical models. The method is called VISTA, Vision by Image/Scene TrAining. The authors have shown the advantages of the proposed model in different practical applications. Here we focus on super-resolution application where the problem is to estimate high resolution details from low resolution images.

Markov Networks for low-level vision problems

A common graphical model for low-level vision problems is Markov networks. For a given "image", [math]y[/math], the underlying "scene" [math]x[/math] should be estimated. The posterior probability, [math] P(x|y)= cP(x,y)[/math] is calculated considering the fact that the parameter [math] c = 1/P(y) [/math] is a constant over [math]x[/math], i.e. the evidence is independent of the scene, and treated as a normalization factor. The best scene estimate depends on the price for making the wrong guess. The best scene estimate [math]\hat{x}[/math] is the minimum mean squared error, MMSE, or the maximum a posterior, MAP. Without any approximation the [math]\hat{x}[/math] is difficult to compute, and hence Markovian assumption is established for the model. Therefore the "image" and "scene" are divided into patches and one node of the Markov network is assigned to each patch. Figure
Fig.1 Markov network for vision problems. Each node in the network describes a local patch of image or scene. Observations, y, have underlying scene explanations, x. Lines in the graph indicate statistical dependencies between nodes.
depicts the undirected graphical model for mentioned problem where the nodes connected by lines indicate statistical dependencies. Each “scene” node is connected to its corresponding “image” node as well as its neighbors.

To make use of Markov networks the unknown parameters should be learned from training data in learning phase, and then in inference phase the “scene” estimation can be made. For a Markov random field, the joint probability over the “scene” [math]x[/math] and the “image” [math] y[/math] is given by:

[math] P(x_1,x_2,...,x_N,y_1,y_2,...,y_N) = \prod_{(i,j)} \psi(x_i,x_j) \prod_{k} \phi(x_k,y_k)\,\,\, (1) [/math]

where [math]\psi[/math] and [math]\phi[/math] are potential functions and they are leaned from training data. In this paper the authors prefer to call these functions compatibility functions. Then one can write the MAP and the MMSE estimates for [math]\hat{x}_j[/math] by marginalizing or taking the maximum over all other variables in the posterior probability, respectively. For discrete variables the expression is:

[math] \hat{x}_{jMMSE} = \sum_{x_j} \sum_{\text{all}\, x_i, i != j} P(x_1,x_2,...,x_N,y_1,y_2,...,y_N)\,\,\, (2) [/math]
[math] \hat{x}_{jMAP} = \arg\max_{x_j} (\max_{\text{all}\, x_i != j} P(x_1,x_2,...,x_N,y_1,y_2,...,y_N))\,\,\, (3) [/math]

For large networks the computation of Eq(2) and Eq(3) are infeasible to evaluate directly; however, the task is easier for network which are trees or chains. This is the motivations of using graphical models to burden the computational complexity.

Inference in Networks without loops

For networks with no loop the inference is the simple “message-passing” rule which enables us to compute MAP and MMSE estimate <ref name="R3"> Jordan, M.I. (Ed.). 1998. Learning in Graphical Models. MIT Press: Cambridge, MA </ref>. For example for the network in Figure 2
Fig.2 Example Markov network without any loop, used for belief propagation example described in text.
the MAP estimation for node [math]j[/math] is determined by:
[math]\begin{matrix} \hat{x}_{1MAP} & = & \arg\max_{x_1} ( \max_{x_2} \max_{x_3} P(x_1,x_2,x_3,y_1,y_2,y_3)\\ & = & \arg\max_{x_1} ( \max_{x_2} \max_{x_3} \phi(x_1,y_1) \phi(x_2,y_2) \phi(x_3,y_3) \psi(x_1,x_2) \psi(x2,x3)\\ & = & \arg\max_{x_1} \phi(x_1,y_1) (\max_{x_2} \psi(x_1,x_2) \phi(x_2,y2)) (\max_{x_3} \psi(x_2, x_3) \phi(x_3,y_3)\,\,\, (4) \end{matrix}[/math]

The similar expressions for [math]x_{2MAP}[/math] and [math]x_{3MAP}[/math] can be used. Equations (3) and (2) can be computed by iterating the following steps. The MAP estimate at node j is

[math] \hat{x}_{jMAP} = \arg\max_{x_j} \phi(x_j, y_j) \prod_{k} M^k_j \,\,\, (5) [/math]

Where k runs over all “scene” node neighbors of node j, and [math] M^k_j [/math] is the message from node k to node j. The [math]M^k_j[/math] message is calculated by:

[math] M^k_j = \max_{x_k} \psi(x_j,x_k) \phi(x_k,y_k) \prod_{i!=j} \hat{M}^l_k \,\,\, (6) [/math]

where [math]\hat{M}^l_k[/math] is [math]M^k_l[/math] from the previous iteration. The initial [math]\hat{M}^k_j[/math]'s are set to column vector of 1’s, with the same dimension as [math]x_j[/math]. Because the initial messages are 1’s at the first iteration, all the message in the network are:

[math] M^2_1 = \max_{x_2} \psi(x_1, x_2) \phi(x_2,y_2)\,\,\,(7) [/math]
[math] M^3_2 = \max_{x_3} \psi(x_2, x_3) \phi(x_3,y_3)\,\,\,(8) [/math]
[math] M^1_2 = \max_{x_1} \psi(x_2, x_1) \phi(x_1,y_1) \,\,\,(9) [/math]
[math] M^2_3 = \max_{x_2} \psi(x_3, x_2) \phi(x_2,y_2) \,\,\,(10) [/math]

The second iteration uses the messages above as the [math]\hat{M}[/math] variables in Eq(6) :

[math] M^2_1 = \max_{x_2} \psi(x_1, x_2) \phi(x_2,y_2)\hat{M}^3_2 \,\,\,(11) [/math]
[math] M^3_2 = \max_{x_3} \psi(x_2, x_3) \phi(x_3,y_3) \,\,\,(12) [/math]
[math] M^2_3 = \max_{x_2} \psi(x_3, x_2) \phi(x_2,y_2)\hat{M}^1_2 \,\,\,(13) [/math]
[math] M^1_2 = \max_{x_1} \psi(x_2, x_1) \phi(x_1,y_1) \,\,\,(14) [/math]

And thus

[math] M^2_1 = \max_{x_2} \psi(x_1, x_2) \phi(x_2,y_2) * \max_{x_3} \psi(x_2,x_3) \phi(x_3,y_3) [/math] (15)

Eventually the MAP estimates for [math]x_1[/math] becomes:

[math] \hat{x}_{1MAP} = \arg\max_{x_1} \phi(x_1,y_1)M^2_1 (16) [/math]

The MMSE estimate Eq(3) has analogous formulae with the [math]\max_{x_k}[/math] of Eq(8) replaced by [math]\sum_{x_k}[/math] and [math]\arg\max_{x_j}[/math] replaced by [math]\sum_{x_j} x_j[/math].

Networks with loops

It is known that finding the posterior distribution for a Markov network with loops is an expensive is an expensive computational task. Several approximation technique have been proposed by different researchers to address this problem. In <ref name="R4"> Weiss, Y. and Freeman, W.T. 1999. Correctness of belief propagation

in Gaussian graphical models of arbitrary topology. Technical Report UCB.CSD-99-1046, Berkeley. </ref>, Weiss and Freeman demonstrate that the above derived belief propagation rules works even in networks with loops. Summary of results from Weiss and Freeman regarding belief propagation results after convergence is provided in Fig.3
Fig.3 Summary of results from Weiss and Freeman (1999) regarding belief propagation results after convergence.

The table shows that MMSE propagation scheme is grunted to converge to the true posterior when used for Gaussian process. For non Gaussian processes, the MAP propagation scheme is grunted to converge to at least a local maximum of the true posterior.

General view of the paper

Basically this paper aims to develop a new super-resolution scheme utilizing Markov random fields by which given low resolution image is resized to required high resolution size reproducing high frequency details. Typical interpolation techniques such as bilinear, nearest neighbor and bicubic expand the size of the low-resolution image, yet the result suffers from blurriness and in some cases blocking artifact. This means that the sharpness of the edges are decreased in the reconstructed image.

According to Freeman, in this method they collect many pairs of high resolution images and their corresponding low-resolution images as the training set. Given low-resolution image by a typical bicubic interpolation algorithm is employed to create a high resolution image which is interpreted as the “image” in Markov network. The hypothesis to reduce the computational burden is: since not all of the low frequencies of the blurred image are useful to predict high frequencies, so not all possible values are taken into consideration. Secondly, both low and high frequencies are scaled multiplicatively for image intensity. Of course this image does not look suitable, and thus they try to estimate original high resolution image which in we may call it “scene” image based on the definitions provided above. In order to do that, the images in training set and also the low-resolution image is divided into patches so that each patch represents the Markov network node (figure1). Therefore, [math]y_i[/math]’s in figure are observed, and thus should be shaded. Subsequently, for each patch in y 10 or 20 nearest patches from the training database are selected using Euclidean distance. The ultimate job is to find the best patch in the candidate set for each patch in y using MAP estimate. In other words, the estimated scene at each patch is always some example from the training set.

Implementation of low-level vision problems using Markov network

The “image” and “scene” are arrays of pixel values, and so the complete representation is cumbersome. In this research, the principle component analysis (PCA) is applied for each patch to find a set of lower dimensional basis function. Moreover, potential functions [math]\psi[/math] and [math]\phi[/math] should be determined. A nice idea is to define Gaussian mixtures over joint spaces [math]x_i*x_j [/math] and [math] x_j*x_k [/math];however, it is very difficult. The authors prefer a discrete representation where the most straightforward approach is to evenly sample all possible states of each image and scene variable at each patch.

For each patch in "image" y a set of 10 or 20 "scene" candidates from the training set are chosen. Figure4
Fig.4 The "image" patch and "scene" are divided into patches. For each "image" patch a collection of candidate scene patches from the training database is chosen. The final task is to find the best patch using inference on Markov networks.
illustrates an example of each patch in y and the associated "scene" candidates.

Learning the Potential (Compatibility) Functions

The potential functions are defined arbitrary, but they have to be introduced wisely. In this paper, a simple way is used to find potential functions. They assume “scene” patches have overlap shown is Figure 5
Fig.5 The compatibility between candidate scene explanations at neighboring nodes is determined by their values in their region of overlap.

Therefore, the scene patches themselves may be used to define potential functions [math]\psi[/math] between nodes [math]x_i[/math]’s. Recall for node [math]x_i[/math] and its neighbor [math]x_j[/math] there are two sets of candidate patches. Let assume the lth candidate in node j and the mth candidate in node k have some overlap. Also, we can think that the pixels in the overlapped region in [math]x_j[/math] ([math]d^l_{kj}[/math]) and their correspondent in [math]x_k[/math] ([math]d^m_{jk}[/math]) are some variation of each other , and thus eventually the potential function between node [math]x_j[/math] and node [math]x_k[/math] are given by:

[math] \psi(x^l_k,x^m_j) = \exp \frac{-|d^l_{jk}-d^m_{kj}|}{2\sigma^2_s} [/math]

Where [math]\sigma_s[/math] has to be determined. The authors assume that the image and scene training samples differ from the "ideal" or original high resolution image by Gaussian noise with covariance <math\sigma_i</math> and [math]\sigma_s[/math], respectively. Therefore, [math]\psi[/math] function between node j and k can be represented by a matrix whose lth row and mth column is [math]\psi(x^l_k, x^m_j)[/math]. Potential function [math]\phi[/math] which is defined between "scene" node x and "image" node y is determined based on another intuitive assumption. We say that a "scene" candidate [math]x^k_l[/math] is compatible with an observed image patch [math]y_0[/math] if the image patch, [math]y^k_l[/math], associated with the scene candidate [math]x^k_l[/math] in the training database matches [math]y_0[/math]. Of course, it will not exactly match, but we may suppose that the training data is “noisy” version of original image resulting in:

[math] \phi(x^l_k, y_k) = \exp \frac{-|y^l_k - y_0|^2}{2\sigma^2_s} [/math]

Super-Resolution

For the super-resolution problem, the input is a low-resolution image, and the “scene” to be estimated is its high resolution version. At the first glance, the task may seem impossible since the high resolution data is missing. However, the human eye is able to identify edges and sharp details in low resolution image and we know these structural information should remain at higher resolution level. The authors attempt to solve this problem using aforementioned Markov model and they name the method VISTA. There are some preprocessing steps in order to increase the efficiency of the training set. First, consider three scales Laplacian pyramid decomposition. The first sub-band, H, represents the detail in high frequency while the second and the third sub-bands indicate the middle, M, and the low, L, frequency components. The assumption is that high frequency band, H, is conditionally independent of the lower frequency bands, given the middle frequency band, M, yielding:

[math] P(H|M,L) = P(H|M) [/math]

Hence, to predict the high frequency components, we will need the middle frequency details, M, not the low frequency band, L. This hypothesis greatly reduces the computation costs. Second, The researchers in this paper assume that statistical relationships between image bands are independent of image contrast. Furthermore, they take the absolute value of the mid-frequency band, and then pass it through a low-pass filter resulting in a normalized mid frequency band. They also do the same procedure for high-frequency band. Figure 5 (a) show the low-resolution image which then is expanded to have the same size as the desired high resolution image using a typical interpolation algorithms such as bicubic method (Figure 5 (b)). The image in (c) in the original high resolution image. Images in Figure 5 (d) and (e) are the first level of the Laplacian pyramid decomposition for "image" and “scene” images, respectively. In other words, the high frequency component in Figure 5 (e) should be estimated using frequency component in Figure 5(d).


Figure5
Fig.6 The "image" patch and "scene" are divided into patches. For each "image" patch a collection of candidate scene patches from the training database is chosen. The final task is to find the best patch using inference on Markov networks.

In order to utilize Markov network for this problem the "image" and "scene" images are divided into local patches as shown in Figure 6 and the final estimate for "scene" image , x,, is the collection which maximizes probability of [math] P(x|y) [/math] using Equation 1.

Figure6
Fig.6 .

Figure 7 illustrates an example of a given patch in y and its corresponding "scene" patches. The patch size is a difficult task since choosing small size gives very little information for estimating the underlying “scene” patches. On the other hand, large patches would make the learning process of [math]\phi[/math]’s very complex. The authors in some papers use 7*7 patch size for low frequency band and 3*3 for high frequency components, but in ref they use 7*7 for the "image" and 5*5 for the "scene".

Figure7
Fig.6 .

Figure 8 compares the results of the proposed approach in this paper and different super-resolution schemes. The interesting thing is the effect of training set on the final result. Because the estimate "scene" patch is always chosen from the training database the final result resembles the training set in some manner. This makes the selection of training images set very important and crucial. The set must contain wide range of images to include different features.

Figure8

One of the most recent works in the field proposes wavelet-domain hmm (whmm) for the super resolution signal reconstruction. <ref>Han, Y., Chen, R., Wu, L., Signal super-resolution reconstruction based on wavelet-domain HMM, Shuju Caiji Yu Chuli/Journal of Data Acquisition and Processing 26 (3), pp. 292-299, 2011.</ref>

Packages

A software package has been developed in Matlab to address this super resolution algorithm. <ref name="R5"> W. T. Freeman, C.E. Liu, http://people.csail.mit.edu/billf/project%20pages/sresCode/Markov%20Random%20Fields%20for%20Super-Resolution.html</ref>

References

<references />