Abstract
Efficient simulation of large-scale biological neural networks is important for studying the working mechanisms of the brain. Allen Brain Institute proposed a large scale computational model of mice primary visual cortex (V1) which is so far the most detailed model of mice V1. The original model is composed of two parts: lateral geniculate nucleus (LGN) model and V1 spiking neural network (SNN) model. The original model has low computational efficiency and does not support GPU and multithreading acceleration. In this work we we propose several techniques for accelerating the original model. We refactored the original LGN model based on PyTorch and V1 SNN model based on NEST-Simulator, and enabled GPU acceleration and multithreading and multiprocess acceleration. Our LGN model achieved 60 times acceleration in computing speed. The building time of V1 SNN model was accelerated by 5.7 times. When using multiple threads and process, our V1 SNN model achieved 17.8 times acceleration on clusters compared with the original model. Our refactored model is helpful for computational research about the mice V1.
This work was supported by the National Key Research and Development Program of China (No. 2021ZD0200301) and by the National Natural Science Foundation of China (Nos. U19B2034, 62061136001, 61836014).
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Antolík, J., Monier, C., Frégnac, Y., Davison, A.P.: A comprehensive data-driven model of cat primary visual cortex. BioRxiv p. 416156 (2019)
Arkhipov, A., et al.: Visual physiology of the layer 4 cortical circuit in silico. PLoS Comput. Biol. 14(11), e1006535 (2018)
Billeh, Y.N., et al.: Systematic integration of structural and functional data into multi-scale models of mouse primary visual cortex. Neuron 106(3), 388–403 (2020)
Dagum, L., Menon, R.: OpenMP: an industry standard API for shared-memory programming. IEEE Comput. Sci. Eng. 5(1), 46–55 (1998)
Dai, K., et al.: Brain Modeling Toolkit: an open source software suite for multiscale modeling of brain circuits. PLoS Comput. Biol. 16(11), e1008386 (2020)
Markram, H., et al.: Reconstruction and simulation of neocortical microcircuitry. Cell 163(2), 456–492 (2015)
Motta, A., et al.: Dense connectomic reconstruction in layer 4 of the somatosensory cortex. Science 366(6469), eaay3134 (2019)
Paszke, A., et al.: PyTorch: an imperative style, high-performance deep learning library. In: Advances in Neural Information Processing Systems 32 (2019)
Potjans, T.C., Diesmann, M.: The cell-type specific cortical microcircuit: relating structure and activity in a full-scale spiking network model. Cereb. Cortex 24(3), 785–806 (2014)
Ramaswamy, S., et al.: The neocortical microcircuit collaboration portal: a resource for rat somatosensory cortex. Front. Neural Circuits 9, 44 (2015)
Reimann, M.W., Anastassiou, C.A., Perin, R., Hill, S.L., Markram, H., Koch, C.: A biophysically detailed model of neocortical local field potentials predicts the critical role of active membrane currents. Neuron 79(2), 375–390 (2013)
Schmidt, M., Bakker, R., Shen, K., Bezgin, G., Diesmann, M., van Albada, S.J.: A multi-scale layer-resolved spiking network model of resting-state dynamics in macaque visual cortical areas. PLoS Comput. Biol. 14(10), e1006359 (2018)
Snir, M., Gropp, W., Otto, S., Huss-Lederman, S., Dongarra, J., Walker, D.: MPI-the Complete Reference: the MPI core, vol. 1. MIT press, Cambridge (1998)
Teeter, C., et al.: Generalized leaky integrate-and-fire models classify multiple neuron types. Nat. Commun. 9(1), 709 (2018). https://doi.org/10.1038/s41467-017-02717-4,https://doi.org/10.1038/s41467-017-02717-4
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Wang, Z., Wang, K., Hu, X. (2022). Accelerating Allen Brain Institute’s Large-Scale Computational Model of Mice Primary Visual Cortex. In: Fang, L., Povey, D., Zhai, G., Mei, T., Wang, R. (eds) Artificial Intelligence. CICAI 2022. Lecture Notes in Computer Science(), vol 13606. Springer, Cham. https://doi.org/10.1007/978-3-031-20503-3_57
Download citation
DOI: https://doi.org/10.1007/978-3-031-20503-3_57
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-20502-6
Online ISBN: 978-3-031-20503-3
eBook Packages: Computer ScienceComputer Science (R0)