cuBNM: GPU-Accelerated Brain Network Modeling

Brain network modeling uses computer simulations to infer about latent neural properties at micro- and mesoscales by fitting brain dynamic models to empirical data of individual subjects or groups. However, computational costs of (individualized) model fitting is a major bottleneck, limiting the practical feasibility of this approach to larger cohorts and more complex models, and highlighting the need for scalable simulation implementations. Here, we introduce cuBNM, a Python package which leverages parallel processing of graphics processing units to massively accelerate simulations of brain network models. We show running simulations on graphics processing units is several hundred times faster compared to central processing units. We demonstrate the usage of cuBNM by running optimization of group-level and individualized low- and high-dimensional models. As examples of the utility of individualized models, we investigated test-retest reliability and heritability of simulated and empirical measures in the Human Connectome Project dataset. We found simulated features were fairly reliable and significantly heritable, suggesting their biological plausibility. Overall, cuBNM enables large-scale simulations of brain network models, opening new avenues for studying latent neural processes across diverse populations, dense networks, and high-dimensional models, which was previously impractical due to computational constraints.