Bayesian Varying-Effects Vector Autoregressive Models for Inference of Brain Connectivity Networks and Covariate Effects in Pediatric Traumatic Brain Injury

Yangfan Ren,Nathan Osborne,Christine B. Peterson,Dana M. DeMaster,Linda Ewing-Cobbs,Marina Vannucci
DOI: https://doi.org/10.48550/arXiv.2405.00535
2024-05-01
Abstract:In this paper, we develop an analytical approach for estimating brain connectivity networks that accounts for subject heterogeneity. More specifically, we consider a novel extension of a multi-subject Bayesian vector autoregressive model that estimates group-specific directed brain connectivity networks and accounts for the effects of covariates on the network edges. We adopt a flexible approach, allowing for (possibly) non-linear effects of the covariates on edge strength via a novel Bayesian nonparametric prior that employs a weighted mixture of Gaussian processes. For posterior inference, we achieve computational scalability by implementing a variational Bayes scheme. Our approach enables simultaneous estimation of group-specific networks and selection of relevant covariate effects. We show improved performance over competing two-stage approaches on simulated data. We apply our method on resting-state fMRI data from children with a history of traumatic brain injury and healthy controls to estimate the effects of age and sex on the group-level connectivities. Our results highlight differences in the distribution of parent nodes. They also suggest alteration in the relation of age, with peak edge strength in children with traumatic brain injury (TBI), and differences in effective connectivity strength between males and females.
Methodology
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is to develop an analytical method for estimating brain connectivity networks while taking into account individual heterogeneity. Specifically, the authors propose a new extension of the multi - agent Bayesian Vector Autoregression (VAR) model, which can estimate group - specific directed brain connectivity networks and take into account the influence of covariates on network edges. This method is particularly suitable for the study of traumatic brain injury (TBI) in children and aims to evaluate the impact of age and gender on group - level connectivity. The method proposed in the paper is named VEVAR (Varying - Effects Vector AutoRegression), which allows for the simultaneous estimation of group - level network structures and edge strengths as (possibly) nonlinear functions of covariate values. By introducing a Gaussian process prior to capture the smooth function of covariate values, the VEVAR model can select the influence of relevant covariates at the group level. In addition, this method uses the Variational Bayes method for posterior inference to improve computational scalability. The authors verified the effectiveness of the VEVAR model through simulated data and actual resting - state fMRI data. The experimental results show that, compared with the existing two - stage methods, VEVAR performs better in group - level edge selection and covariate effect selection. When applied to the data of TBI children and healthy control groups, the research results reveal the altered relationship between age and TBI children, as well as the differences in effective connection strength between males and females. These findings contribute to a deeper understanding of the impact of TBI on children's brain development and its complex relationship with age and gender.