- Research
- Open access
- Published:
Dependent Gaussian mixture models for source separation
EURASIP Journal on Advances in Signal Processing volume 2012, Article number: 239 (2012)
Abstract
Source separation is a common task in signal processing and is often analogous to factor analysis. In this study, we look at a factor analysis model for source separation of multi-spectral image data where prior information about the sources and their dependencies is quantified as a multivariate Gaussian mixture model with an unknown number of factors. Variational Bayes techniques for model parameter estimation are used. The development of this methodology is motivated by the need to bring an efficient solution to the separation of components in the microwave radiation maps that are being obtained by the satellite mission Planck which has the objective of uncovering cosmic microwave background radiation. The proposed algorithm successfully incorporates a rich variety of prior information available to us in this problem in contrast to many previous solutions that assume completely blind separation of the sources. Results on realistic simulations of Planck maps and on Wilkinson microwave anisotropy probe fifth year images are shown. The technique suggested is easily applicable to other source separation applications by modifying some of the priors.
Introduction
The discovery of the cosmic microwave background (CMB) is a strong evidence for the Big Bang theory of the formation and development of the universe. According to the theory, the early universe was smaller and hotter but cooled as it expanded. Once the temperature cooled to about 3000 K, photons were free to propagate without being scattered off ionized matter; the CMB is an image of this event and is visible across the entire sky. Three satellites have been launched to measure the CMB: the cosmic background explorer, Wilkinson microwave anisotropy probe (WMAP) and most recently the Planck surveyor. Planck is the highest resolution data to date, of the order of 107 pixels across the sky measured at nine channels.
Unfortunately, the signals measured by these satellites as shown in Figure 1 contain radiation not only from CMB, but also contributions from a number of other sources, namely foreground radiations and extragalactic sources in addition to antenna receiver noise. Foreground sources from our galaxy include synchrotron, dust, and free–free emission. Therefore, the separation of the CMB signal from other sources is an important stage in the production of CMB maps [1].
To date, there have been several attempts to achieve it in a Bayesian framework using both (a) Gaussian mixture model (GMM) prior [3], and (b) Markov Random Field (MRF) prior [4, 5]. Full sky maps at low resolution through MCMC, using masks to reduce the effect of the signal in the galactic plane, were described in [6]. Some of these are fully Bayesian source separation methods which are developed to separate the underlying CMB from the mixed observed signals of extraterrestrial microwaves made at several frequencies.
A common assumption among works in the literature is the independence of the cosmological sources. Although it is well known that CMB is independent from the rest of the sources, the galactic sources demonstrate significant statistical dependence among themselves, as stated in [1]. Recently, a small number of researchers have started addressing this problem [7, 8]. Various dependent component analysis approaches are compared in [9], demonstrating their superior performance with respect to classical ICA.
In this study, we present a dependent components model for source separation of multi-spectral image data, where prior information about the sources and between-source dependencies is quantified as a multivariate GMM, using variational Bayes techniques for model parameter estimation. This article can thus be considered as an extension of [3], modeling dependencies between-sources through generalizing the prior to multivariate GMM.
The rest of the article is structured as follows. The next section gives the model for the mixing problem and describes the hierarchical Bayesian model that we use, including the prior we assume for the sources. Section “Implementing the source separation” describes the variational Bayes approach we use for the implementation of the separation. Section “Examples” provides results on both synthetic Planck and real WMAP images. Finally, we provide a discussion of the results in the last section.
Model
The model description is defined in terms of the microwave source separation problem, where there are n f maps of the sky at frequencies , each map consisting of J pixels. The data are denoted . The source model consists of n s sources and is represented by the vectors , with each component representing the amplitude of a physical source of microwaves. We assume that the d j can be represented as a linear combination of the s j :
where A is an n f × n s “mixing” matrix and e j is a vector of n f independent Gaussian error terms with precisions (inverse variances) . For convenience, define
to represent all data and sources.
We assume dependence between the sources, defined by a prior distribution p(S|ψ) with parameters ψ. The goal is to estimate the S and the parameters ψ associated with the model for S, given observation of D. The noise variances τ and the mixing matrix A are assumed known. GMM are used to represent the non-Gaussian sources, in which case it is an example of a model known as a mixture of factor analyzers [10]. As in [10], we adopt a Bayesian approach to the data fitting, implemented by a variational Bayes approach.
Bayesian inference will be based on the posterior distribution, which following the above description can be factorized as
Each element of this distribution is defined next in turn.
Noise structure
Gaussian error, e j , is assumed independent within and between pixels j and frequency, which gives
where Ai· is the i th row of A.
Mixing matrix structure
In this application, A is parameterized and denoted A(θ). Each column of A(θ) is the contribution to the observation of a source at different frequencies, which is written as a function of the frequencies and θ. These parameterizations are approximations that come from the current state of knowledge about how the sources are generated. Here, we merely state the parameterization that we are going to use, and refer to [11] for a more detailed exposition on the background to them. Some restrictions are usually placed on A(θ)in order to force a unique solution; this is achieved here by setting the first row of A(θ) to be ones.
It is assumed that the CMB is the first source and therefore it corresponds to the first column of A(θ). It is modeled as a black body at a temperature, and its contribution is a known constant at each frequency. The parametrization of the mixing matrix is given as
where
T0 = 2.725K is the average CMB temperature, h is the Planck constant, and kB is Boltzmann’s constant. The ratio g(ν i ) / g(ν1) is designed to ensure that A11(θ) = 1 as we constraint the first row of A(θ) to be ones.
where T1 = 18.1K is the assumed thermodynamical temperature of the dust grains, and column 2 corresponds to synchrotron, column 3 to galactic dust, and column 4 is free–free emission. There are three unknown model parameters for A, for synchrotron , the spectral indices for dust κ d ∈ {κ d : 1 ≤ κ d ≤ 2}, and for free–free emission .
The sources
The distribution of s j is modeled as a GMM with m factors. The model proposed allows for between-source dependence; the vector of sources at a pixel is a mixture of multivariate Gaussians
where
for mixture component weights w a , mean vectors μ a , and precision matrices Q a , so that ψ is all the w a , μ a and Q a , with a = 1,…,m. Note that, in the standard inflationary cosmological model the CMB is a single multivariate Gaussian (m = 1) while the Galactic foregrounds might require m > 1 to be correctly modeled. In order to fulfill this, we set the CMB for components 2,…,m of the multivariate mixture to be exactly zero in the implementation.
Priors
The remaining term in Equation (2) is p(ψ). We use the conjugate prior distributions [12] that facilitate the computation of the posterior and yet flexible enough to incorporate good prior information: Gaussians for the component means, Dirichlet for the component weights, and Wishart for precision matrices. In the microwave source application, background knowledge about the magnitude of the sources can be incorporated through specifying values of the parameters of these prior distributions. This prior specification follows [13], who discuss how to specify these values in more detail.
Implementing the source separation
The posterior developed in the previous section does not lend itself to an analytical solution. MCMC techniques are one approach that let us evaluate complicated integrals by sampling rather than by analytical or numerical methods. The main criticism of Bayesian source separation with sampling methods, MCMC in particular, is their computational load and slow convergence. Regarding the speed, they cannot compete with methods such as FastICA [14, 15].
There are several approaches to speed up the algorithm, such as the strategies suggested in [16]. In the image source separation problem framework, the Langevin sampling scheme has been implemented [4], as a way to obtain a faster MC algorithm.
In this study, the source separation model presented in Section “Model” is implemented by a variational Bayesian approach [10, 17, 18], that allows for more efficient inference when dealing with large data when compared with MCMC techniques. In essence, given the data D and a model with parameters θ and latent variables Z, the variational Bayes method is based on approximating the posterior distribution p(Z,θ|D) with a factorial approximation q(Z,θ|ϕ) = q(Z|ϕ Z )q(θ|ϕ θ ), where ϕ are the variational parameters. The approximation is fitted by minimizing the Kullback–Leibler divergence between q and p, or equivalently maximizing a lower bound on marginal log-likelihood of the data.
Attias [19] has recently developed a fully Bayesian approach to GMM with a variational approximation to the posterior that, when choosing conjugate priors, leads to the following components: Wishart densities for the precisions, Q a ; Normal densities for the means, μ a ; and a Dirichlet for the mixing coefficients, p; and a discrete distribution for the indicator posteriors, z j , which indicates the component that explains information in pixel j. We further derived the variational approximation to the marginal posterior of sources, s j , which turns out to be a multivariate Gaussian distribution. In brief
and
where MVN stands for multivariate normal distribution and Ψ denotes the digamma function. Note that q(z j = a) is the probability that component a is responsible for information in pixel j in sources, s j .
The quantities of interest, i.e., the hyper-parameters to be computed, , , λ, ξ a , β a , η a , and V a for j = 1,…,J and a = 1,…,m, have the following values:
with
Computations were carried out using Matlab.
Examples
Analysis of simulated data
The synthetic Planck data were generated using the Planck Sky Model (PSM) package. The PSM is a set of IDL codes created by the Planck WG2 team in order to provide realistic simulations of the sky at the Planck frequencies [20]. Figure 2 shows data obtained from realistic simulations of CMB, synchrotron, and galactic dust on a 512×512 patch. The original sources are shown in Figure 3. The data were generated at the nine frequencies that are observed by Planck from 30 to 857 GHz. The mixing matrix used was as defined in Section “Mixing matrix structure” with κ s = −2.9 and κ d = 2.0. Noise precisions were those published by the Planck research team [21]. After exploring several values for m, the number of components in the GMM source model was fixed to be m = 1, as it provided the best fit, taking into account the compromise between fit and number of parameters in the model.
Figure 4 shows an estimate of CMB, along with a scatter plot of this estimate against the true value, as shown in Figure 3. Such an estimate is the average of the samples obtained for the first column of A⋆, which corresponds to CMB. We see from the scatter plot and from comparison with Figure 3 that the reconstruction of CMB is very accurate here. The same is true for the other two sources, as shown in Figure 5.
Table 1 shows the mean of the parameters of the model. Regarding the between-sources dependence structure, posterior estimates of are approximately 0, suggesting independence between CMB and the other sources, as expected. On the other hand, posterior estimate of , indicating dependence between synchrotron and galactic dust.
Analysis of a WMAP year 5 patch
The WMAP [22] was launched in 2001 and data collection activities finished in 2010. It observes 5 frequencies from 22 to 90 GHz. Figure 6 shows a patch of 5-year WMAP data.
The algorithm was implemented with four sources (CMB, synchrotron, dust, and free–free emission). The noise precisions were assumed to be the published values for WMAP detectors. The spectral density for free–free emission was fixed at −2.14 (following [11]) and the synchrotron and dust spectral indices were as in the first example. The number of components in the GMM source model were fixed to be m = 2, following the same reasoning as in the simulation study. Informative priors were placed on the GMM parameters, based on discussions on the expected marginal properties of the sources. Table 2 shows the mean of the mixture parameters of the model. Figure 7 shows the estimated CMB. The result obtained is in agreement with previous work [3], as can be appreciated in Figure 8 that shows an histogram of the differences between the estimated CMB using the approach presented here and the estimated CMB obtained in [3].
Finally, in order to show the fit of the data to the model, Figure 9 is a scatter plot of the observed value of the d jk with the standardized residuals, with one figure for each frequency k = 1,…,5.
Conclusion
A fully Bayesian factor analysis algorithm has been presented and applied to a multi-channel image source separation problem, where dependencies between sources are modeled as a multivariate GMM. The algorithm performs very well on simulated Planck data and has been applied to data from WMAP.
In this study, we extend previous approaches [3] by allowing the source priors to be a mixture of multivariate Gaussian distributions for each pixel.
The development of this methodology is motivated by the need to bring an efficient solution to the separation of components in the microwave radiation maps to be obtained by the satellite mission Planck which has the objective of uncovering CMB radiation. The proposed algorithm successfully incorporates a rich variety of prior information available to us in this problem in contrast to most of the previous work that assumes completely blind separation of the sources. Further, the variational approach presented here overcomes the convergence problems of the MCMC stated in [23], when dealing with large datasets such as that will be provided by the satellite mission Planck.
In the analysis of simulated data, the number of components in the GMM source model turned out to be m = 1. This means that sources are multivariate Gaussian a priori. On the other hand, for real data, the number of components is m = 2. In blind source separation problem, identifiability relies on the independence of the sources. In this study, in spite of modeling the sources as Gaussians when m = 1, identifiability is obtained because of the prior information which is incorporated to the model, given structure to the mixing matrix.
Another type of dependence is that a source is spatially correlated. Spatial dependence is most conveniently modeled by a Gaussian MRF and some preliminary work on this idea can be found in [5]. Combining with cross source correlations, one might ultimately consider a mixture of multivariate Gaussian MRF as a prior for the sources. Implementing the analysis with such a prior would be a significant challenge computationally; we hypothesize that it will be difficult to derive a well-behaved MCMC approach. Other functional approximations, such as that of [24], offer feasible alternative to computing the posterior distribution in this case.
Finally, although the technique was developed for the astrophysical source separation problem in mind, it is general and it is applicable to other source separation problems as well.
References
Kuruoglu EE: Bayesian source separation for cosmology. IEEE Signal Process. Mag 2010, 27: 43-54.
[http://lambda.gsfc.nasa.gov/product/map/current/m_products.cfm]
Wilson SP, Kuruoglu EE, Salerno E: Fully Bayesian source separation of astrophysical images modelled by a mixture of Gaussians. IEEE J. Sel. Topics Signal Process 2008, 2(5):685-696.
Kayabol K, Kuruoglu EE, Sanz JL, Sankur B, Salerno E, Herranz D: Adaptive Langevin sampler for separation of t-distribution modelled astrophysical maps. IEEE Trans. Image Process 2010, 19(9):2357-2368.
Kayabol K, Kuruoglu EE, Sankur B: Bayesian separation of images modeled with MRFs using MCMC. IEEE Trans. Image Process 2009, 18(5):982-994.
Dickinson C, Eriksen HK, Banday AJ, Jewell JB, Gorski KM, Huey G, Lawrence CR, O’Dwyer IJ, Wandelt BD: Bayesian component and separation cosmic microwave background estimation for the five-year WMAP temperature data. Astrophys. J 2010, 705: 1607-1623.
Bedini L, Herranz D, Salerno E, Baccigalupi C, Kuruoglu EE, Tonazzini A: Separation of correlated astrophysical sources using multiple-lag data covariance matrices. EURASIP J. Appl. Signal Process 2005, 2005(15):2400-2412. 10.1155/ASP.2005.2400
Bonaldi A, Bedini L, Salerno E, Baccigalupi C, De Zott G: Estimating the spectral indices of correlated astrophysical foregrounds by a second-order statistical approach. Monthly Notices R. Astron. Soc 2006, 373: 271-279. 10.1111/j.1365-2966.2006.11025.x
Kuruoglu EE: Dependent component analysis for cosmology. Lecture Notes Comput. Sci 2010, 6365: 538-545. 10.1007/978-3-642-15995-4_67
Ghahramani Z, Beal M: Variational inference for Bayesian mixtures of factor analysers. In Advances in Neural Information Processing Systems. Edited by: Solla SA, Leen TK, Muller KR. MIT Press, Cambridge, MA); 2000.
Eriksen HK, Dickinson C, Lawrence CR, Baccigalupi C, Banday AJ, Gorski KM, Hansen FK, Lilje PB, Pierpaoli E, Smith KM, Vanderlinde K: C M B component separation by parameter estimation. Astrophys. J 2006, 641: 665-682. 10.1086/500499
Lee PM: Bayesian Statistics: An Introduction. Hodder Arnold H& S, London; 2004.
Richardson S, Green P: On Bayesian analysis of mixtures with an unknown number of components (with discussion). J. R. Stat. Soc. Ser. B 1997, 59: 731-792. 10.1111/1467-9868.00095
Hyvärinen A: Fast and robust fixed-point algorithms for independent component analysis. IEEE Trans. Neural Netw 1999, 10(3):626-634. 10.1109/72.761722
Leach S, Cardoso JF, Baccigalupi C, Barreiro R, Betoule M, Bobin J, Bonaldi A, Delabrouille J, De Zotti G, Dickinson C, Eriksen HK, González-Nuevo J, Hansen FK, Herranz D, Le Jeune M, López-Caniego M, Martínez-González E, Massardi M, Melin JB, Miville-Deschênes MA, Patanchon G, Prunet S, Ricciardi S, Salerno E, Sanz JL, Starck JL, Stivoli F, Stolyarov V, Stompor R, Vielva P: Component separation methods for the PLANCK mission. Astron. Astrophys 2008, 491(2):597-615. 10.1051/0004-6361:200810116
Gilks WR, Richardson S, Spiegelhalter DJ: Markov Chain Monte Carlo in Practice. CRC Press, Boca Raton, FL; 1996.
McKay DJC: Information Theory, Inference and Learning Algorithms. Cambridge University Press, Cambridge; 2003.
Bishop CM: Pattern Recognition and Machine Learning. Springer, New York; 2006.
Attias H: A Variational Bayesian Framework for Graphical Models. MIT Press, Cambridge, MA; 2000.
Collaboration P: The pre-launch plack sky model: a model of sky emission at submillimetre to centimetre wavelengths (in preparation).
[http://sci.esa.int/science-e/www/object/index.cfm?fobjectid=34730%26fbodylongid=1595]
Wilson SP, Kuruoglu EE, Quirós A: Bayesian factor analysis using Gaussian mixture sources, with application to separation of the cosmic microwave background. 2nd International Workshop on Cognitive Information Processing 2010.
Rue H, Martino S, Chopin N: Approximate Bayesian inference for latent Gaussian models using integrated nested Laplace approximations. J. R. Stat. Soc. Ser. B 2008, 71: 319-392.
Acknowledgements
Simon Wilson was supported by the STATICA project, funded by the Principal Investigator program of Science Foundation Ireland, contract number 08/IN.1/I1879. The authors acknowledge the use of the PSM, developed by the Component Separation Working Group (WG2) of the Planck Collaboration.
Author information
Authors and Affiliations
Corresponding author
Additional information
Competing interests
Both authors declare that they have no competing interests.
Authors’ original submitted files for images
Below are the links to the authors’ original submitted files for images.
Rights and permissions
Open Access This article is distributed under the terms of the Creative Commons Attribution 2.0 International License (https://creativecommons.org/licenses/by/2.0), which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
About this article
Cite this article
Quirós, A., Wilson, S.P. Dependent Gaussian mixture models for source separation. EURASIP J. Adv. Signal Process. 2012, 239 (2012). https://doi.org/10.1186/1687-6180-2012-239
Received:
Accepted:
Published:
DOI: https://doi.org/10.1186/1687-6180-2012-239