Flexible and efficient simulation-based inference for models of decision-making
Abstract
Inferring parameters of computational models that capture experimental data is a central task in cognitive neuroscience. Bayesian statistical inference methods usually require the ability to evaluate the likelihood of the model—however, for many models of interest in cognitive neuroscience, the associated likelihoods cannot be computed efficiently. Simulation-based inference (SBI) offers a solution to this problem by only requiring access to simulations produced by the model. Previously, Fengler et al. introduced Likelihood Approximation Networks (LAN, Fengler et al., 2021) which make it possible to apply SBI to models of decision-making, but require billions of simulations for training. Here, we provide a new SBI method that is substantially more simulation-efficient. Our approach, Mixed Neural Likelihood Estimation (MNLE), trains neural density estimators on model simulations to emulate the simulator, and is designed to capture both the continuous (e.g., reaction times) and discrete (choices) data of decision-making models. The likelihoods of the emulator can then be used to perform Bayesian parameter inference on experimental data using standard approximate inference methods like Markov Chain Monte Carlo sampling. We demonstrate MNLE on two variants of the drift-diffusion model (DDM) and show that it is substantially more efficient than LANs: MNLE achieves similar likelihood accuracy with six orders of magnitude fewer training simulations, and is significantly more accurate than LANs when both are trained with the same budget. This enables researchers to perform SBI on custom-tailored models of decision-making, leading to fast iteration of model design for scientific discovery.
Data availability
We implemented MNLE as part of the open source package for SBI, sbi, available at https://github. com/mackelab/sbi. Code for reproducing the results presented here, and tutorials on how to apply MNLE to other simulators using sbi can be found at https://github.com/mackelab/mnle-for-ddms.
Article and author information
Author details
Funding
Deutsche Forschungsgemeinschaft (SFB 1233)
- Jan-Matthis Lueckmann
- Jakob H Macke
Deutsche Forschungsgemeinschaft (SPP 2041)
- Jan Boelts
- Jakob H Macke
Deutsche Forschungsgemeinschaft (Germany's Excellence Strategy MLCoE)
- Jan Boelts
- Jan-Matthis Lueckmann
- Richard Gao
- Jakob H Macke
Bundesministerium für Bildung und Forschung (ADIMEM,FKZ 01IS18052 A-D)
- Jan-Matthis Lueckmann
- Jakob H Macke
HORIZON EUROPE Marie Sklodowska-Curie Actions (101030918)
- Richard Gao
Bundesministerium für Bildung und Forschung (Tübingen AI Center,FKZ 01IS18039A)
- Jan Boelts
- Jakob H Macke
The funders had no role in study design, data collection and interpretation, or the decision to submit the work for publication.
Copyright
© 2022, Boelts et al.
This article is distributed under the terms of the Creative Commons Attribution License permitting unrestricted use and redistribution provided that the original author and source are credited.
Metrics
-
- 3,481
- views
-
- 680
- downloads
-
- 24
- citations
Views, downloads and citations are aggregated across all versions of this paper published by eLife.
Download links
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)
Further reading
-
- Neuroscience
Synaptic heterogeneity is a hallmark of nervous systems that enables complex and adaptable communication in neural circuits. To understand circuit function, it is thus critical to determine the factors that contribute to the functional diversity of synapses. We investigated the contributions of voltage-gated calcium channel (VGCC) abundance, spatial organization, and subunit composition to synapse diversity among and between synapses formed by two closely related Drosophila glutamatergic motor neurons with distinct neurotransmitter release probabilities (Pr). Surprisingly, VGCC levels are highly predictive of heterogeneous Pr among individual synapses of either low- or high-Pr inputs, but not between inputs. We find that the same number of VGCCs are more densely organized at high-Pr synapses, consistent with tighter VGCC-synaptic vesicle coupling. We generated endogenously tagged lines to investigate VGCC subunits in vivo and found that the α2δ–3 subunit Straightjacket along with the CAST/ELKS active zone (AZ) protein Bruchpilot, both key regulators of VGCCs, are less abundant at high-Pr inputs, yet positively correlate with Pr among synapses formed by either input. Consistently, both Straightjacket and Bruchpilot levels are dynamically increased across AZs of both inputs when neurotransmitter release is potentiated to maintain stable communication following glutamate receptor inhibition. Together, these findings suggest a model in which VGCC and AZ protein abundance intersects with input-specific spatial and molecular organization to shape the functional diversity of synapses.
-
- Neuroscience
The integrated stress response (ISR) is a conserved pathway in eukaryotic cells that is activated in response to multiple sources of cellular stress. Although acute activation of this pathway restores cellular homeostasis, intense or prolonged ISR activation perturbs cell function and may contribute to neurodegeneration. DNL343 is an investigational CNS-penetrant small-molecule ISR inhibitor designed to activate the eukaryotic initiation factor 2B (eIF2B) and suppress aberrant ISR activation. DNL343 reduced CNS ISR activity and neurodegeneration in a dose-dependent manner in two established in vivo models – the optic nerve crush injury and an eIF2B loss of function (LOF) mutant – demonstrating neuroprotection in both and preventing motor dysfunction in the LOF mutant mouse. Treatment with DNL343 at a late stage of disease in the LOF model reversed elevation in plasma biomarkers of neuroinflammation and neurodegeneration and prevented premature mortality. Several proteins and metabolites that are dysregulated in the LOF mouse brains were normalized by DNL343 treatment, and this response is detectable in human biofluids. Several of these biomarkers show differential levels in CSF and plasma from patients with vanishing white matter disease (VWMD), a neurodegenerative disease that is driven by eIF2B LOF and chronic ISR activation, supporting their potential translational relevance. This study demonstrates that DNL343 is a brain-penetrant ISR inhibitor capable of attenuating neurodegeneration in mouse models and identifies several biomarker candidates that may be used to assess treatment responses in the clinic.