Interrogating theoretical models of neural computation with emergent property inference

  1. Sean R Bittner
  2. Agostina Palmigiano
  3. Alex T Piet
  4. Chunyu A Duan
  5. Carlos D Brody
  6. Kenneth D Miller
  7. John Cunningham  Is a corresponding author
  1. Department of Neuroscience, Columbia University, United States
  2. Princeton Neuroscience Institute, United States
  3. Princeton University, United States
  4. Allen Institute for Brain Science, United States
  5. Institute of Neuroscience, Chinese Academy of Sciences, China
  6. Howard Hughes Medical Institute, United States
  7. Department of Statistics, Columbia University, United States
5 figures and 1 additional file

Figures

Figure 1 with 4 supplements
Emergent property inference in the stomatogastric ganglion.

(A) Conductance-based subcircuit model of the STG. (B) Spiking frequency ω(𝐱;𝐳) is an emergent property statistic. Simulated at gel=4.5 nS and gsynA=3 nS. (C) The emergent property of intermediate hub frequency. Simulated activity traces are colored by log probability of generating parameters in the EPI distribution (Panel E). (D) For a choice of circuit model and emergent property, EPI learns a deep probability distribution of parameters 𝐳. (E) The EPI distribution producing intermediate hub frequency. Samples are colored by log probability density. Contours of hub neuron frequency error are shown at levels of 0.525, 0.53, … 0.575 Hz (dark to light gray away from mean). Dimension of sensitivity 𝐯1 (solid arrow) and robustness 𝐯2 (dashed arrow). (F) (Top) The predictions of the EPI distribution. The black and gray dashed lines show the mean and two standard deviations according the emergent property. (Bottom) Simulations at the starred parameter values.

Figure 1—figure supplement 1
Emergent property inference in a 2D linear dynamical system.

(A) Two-dimensional linear dynamical system model, where real entries of the dynamics matrix A are the parameters. (B) The EPI distribution for a two-dimensional linear dynamical system with τ=1 that produces an average of 1 Hz oscillations with some small amount of variance. Dashed lines indicate the parameter axes. (C) Entropy throughout the optimization. At the beginning of each augmented lagrangian epoch (imax=2,000 iterations), the entropy dipped due to the shifted optimization manifold where emergent property constraint satisfaction is increasingly weighted. (D) Emergent property moments throughout optimization. At the beginning of each augmented lagrangian epoch, the emergent property moments adjust closer to their constraints.

Figure 1—figure supplement 2
Analytic contours of inferred EPI distribution.

(A) Probability contours in the a1,1-a2,2 plane were derived from the relationship to emergent property statistic of growth/decay factor real(λ1). (B) Probability contours in the a1,2-a2,1 plane were derived from the emergent property statistic of oscillation frequency 2πimag(λ1).

Figure 1—figure supplement 3
Sampled dynamical systems 𝐳q𝜽(𝐳𝒳) and their simulated activity from 𝐱(t=0)=[22,-22] colored by log probability.

(A) Each dimension of the simulated trajectories throughout time. (B) The simulated trajectories in phase space.

Figure 1—figure supplement 4
EPI optimization of the STG model producing network syncing.

(A) Entropy throughout optimization. (B) The emergent property statistic means and variances converge to their constraints at 25,000 iterations following the fifth augmented lagrangian epoch.

Figure 2 with 3 supplements
Inferring recurrent neural networks with stable amplification.

(A) Wall time of EPI (blue), SNPE (orange), and SMC-ABC (green) to converge on RNN connectivities producing stable amplification. Each dot shows convergence time for an individual random seed. For reference, the mean wall time for EPI to achieve its full constraint convergence (means and variances) is shown (blue line). (B) Simulation count of each algorithm to achieve convergence. Same conventions as A. (C) The predictive distributions of connectivities inferred by EPI (blue), SNPE (orange), and SMC-ABC (green), with reference to 𝐱0=𝝁 (gray star). (D) Simulations of networks inferred by each method (τ=100ms). Each trace (15 per algorithm) corresponds to simulation of one z. (Below) Ratio of obtained samples producing stable amplification, stable monotonic decay, and instability.

Figure 2—figure supplement 1
Architecture parameter comparison of EPI and SNPE.

Number of parameters in deep probability distribution architectures of EPI (blue) and SNPE (orange) by RNN size (N).

Figure 2—figure supplement 2
SNPE convergence was enabled by increasing nround, not natom.

(A) Difference of mean predictions 𝐱0 throughout optimization at N=50 with by simulation count (left) and wall time (right) of SNPE with nround=5,000 (light orange), SNPE with nround=25,000 (dark orange), and EPI (blue). Each line shows an individual random seed. (B) Same conventions as A at N=100 of SNPE with natom=100 (light orange) and natom=1,000 (dark orange). (C) Same conventions as A at N=100 of SNPE with nround=25,000 (light orange) and nround=250,000 (dark orange).

Figure 2—figure supplement 3
Model characteristics affect predictions of posteriors inferred by SNPE, while predictions of parameters inferred by EPI remain fixed.

(A) Predictive distribution of EPI (blue) and SNPE (orange) inferred connectivity of RNNs exhibiting stable amplification with N=2 (top), N=10 (bottom), g=0.01 (left), and g=0.1 (right). (B) Entropy of parameter distribution approximations throughout optimization with N=2 (top), N=10 (bottom), g=0.1 (dark shade), and g=0.01 (light shade). (C) Validation log probabilities throughout SNPE optimization. Same conventions as B. (D) Adherence to EPI constraints. Same conventions as B.

Figure 3 with 4 supplements
Emergent property inference in the stochastic stabilized supralinear network (SSSN).

(A) Four-population model of primary visual cortex with excitatory (black), parvalbumin (blue), somatostatin (red), and VIP (green) neurons (excitatory and inhibitory projections filled and unfilled, respectively). Some neuron-types largely do not form synaptic projections to others (|(Wα1,α2)|<0.025). Each neural population receives a baseline input 𝐡b, and the E- and P-populations also receive a contrast-dependent input 𝐡c. Additionally, each neural population receives a slow noisy input ϵ. (B) Transient network responses of the SSSN model. Traces are independent trials with varying initialization 𝐱(0) and noise ϵ. (C) Mean (solid line) and standard deviation sE(𝐱;𝐳) (shading) across 100 trials. (D) EPI distribution of noise parameters 𝐳 conditioned on E-population variability. The EPI predictive distribution of sE(𝐱;𝐳) is show on the bottom-left. (E) (Top) Enlarged visualization of the σE-σP marginal distribution of EPI q𝜽(𝐳𝒳(5Hz)) and q𝜽(𝐳𝒳(10Hz)). Each black dot shows the mode at each σP. The arrows show the most sensitive dimensions of the Hessian evaluated at these modes. (F) The predictive distributions of σE2+σP2 of each inferred distribution q𝜽(𝐳𝒳(5Hz)) and q𝜽(𝐳𝒳(10Hz)).

Figure 3—figure supplement 1
EPI inferred distribution for 𝒳(10Hz).
Figure 3—figure supplement 2
EPI optimization.

(A) Entropy throughout optimization. (B) The emergent property statistic means and variances converge to their constraints at 8000 iterations following the fourth augmented lagrangian epoch.

Figure 3—figure supplement 3
EPI predictive distributions of the sum of squares of each pair of noise parameters.
Figure 3—figure supplement 4
SSSN simulations for small increases in neuron-type population input (left); average (solid) and standard deviation (shaded) of stochastic fluctuations of responses (right).
Figure 4 with 5 supplements
Inferring rapid task switching networks in superior colliculus.

(A) Rapid task switching behavioral paradigm (see text). (B) Model of superior colliculus (SC). Neurons: LP - Left Pro, RP - Right Pro, LA - Left Anti, RA - Right Anti. Parameters: sW - self, hW - horizontal, vW -vertical, dW - diagonal weights. (C) The EPI inferred distribution of rapid task switching networks. Red/purple parameters indicate modes 𝐳*(sW) colored by sW. Sensitivity vectors 𝐯1(𝐳*) are shown by arrows. (Bottom-left) EPI predictive distribution of task accuracies. (D) Mean and standard error (Ntest = 25, bars not visible) of accuracy in Pro (top) and Anti (bottom) tasks after perturbing connectivity away from mode along 𝐯1(𝐳*) (left), 𝐯task (middle), and 𝐯diag (right).

Figure 4—figure supplement 1
Task accuracy by EPI inferred SC network connectivity.

(A) Same pairplot as Figure 4C colored by Pro-task accuracy. (B) Same as A colored by Anti-task accuracy. (C) Connectivity parameters of EPI distributions versus task accuracies. β is slope coefficient of linear regression, r is correlation, and p is the two-tailed p-value.

Figure 4—figure supplement 2
SC network simulations by regime.

(A) Simulations in network regime 1: 𝐳*(sW=-0.75). (B) Simulations in network regime 2: 𝐳*(sW=0.75).

Figure 4—figure supplement 3
Eigenmodes of SC connectivity.

(A) Invariant eigenvectors of connectivity matrix W. (B) Accuracies for connectivity perturbations when changing λall and λside (λtask and λdiag shown in Figure 4D).

Figure 4—figure supplement 4
EPI optimization of the SC model producing rapid task switching.

(A) Entropy throughout optimization. (B) The emergent property statistic means and variances converge to their constraints at 20,000 iterations following the tenth augmented lagrangian epoch.

Figure 4—figure supplement 5
SC connectivities obtained through brute-force sampling.

(A) Rapid task switching SC connectivities obtained from random sampling. (B) Task accuracies of the inferred distributions from random sampling (top) and EPI (bottom).

Figure 5 with 2 supplements
Responses to optogenetic perturbation by connectivity regime.

(A) Mean and standard error (bars) across recording sessions of task error following delay period optogenetic inactivation in rats. (B) Mean and standard deviation (bars) of task error induced by delay period inactivation of varying optogenetic strength γ across the EPI distribution. (C) (Left) Mean and standard error of Pro and Anti error from regime 1 to regime 2 at γ=0.675. (Right) Correlations of connectivity eigenvalues with Anti error from regime 1 to regime 2 at γ=0.675. (D) (Left) Mean and standard deviation (shading) of responses of the SC model at the mode of the EPI distribution to delay period inactivation at γ=0.85. Accuracy in Pro (top) and Anti (bottom) task is shown as a percentage. (Right) Anti-accuracy following delay period inactivation at γ=0.85 versus accuracy in the Pro-task across connectivities in the EPI distribution.

Figure 5—figure supplement 1
SC responses to delay period inactivation at Anti error saturating levels.

(Left) Mean and standard error of Pro- and Anti-error from regime 1 to regime 2 at γ=0.85.

(Right) Correlations of connectivity eigenvalues with Anti-error from regime 1 to regime 2 at γ=0.85.

Figure 5—figure supplement 2
SC responses to delay period inactivation at experiment matching levels.

(Left) Mean and standard deviation (shading) of responses of the SC model at the mode of the EPI distribution to delay period inactivation at γ=0.675.

Accuracy in Pro (top) and Anti (bottom) task is shown as a percentage. (Right) Anti-accuracy following delay period inactivation at γ=0.675 versus accuracy in the Pro-task across connectivities in the EPI distribution.

Additional files

Download links

A two-part list of links to download the article, or parts of the article, in various formats.

Downloads (link to download the article as PDF)

Open citations (links to open the citations from this article in various online reference manager services)

Cite this article (links to download the citations from this article in formats compatible with various reference manager tools)

  1. Sean R Bittner
  2. Agostina Palmigiano
  3. Alex T Piet
  4. Chunyu A Duan
  5. Carlos D Brody
  6. Kenneth D Miller
  7. John Cunningham
(2021)
Interrogating theoretical models of neural computation with emergent property inference
eLife 10:e56265.
https://doi.org/10.7554/eLife.56265