A Differentiable Approach to Multi-scale Brain Modeling

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Chaoming, Lyu, Muyang, Zhang, Tianqiu, He, Sichao, Wu, Si
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917786195853312
author Wang, Chaoming
Lyu, Muyang
Zhang, Tianqiu
He, Sichao
Wu, Si
author_facet Wang, Chaoming
Lyu, Muyang
Zhang, Tianqiu
He, Sichao
Wu, Si
contents We present a multi-scale differentiable brain modeling workflow utilizing BrainPy, a unique differentiable brain simulator that combines accurate brain simulation with powerful gradient-based optimization. We leverage this capability of BrainPy across different brain scales. At the single-neuron level, we implement differentiable neuron models and employ gradient methods to optimize their fit to electrophysiological data. On the network level, we incorporate connectomic data to construct biologically constrained network models. Finally, to replicate animal behavior, we train these models on cognitive tasks using gradient-based learning rules. Experiments demonstrate that our approach achieves superior performance and speed in fitting generalized leaky integrate-and-fire and Hodgkin-Huxley single neuron models. Additionally, training a biologically-informed network of excitatory and inhibitory spiking neurons on working memory tasks successfully replicates observed neural activity and synaptic weight distributions. Overall, our differentiable multi-scale simulation approach offers a promising tool to bridge neuroscience data across electrophysiological, anatomical, and behavioral scales.
format Preprint
id arxiv_https___arxiv_org_abs_2406_19708
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Differentiable Approach to Multi-scale Brain Modeling
Wang, Chaoming
Lyu, Muyang
Zhang, Tianqiu
He, Sichao
Wu, Si
Neural and Evolutionary Computing
Artificial Intelligence
Computational Engineering, Finance, and Science
Neurons and Cognition
We present a multi-scale differentiable brain modeling workflow utilizing BrainPy, a unique differentiable brain simulator that combines accurate brain simulation with powerful gradient-based optimization. We leverage this capability of BrainPy across different brain scales. At the single-neuron level, we implement differentiable neuron models and employ gradient methods to optimize their fit to electrophysiological data. On the network level, we incorporate connectomic data to construct biologically constrained network models. Finally, to replicate animal behavior, we train these models on cognitive tasks using gradient-based learning rules. Experiments demonstrate that our approach achieves superior performance and speed in fitting generalized leaky integrate-and-fire and Hodgkin-Huxley single neuron models. Additionally, training a biologically-informed network of excitatory and inhibitory spiking neurons on working memory tasks successfully replicates observed neural activity and synaptic weight distributions. Overall, our differentiable multi-scale simulation approach offers a promising tool to bridge neuroscience data across electrophysiological, anatomical, and behavioral scales.
title A Differentiable Approach to Multi-scale Brain Modeling
topic Neural and Evolutionary Computing
Artificial Intelligence
Computational Engineering, Finance, and Science
Neurons and Cognition
url https://arxiv.org/abs/2406.19708