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,937
- views
-
- 751
- downloads
-
- 31
- 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
-
- Computational and Systems Biology
- Neuroscience
Accumulating evidence to make decisions is a core cognitive function. Previous studies have tended to estimate accumulation using either neural or behavioral data alone. Here, we develop a unified framework for modeling stimulus-driven behavior and multi-neuron activity simultaneously. We applied our method to choices and neural recordings from three rat brain regions—the posterior parietal cortex (PPC), the frontal orienting fields (FOF), and the anterior-dorsal striatum (ADS)—while subjects performed a pulse-based accumulation task. Each region was best described by a distinct accumulation model, which all differed from the model that best described the animal’s choices. FOF activity was consistent with an accumulator where early evidence was favored while the ADS reflected near perfect accumulation. Neural responses within an accumulation framework unveiled a distinct association between each brain region and choice. Choices were better predicted from all regions using a comprehensive, accumulation-based framework and different brain regions were found to differentially reflect choice-related accumulation signals: FOF and ADS both reflected choice but ADS showed more instances of decision vacillation. Previous studies relating neural data to behaviorally inferred accumulation dynamics have implicitly assumed that individual brain regions reflect the whole-animal level accumulator. Our results suggest that different brain regions represent accumulated evidence in dramatically different ways and that accumulation at the whole-animal level may be constructed from a variety of neural-level accumulators.
-
- Neuroscience
The question as to whether animals taste cholesterol taste is not resolved. This study investigates whether the fruit fly, Drosophila melanogaster, is capable of detecting cholesterol through their gustatory system. We found that flies are indifferent to low levels of cholesterol and avoid higher levels. The avoidance is mediated by gustatory receptor neurons (GRNs), demonstrating that flies can taste cholesterol. The cholesterol-responsive GRNs comprise a subset that also responds to bitter substances. Cholesterol detection depends on five ionotropic receptor (IR) family members, and disrupting any of these genes impairs the flies' ability to avoid cholesterol. Ectopic expressions of these IRs in GRNs reveals two classes of cholesterol receptors, each with three shared IRs and one unique subunit. Additionally, expressing cholesterol receptors in sugar-responsive GRNs confers attraction to cholesterol. This study reveals that flies can taste cholesterol, and that the detection depends on IRs in GRNs.