用随机微分方程建模神经动态,融合网络与物理模型,更少参数实现精准预测。
Generative Modeling of Neural Dynamics via Latent Stochastic Differential Equations
- 以连续时间随机微分方程建模神经活动,结合可微漂移与扩散项。
- 在三类数据上预测神经与行为响应,性能媲美黑箱模型但参数少10倍。
- 支持模型混合与解释,适合需要可解释性与不确定性估计的研究者。
我们提出一种概率框架,用于构建生物神经系统的计算模型。该框架将生理记录视为潜在连续时间随机动力系统在离散时间的不完整观测,系统通过状态演化实现计算。我们采用耦合的可微漂移与扩散函数的随机微分方程系统,并使用变分推断估计其状态与参数。该框架可无缝集成文献中的数学模型、神经网络或两者混合,用于学习与比较不同模型。我们在框架中开发了一种混合模型,结合耦合振子与神经网络,从单细胞记录中捕捉潜藏群体动态。在涵盖不同物种、脑区和行为任务的三个神经科学数据集上评估,该模型在预测刺激诱发的神经与行为反应方面表现优异,优于复杂黑箱方法,同时参数量少一个数量级,提供不确定性估计,并具备自然可解释的语言表达。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。