Variational Mixture Models with Gamma or inverse-Gamma components

Mixture models with Gamma and or inverse-Gamma distributed mixture components are useful for medical image tissue segmentation or as post-hoc models for regression coefficients obtained from linear regression within a Generalised Linear Modeling framework (GLM), used in this case to separate stochastic (Gaussian) noise from some kind of positive or negative "activation" (modeled as Gamma or inverse-Gamma distributed). To date, the most common choice in this context it is Gaussian/Gamma mixture models learned through a maximum likelihood (ML) approach; we recently extended such algorithm for mixture models with inverse-Gamma components. Here, we introduce a fully analytical Variational Bayes (VB) learning framework for both Gamma and/or inverse-Gamma components. We use synthetic and resting state fMRI data to compare the performance of the ML and VB algorithms in terms of area under the curve and computational cost. We observed that the ML Gaussian/Gamma model is very expensive specially when considering high resolution images; furthermore, these solutions are highly variable and they occasionally can overestimate the activations severely. The Bayesian Gauss-Gamma is in general the fastest algorithm but provides too dense solutions. The maximum likelihood Gaussian/inverse-Gamma is also very fast but provides in general very sparse solutions. The variational Gaussian/inverse-Gamma mixture model is the most robust and its cost is acceptable even for high resolution images. Further, the presented methodology represents an essential building block that can be directly used in more complex inference tasks, specially designed to analyse MRI-fMRI data; such models include for example analytical variational mixture models with adaptive spatial regularization or better source models for new spatial blind source separation approaches.

[1]  Mark W. Woolrich,et al.  FSL , 2012, NeuroImage.

[2]  B. Lindsay Mixture models : theory, geometry, and applications , 1995 .

[3]  A. Rosser A.I.D.S. , 1986, Maryland medical journal.

[4]  Stephen M. Smith,et al.  Probabilistic independent component analysis for functional magnetic resonance imaging , 2004, IEEE Transactions on Medical Imaging.

[5]  Nasir M. Rajpoot,et al.  A gamma-gaussian mixture model for detection of mitotic cells in breast cancer histopathology images , 2012, Proceedings of the 21st International Conference on Pattern Recognition (ICPR2012).

[6]  David J. C. MacKay,et al.  Information Theory, Inference, and Learning Algorithms , 2004, IEEE Transactions on Information Theory.

[7]  Mark W. Woolrich,et al.  Variational bayes inference of spatial mixture models for segmentation , 2006, IEEE Transactions on Medical Imaging.

[8]  Jean-Michel Marin,et al.  Bayesian Modelling and Inference on Mixtures of Distributions , 2005 .

[9]  Stephen M. Smith,et al.  A global optimisation method for robust affine registration of brain images , 2001, Medical Image Anal..

[10]  Allan Hanbury,et al.  Computer Analysis of Images and Patterns, 12th International Conference, CAIP 2007, Vienna, Austria, August 27-29, 2007, Proceedings , 2007, CAIP.

[11]  Jian Wang,et al.  Maximum likelihood estimation for compound-gaussian clutter with inverse gamma texture , 2007, IEEE Transactions on Aerospace and Electronic Systems.

[12]  Andrew R. Webb,et al.  Bayesian gamma mixture model approach to radar target recognition , 2003 .

[13]  Yuhong Yang,et al.  Information Theory, Inference, and Learning Algorithms , 2005 .

[14]  N. Filippini,et al.  Group comparison of resting-state FMRI data using multi-subject ICA and dual regression , 2009, NeuroImage.

[15]  Mark W. Woolrich,et al.  Bayesian analysis of neuroimaging data in FSL , 2009, NeuroImage.

[16]  Mark W. Woolrich,et al.  Advances in functional and structural MR image analysis and implementation as FSL , 2004, NeuroImage.

[17]  Christopher M. Bishop,et al.  Pattern Recognition and Machine Learning (Information Science and Statistics) , 2006 .

[18]  Jiabin Wang,et al.  Variational Bayes Inference Based Segmentation of Heterogeneous Lymphoma Volumes in Dual-Modality PET-CT Images , 2011, 2011 International Conference on Digital Image Computing: Techniques and Applications.

[19]  Hongjun Song,et al.  Radar detection based on compound-Gaussian model with inverse gamma texture , 2011 .

[20]  Mark W. Woolrich,et al.  Mixture models with adaptive spatial regularization for segmentation with an application to FMRI data , 2005, IEEE Transactions on Medical Imaging.

[21]  Hanqing Lu,et al.  A variational inference based approach for image segmentation , 2008, 2008 19th International Conference on Pattern Recognition.

[22]  Daniel Fink A Compendium of Conjugate Priors , 1997 .

[23]  Jean-Baptiste Poline,et al.  Bayesian Joint Detection-Estimation of Brain Activity Using MCMC With a Gamma-Gaussian Mixture Prior Model , 2006, 2006 IEEE International Conference on Acoustics Speech and Signal Processing Proceedings.

[24]  C. F. Beckmann,et al.  Estimating an Inverse Gamma distribution , 2016, 1605.01019.