Generative Modeling of Neural Dynamics via Latent Stochastic Differential Equations

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: ElGazzar, Ahmed, van Gerven, Marcel
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915782147964928
author ElGazzar, Ahmed
van Gerven, Marcel
author_facet ElGazzar, Ahmed
van Gerven, Marcel
contents We propose a probabilistic framework for developing computational models of biological neural systems. In this framework, physiological recordings are viewed as discrete-time partial observations of an underlying continuous-time stochastic dynamical system which implements computations through its state evolution. To model this dynamical system, we employ a system of coupled stochastic differential equations with differentiable drift and diffusion functions and use variational inference to infer its states and parameters. This formulation enables seamless integration of existing mathematical models in the literature, neural networks, or a hybrid of both to learn and compare different models. We demonstrate this in our framework by developing a generative model that combines coupled oscillators with neural networks to capture latent population dynamics from single-cell recordings. Evaluation across three neuroscience datasets spanning different species, brain regions, and behavioral tasks show that these hybrid models achieve competitive performance in predicting stimulus-evoked neural and behavioral responses compared to sophisticated black-box approaches while requiring an order of magnitude fewer parameters, providing uncertainty estimates, and offering a natural language for interpretation.
format Preprint
id arxiv_https___arxiv_org_abs_2412_12112
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Generative Modeling of Neural Dynamics via Latent Stochastic Differential Equations
ElGazzar, Ahmed
van Gerven, Marcel
Neurons and Cognition
Machine Learning
We propose a probabilistic framework for developing computational models of biological neural systems. In this framework, physiological recordings are viewed as discrete-time partial observations of an underlying continuous-time stochastic dynamical system which implements computations through its state evolution. To model this dynamical system, we employ a system of coupled stochastic differential equations with differentiable drift and diffusion functions and use variational inference to infer its states and parameters. This formulation enables seamless integration of existing mathematical models in the literature, neural networks, or a hybrid of both to learn and compare different models. We demonstrate this in our framework by developing a generative model that combines coupled oscillators with neural networks to capture latent population dynamics from single-cell recordings. Evaluation across three neuroscience datasets spanning different species, brain regions, and behavioral tasks show that these hybrid models achieve competitive performance in predicting stimulus-evoked neural and behavioral responses compared to sophisticated black-box approaches while requiring an order of magnitude fewer parameters, providing uncertainty estimates, and offering a natural language for interpretation.
title Generative Modeling of Neural Dynamics via Latent Stochastic Differential Equations
topic Neurons and Cognition
Machine Learning
url https://arxiv.org/abs/2412.12112