The authors use a number of well-defined dynamical systems, such as the van der Pol oscillator, as the hidden rules to simulate spike train data, and they demonstrate the approach’s ability to retrieve the dynamics defined by those equations. The key feature of this approach is that the Gaussian process description allows them to obtain an estimate of the entire landscape (or phase portrait), even where no observations were directly made. It’s like being able to distinguish the entire body of Mozart’s works after hearing just a handful of his compositions. Because of this, we can directly compare the simulated ground truth and the learned dynamics visually, noting a close similarity between the two, especially in important regions of the state space such as fixed points (see figure, right). It will be really exciting to see what the full manifolds learned from real spiking data look like, especially in regions where data is not directly
Daniel Hernandez, working with John Cunningham and Liam Paninski, SCGB investigators at Columbia University, developed Variational Inference for Nonlinear Dynamics (VIND) as another tool to infer hidden structure in neural population activity. VIND takes the same approach in assuming that there is a set of latent or hidden variables whose evolution traces out the manifold, and that the recorded neural activity is a projection of those hidden variables. The algorithm accordingly learns two components: how the hidden variables evolve, and how spikes are generated from them. For the first part, VIND approximates a nonlinear dynamical system as a ‘locally linear’ one, which means the equations are different depending on the location in state space and are learned using an artificial neural network. For the second part, VIND assumes that the spiking data we observe are randomly drawn from complicated probability distributions parameterized by the hidden variables, which are often intractable to learn directly. In that sense, the spikes we observe have an almost improvisational feel, like jazz — the musicians are following a concrete composition, but how they play it varies from one show to another. That variation makes the underlying composition more difficult to decipher, but VIND approximates the complicated probability distributions with Gaussian or Poisson distributions (i.e., variational inference).
Validating their approach on simulated data from the classic Lorenz system (the butterfly attractor), as well as various electrophysiological datasets, they are able to retrieve and denoise the underlying low-dimensional latent dynamics, showing that the method has better performance and speed than several others. Moreover, VIND is able to learn the rules of a system (e.g., single-neuron voltage dynamics) and predict many steps into the future given a different starting point, which is very useful. Its ability to predict future behavior also implies that the system does, to some degree, follow a set of dynamical equations, cashing in on the original assumptions. To the detriment of my analogy, symphonies (at least human-composed ones) do not actually follow a set of dynamical equations as they evolve. Nevertheless, after listening to your favorite artist over and over again, you might have the experience of being able to guess the chord sequence of their new song after hearing just the first few seconds.
Finally, Dan O’Shea, working with David Sussillo and Krishna Shenoy, SCGB investigators at Stanford University, presented his work extending the functionality of LFADS (latent factor analysis via dynamical systems) and its application on a massive mouse whole-cortex Neuropixels dataset. LFADS, previously discussed in a SCGB research summary by Grace Lindsay, similarly assumes that latent neural variables evolve according to a nonlinear dynamical system. LFADS uses techniques from machine learning to train a recurrent neural network (RNN) that can reproduce single-trial neural dynamics. At its core, an RNN is a complex but flexible nonlinear dynamical system, whose connection weights can be tuned through optimization to approximate other dynamical systems without learning the explicit rules. In addition, LFADS is capable of ‘dynamical neural stitching,’ a feature that allows researchers to learn shared neural dynamics for a behavior in datasets where separate groups of neurons were recorded in multiple, sequential experimental sessions. This is like recording the wind, percussion and brass sections playing the same piece on different days and then stitching the tracks together to infer the hidden composition. Check out Dan’s explanatory Twitter thread, with beautiful animations of stitched neural trajectories.