Inferring stochastic low-rank recurrent neural networks from neural data

2406.16749

YC

0

Reddit

0

Published 6/26/2024 by Matthijs Pals, A Erdem Sau{g}tekin, Felix Pei, Manuel Gloeckler, Jakob H Macke
Inferring stochastic low-rank recurrent neural networks from neural data

Abstract

A central aim in computational neuroscience is to relate the activity of large populations of neurons to an underlying dynamical system. Models of these neural dynamics should ideally be both interpretable and fit the observed data well. Low-rank recurrent neural networks (RNNs) exhibit such interpretability by having tractable dynamics. However, it is unclear how to best fit low-rank RNNs to data consisting of noisy observations of an underlying stochastic system. Here, we propose to fit stochastic low-rank RNNs with variational sequential Monte Carlo methods. We validate our method on several datasets consisting of both continuous and spiking neural data, where we obtain lower dimensional latent dynamics than current state of the art methods. Additionally, for low-rank models with piecewise linear nonlinearities, we show how to efficiently identify all fixed points in polynomial rather than exponential cost in the number of units, making analysis of the inferred dynamics tractable for large RNNs. Our method both elucidates the dynamical systems underlying experimental recordings and provides a generative model whose trajectories match observed trial-to-trial variability.

Create account to get full access

or

If you already have an account, we'll log you in

Overview

  • This paper presents a novel approach to inferring stochastic low-rank recurrent neural networks from neural data.
  • The researchers propose a method for identifying the underlying dynamical structure of neural populations, which can provide insights into the computational mechanisms governing neural activity.
  • The approach combines techniques from Toward Stochastic Neural Computing, Recurrent Deep Kernel Learning for Dynamical Systems, and Latent Variable Sequence Identification for Cognitive Models of Neural to infer the low-dimensional dynamical structure underlying neural population activity.

Plain English Explanation

The human brain is a complex system with billions of interconnected neurons that work together to process information and generate behavior. Researchers are often interested in understanding the underlying computational mechanisms that govern neural activity. This paper presents a new approach to uncover the hidden dynamical structure of neural populations by analyzing the patterns in the data recorded from neural recordings.

The key idea is to model the neural activity as a stochastic low-rank recurrent neural network, where the low-dimensional dynamics capture the essential computations performed by the brain. The researchers develop a method to infer the parameters of this model directly from the neural data, without requiring detailed knowledge of the neural circuit architecture. This allows them to gain insights into the fundamental computational principles underlying neural information processing, which could have important implications for understanding brain function and developing new brain-inspired AI systems.

Technical Explanation

The paper proposes a framework for inferring the parameters of a stochastic low-rank recurrent neural network (SLRNN) model from neural data. The SLRNN model consists of a low-dimensional latent state that evolves over time according to a stochastic dynamical system, and an observation model that maps the latent state to the observed neural activity.

The researchers develop a variational inference approach to learn the SLRNN model parameters, building on techniques from Toward Stochastic Neural Computing, Recurrent Deep Kernel Learning for Dynamical Systems, and Latent Variable Sequence Identification for Cognitive Models of Neural. The key steps involve:

  1. Defining the SLRNN model, which consists of a low-rank recurrent transition matrix and a linear observation model.
  2. Deriving a variational lower bound on the log-likelihood of the neural data under the SLRNN model.
  3. Optimizing the variational parameters and model parameters to maximize the lower bound.

The resulting SLRNN model can capture the essential low-dimensional dynamics underlying the neural population activity, providing a compact representation of the computations performed by the neural circuit. The authors demonstrate the effectiveness of their approach on both simulated and real neural datasets, showing that the inferred SLRNN models can accurately predict the neural responses and uncover meaningful latent state dynamics.

Critical Analysis

The proposed framework for inferring stochastic low-rank recurrent neural networks from neural data is a promising approach that builds on several recent advances in dynamical systems modeling and variational inference. However, the paper also acknowledges several important limitations and areas for further research:

  1. The SLRNN model assumes a linear observation model, which may not always be appropriate for capturing the complex, nonlinear relationships between neural activity and the underlying computations. Extending the model to handle more flexible observation functions could improve its expressive power.
  2. The variational inference procedure relies on several approximations and assumptions, such as the Gaussian form of the posterior distribution. Exploring alternative inference techniques, such as those used in Stretched Measured Neural Predictions of Complex Network Dynamics or Rethinking Spiking Neural Networks as State-Space, may lead to more accurate and robust parameter estimates.
  3. The method assumes that the neural population activity can be well-described by a low-dimensional latent dynamical system. While this assumption is often reasonable, there may be cases where the underlying computations are better captured by more complex, high-dimensional models.

Overall, the paper presents an interesting and potentially valuable approach for uncovering the hidden computational structure of neural systems. However, further research is needed to address the identified limitations and explore the broader applicability of the method to a wider range of neural data and computational models.

Conclusion

This paper introduces a novel framework for inferring stochastic low-rank recurrent neural networks from neural data. The key innovation is the development of a variational inference approach to learn the parameters of a compact SLRNN model that can capture the essential low-dimensional dynamics underlying neural population activity.

The proposed method has the potential to provide important insights into the computational mechanisms governing neural information processing, which could have far-reaching implications for our understanding of brain function and the development of new brain-inspired AI systems. While the approach has several limitations that require further research, the paper represents an important step forward in the field of computational neuroscience and dynamical systems modeling of neural data.



This summary was produced with help from an AI and may contain inaccuracies - check out the links to read the original source documents!

Related Papers

šŸ§ 

Toward stochastic neural computing

Yang Qi, Zhichao Zhu, Yiming Wei, Lu Cao, Zhigang Wang, Jie Zhang, Wenlian Lu, Jianfeng Feng

YC

0

Reddit

0

The highly irregular spiking activity of cortical neurons and behavioral variability suggest that the brain could operate in a fundamentally probabilistic way. Mimicking how the brain implements and learns probabilistic computation could be a key to developing machine intelligence that can think more like humans. In this work, we propose a theory of stochastic neural computing (SNC) in which streams of noisy inputs are transformed and processed through populations of nonlinearly coupled spiking neurons. To account for the propagation of correlated neural variability, we derive from first principles a moment embedding for spiking neural network (SNN). This leads to a new class of deep learning model called the moment neural network (MNN) which naturally generalizes rate-based neural networks to second order. As the MNN faithfully captures the stationary statistics of spiking neural activity, it can serve as a powerful proxy for training SNN with zero free parameters. Through joint manipulation of mean firing rate and noise correlations in a task-driven way, the model is able to learn inference tasks while simultaneously minimizing prediction uncertainty, resulting in enhanced inference speed. We further demonstrate the application of our method to Intel's Loihi neuromorphic hardware. The proposed theory of SNC may open up new opportunities for developing machine intelligence capable of computing uncertainty and for designing unconventional computing architectures.

Read more

4/23/2024

Recurrent Deep Kernel Learning of Dynamical Systems

Recurrent Deep Kernel Learning of Dynamical Systems

Nicol`o Botteghi, Paolo Motta, Andrea Manzoni, Paolo Zunino, Mengwu Guo

YC

0

Reddit

0

Digital twins require computationally-efficient reduced-order models (ROMs) that can accurately describe complex dynamics of physical assets. However, constructing ROMs from noisy high-dimensional data is challenging. In this work, we propose a data-driven, non-intrusive method that utilizes stochastic variational deep kernel learning (SVDKL) to discover low-dimensional latent spaces from data and a recurrent version of SVDKL for representing and predicting the evolution of latent dynamics. The proposed method is demonstrated with two challenging examples -- a double pendulum and a reaction-diffusion system. Results show that our framework is capable of (i) denoising and reconstructing measurements, (ii) learning compact representations of system states, (iii) predicting system evolution in low-dimensional latent spaces, and (iv) quantifying modeling uncertainties.

Read more

5/31/2024

Latent Variable Sequence Identification for Cognitive Models with Neural Bayes Estimation

Latent Variable Sequence Identification for Cognitive Models with Neural Bayes Estimation

Ti-Fen Pan, Jing-Jing Li, Bill Thompson, Anne Collins

YC

0

Reddit

0

Extracting time-varying latent variables from computational cognitive models is a key step in model-based neural analysis, which aims to understand the neural correlates of cognitive processes. However, existing methods only allow researchers to infer latent variables that explain subjects' behavior in a relatively small class of cognitive models. For example, a broad class of relevant cognitive models with analytically intractable likelihood is currently out of reach from standard techniques, based on Maximum a Posteriori parameter estimation. Here, we present an approach that extends neural Bayes estimation to learn a direct mapping between experimental data and the targeted latent variable space using recurrent neural networks and simulated datasets. We show that our approach achieves competitive performance in inferring latent variable sequences in both tractable and intractable models. Furthermore, the approach is generalizable across different computational models and is adaptable for both continuous and discrete latent spaces. We then demonstrate its applicability in real world datasets. Our work underscores that combining recurrent neural networks and simulation-based inference to identify latent variable sequences can enable researchers to access a wider class of cognitive models for model-based neural analyses, and thus test a broader set of theories.

Read more

6/24/2024

Rethinking Spiking Neural Networks as State Space Models

Rethinking Spiking Neural Networks as State Space Models

Malyaban Bal, Abhronil Sengupta

YC

0

Reddit

0

Spiking neural networks (SNNs) are posited as a biologically plausible alternative to conventional neural architectures, with their core computational framework resting on the extensively studied leaky integrate-and-fire (LIF) neuron design. The stateful nature of LIF neurons has spurred ongoing discussions about the ability of SNNs to process sequential data, akin to recurrent neural networks (RNNs). Despite this, there remains a significant gap in the exploration of current SNNs within the realm of long-range dependency tasks. In this study, to extend the analysis of neuronal dynamics beyond simplistic LIF mechanism, we present a novel class of stochastic spiking neuronal model grounded in state space models. We expand beyond the scalar hidden state representation of LIF neurons, which traditionally comprises only the membrane potential, by proposing an n-dimensional hidden state. Additionally, we enable fine-tuned formulation of neuronal dynamics across each layer by introducing learnable parameters, as opposed to the fixed dynamics in LIF neurons. We also develop a robust framework for scaling these neuronal models to deep SNN-based architectures, ensuring efficient parallel training while also adeptly addressing the challenge of non-differentiability of stochastic spiking operation during the backward phase. Our models attain state-of-the-art performance among SNN models across diverse long-range dependency tasks, encompassing the Long Range Arena benchmark, permuted sequential MNIST, and the Speech Command dataset. Moreover, we provide an analysis of the energy efficiency advantages, emphasizing the sparse activity pattern intrinsic to this spiking model.

Read more

6/6/2024