Rapid, interpretable, data-driven models of neural dynamics using Recurrent Mechanistic Models

Thiago B Burghi,Maria Ivanova,Ekaterina O Morozova,Huaxinyu Wang,Eve Marder,Timothy O'Leary
DOI: https://doi.org/10.1101/2024.10.10.617633
2024-10-12
Abstract:Obtaining predictive models of a neural system is notoriously challenging. Detailed models suffer from excess model complexity and are difficult to fit efficiently. Simplified models must negotiate a tradeoff between tractability, predictive power and ease of interpretation. We present a new modelling paradigm for estimating predictive mechanistic-like models of neurons and small circuits that navigates these issues using methods from systems theory. The key insight is that membrane currents can be modelled using two scalable system components optimized for learning: linear state space models, and nonlinear artificial neural networks (ANNs). Combining these components, we construct two types of membrane currents: lumped currents, which are more flexible, and data-driven conductance-based currents, which are more interpretable. The resulting class of models --- which we call Recurrent Mechanistic Models (RMMs) --- can be trained in a matter of seconds to minutes on intracellular recordings during an electrophysiology experiment, representing a step change in performance over previous approaches. As a proof-of-principle, we use RMMs to learn the dynamics of two groups of neurons, and their synaptic connections, in the Stomatogastric Ganglion (STG), a well-known central pattern generator. We show that RMMs are efficiently trained using teacher forcing and multiple-shooting. Due to their reliability, efficiency and interpretability, RMMs enable qualitatively new kinds of experiments using predictive models in closed-loop neurophysiology experiments, and for online estimation of neural properties in living preparations.
Neuroscience
What problem does this paper attempt to address?
### The Problem the Paper Attempts to Solve The paper aims to address the challenge of predictive modeling in neural systems. Specifically, while detailed traditional biophysical models can capture the complex dynamics of neurons and small circuits, these models are too complex to efficiently fit experimental data. On the other hand, purely data-driven models can fit data well, but their general structure limits mechanistic insight and may provide incorrect predictions outside the training data. The paper proposes a new modeling framework—**Recurrent Mechanistic Models (RMMs)**—which combines linear state-space models and artificial neural networks (ANNs) to quickly estimate neuronal behavior. RMMs can be trained on experimental data within minutes, significantly improving performance. Additionally, RMMs offer interpretability, allowing researchers to estimate neural properties in real-time during closed-loop neurophysiological experiments. Specifically, the paper demonstrates how RMMs can be used to learn the dynamics of 2 groups of neurons and their synaptic connections, validated in the Stomatogastric Ganglion (STG). The STG is a well-known Central Pattern Generator (CPG). The results show that RMMs have excellent predictive capabilities during experiments and can recover fundamental features of membrane conductance, thereby supporting mechanistic interpretation.