Abstract:Recurrent neural networks (RNNs) are powerful models for processing time-series data, but it remains challenging to understand how they function. Improving this understanding is of substantial interest to both the machine learning and neuroscience communities. The framework of reverse engineering a trained RNN by linearizing around its fixed points has provided insight, but the approach has significant challenges. These include difficulty choosing which fixed point to expand around when studying RNN dynamics and error accumulation when reconstructing the nonlinear dynamics with the linearized dynamics. We present a new model that overcomes these limitations by co-training an RNN with a novel switching linear dynamical system (SLDS) formulation. A first-order Taylor series expansion of the co-trained RNN and an auxiliary function trained to pick out the RNN's fixed points govern the SLDS dynamics. The results are a trained SLDS variant that closely approximates the RNN, an auxiliary function that can produce a fixed point for each point in state-space, and a trained nonlinear RNN whose dynamics have been regularized such that its first-order terms perform the computation, if possible. This model removes the post-training fixed point optimization and allows us to unambiguously study the learned dynamics of the SLDS at any point in state-space. It also generalizes SLDS models to continuous manifolds of switching points while sharing parameters across switches. We validate the utility of the model on two synthetic tasks relevant to previous work reverse engineering RNNs. We then show that our model can be used as a drop-in in more complex architectures, such as LFADS, and apply this LFADS hybrid to analyze single-trial spiking activity from the motor system of a non-human primate.

Structure-Exploiting variational inference for recurrent switching linear dynamical systems

Recurrent switching linear dynamical systems

Parsing neural dynamics with infinite recurrent switching linear dynamical systems

Variational learning for switching state-space models

Collapsed Amortized Variational Inference for Switching Nonlinear Dynamical Systems

Variational Inference for Continuous-Time Switching Dynamical Systems

Rapid Identification of Switched Systems: A Data-Driven Method in Variational Framework

Tree-Structured Recurrent Switching Linear Dynamical Systems for Multi-Scale Modeling

Bayesian Inference of Recurrent Switching Linear Dynamical Systems with Higher-Order Dependence

Reverse engineering recurrent neural networks with Jacobian switching linear dynamical systems

Modeling Latent Neural Dynamics with Gaussian Process Switching Linear Dynamical Systems

Learning and Inferring Motion Patterns using Parametric Segmental Switching Linear Dynamic Systems

Switching Autoregressive Low-rank Tensor Models

Recurrent Switching Dynamical Systems Models for Multiple Interacting Neural Populations

Decomposed Linear Dynamical Systems (dLDS) for learning the latent components of neural dynamics

Hybrid Recurrent Models Support Emergent Descriptions for Hierarchical Planning and Control

Long Range Switching Time Series Prediction via State Space Model

On the Identifiability of Switching Dynamical Systems

Discovering group dynamics in coordinated time series via hierarchical recurrent switching-state models