Efficient Hierarchical Bayesian Inference for Spatio-temporal Regression Models in Neuroimaging

Abstract

Several problems in neuroimaging and beyond require inference on the parameters of multi-task sparse hierarchical regression models. Examples include M/EEG inverse problems, neural encoding models for task-based fMRI analyses, and climate science. In these domains, both the model parameters to be inferred and the measurement noise may exhibit a complex spatio-temporal structure. Existing work either neglects the temporal structure or leads to computationally demanding inference schemes. Overcoming these limitations, we devise a novel flexible hierarchical Bayesian framework within which the spatio-temporal dynamics of model parameters and noise are modeled to have Kronecker product covariance structure. Inference in our framework is based on majorization-minimization optimization and has guaranteed convergence properties. Our highly efficient algorithms exploit the intrinsic Riemannian geometry of temporal autocovariance matrices. For stationary dynamics described by Toeplitz matrices, the theory of circulant embeddings is employed. We prove convex bounding properties and derive update rules of the resulting algorithms. On both synthetic and real neural data from M/EEG, we demonstrate that our methods lead to improved performance.

arXiv commentAccepted to the 35th Conference on Neural Information Processing Systems (NeurIPS 2021)

References in corpus (16)

• Sparse Signal Recovery with Temporally Correlated Source Vectors Using Sparse Bayesian Learning
• Discrete, 3D distributed, linear imaging methods of electric neuronal activity. Part 1: exact, zero error localization
• Multi Snapshot Sparse Bayesian Learning for DOA Estimation
• Conic geometric optimisation on the manifold of positive definite matrices
• Robust Estimation of Structured Covariance Matrix for Heavy-Tailed Elliptical Distributions
• Covariance Estimation in High Dimensions via Kronecker Product Expansions
• Massive MIMO Unsourced Random Access
• Geometric Mean Metric Learning
• Geodesic Convex Optimization: Differentiation on Manifolds, Geodesics, and Convexity
• A Proximal Approach for a Class of Matrix Optimization Problems
• A hierarchical Bayesian perspective on majorization-minimization for non-convex sparse regression: application to M/EEG source imaging
• Handling correlated and repeated measurements with the smoothed multivariate square-root Lasso
• Dependent relevance determination for smooth and structured sparse regression
• Regularizing Solutions to the MEG Inverse Problem Using Space-Time Separable Covariance Functions
• Statistical control for spatio-temporal MEG/EEG source imaging with desparsified multi-task Lasso
• Thermal Source Localization Through Infinite-Dimensional Compressed Sensing