/swyft

Truncated marginal neural ratio estimation estimation with PyTorch.

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

swyft

PyPI version codecov

swyft is the official implementation of Truncated Marginal Neural Ratio Estimation (TMNRE), a hyper-efficient, simulation-based inference technique for complex data and expensive simulators.

swyft:

  • estimates likelihood-to-evidence ratios for arbitrary marginal posteriors; they typically require fewer simulations than the corresponding joint.
  • performs targeted inference by prior truncation, combining simulation efficiency with empirical testability.
  • seamless reuses simulations drawn from previous analyses, even with different priors.
  • integrates dask and zarr to make complex simulation easy.

swyft is designed to solve the Bayesian inverse problem when the user has access to a simulator that stochastically maps parameters to observational data. In scientific settings, a cost-benefit analysis often favors approximating the posterior marginality; swyft provides this functionality. The package additionally implements our prior truncation technique, routines to empirically test results by estimating the expected coverage, and a dask simulator manager with zarr storage to simplify use with complex simulators.

Related