A large-scale neural network training framework for generalized estimation of single-trial population dynamics

Abstract

Recent technical advances have enabled recording of increasingly large populations of neural activity, even during natural, unstructured behavior. Deep sequential autoencoders are the current state-of-the-art for uncovering dynamics from these datasets. However, these highly complex models include many non-trainable hyperparameters (HPs) that are typically hand tuned with reference to supervisory information (e.g., behavioral data). This process is cumbersome and time consuming and biases model selection toward models with good representations of individual supervisory variables. Additionally, it cannot be applied to cognitive areas or unstructured tasks for which supervisory information is unavailable. Here we demonstrate AutoLFADS, an automated model-tuning framework that can characterize dynamics using only neural data, without the need for supervisory information. This enables inference of dynamics out-of-the-box in diverse brain areas and behaviors, which we demonstrate on several datasets: motor cortex during free-paced reaching, somatosensory cortex during reaching with perturbations, and dorsomedial frontal cortex during cognitive timing tasks. We also provide a cloud software package and comprehensive tutorials that enable new users to apply the method without dedicated computing resources.