sbi reloaded: a toolkit for simulation-based inference workflows

2024/11/26 by Jan Boelts, Michael Deistler, Boelts, Jan +63 · 12 citations
Decision Sciences · #FOS: Computer and information sciences #Machine Learning (cs.LG) #Scientific Computing and Data Management #Simulation Techniques and Applications

paper · pdf · doi:10.48550/arxiv.2411.17337

openalex publication_date 2024/11/26 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/30

Abstract

Scientists and engineers use simulators to model empirically observed phenomena. However, tuning the parameters of a simulator to ensure its outputs match observed data presents a significant challenge. Simulation-based inference (SBI) addresses this by enabling Bayesian inference for simulators, identifying parameters that match observed data and align with prior knowledge. Unlike traditional Bayesian inference, SBI only needs access to simulations from the model and does not require evaluations of the likelihood function. In addition, SBI algorithms do not require gradients through the simulator, allow for massive parallelization of simulations, and can perform inference for different observations without further simulations or training, thereby amortizing inference. Over the past years, we have developed, maintained, and extended sbi, a PyTorch-based package that implements Bayesian SBI algorithms based on neural networks. The sbi toolkit implements a wide range of inference methods, neural network architectures, sampling methods, and diagnostic tools. In addition, it provides well-tested default settings, but also offers flexibility to fully customize every step of the simulation-based inference workflow. Taken together, the sbi toolkit enables scientists and engineers to apply state-of-the-art SBI methods to black-box simulators, opening up new possibilities for aligning simulations with empirically observed data.

Cited by

• A Probabilistic Framework for LLM-Based Model Discovery
• Minimum Distance Summaries for Robust Neural Posterior Estimation
• Transformer Embeddings for Fast Microlensing Inference
• Machine Learning-Informed 3+1 Sterile Neutrino Global Fits using Posterior Density Estimation of Electron Disappearance Data
• Black-hole ringdown with templates capturing spin precession: a critical re-analysis of GW190521
• Assessment of Simulation-based Inference Methods for Stochastic Compartmental Models
• Scalable Simulation-Based Model Inference with Test-Time Complexity Control
• A differentiable model of supply-chain shocks
• Reliable Parameter Inference for the Epoch of Reionization using Balanced Neural Ratio Estimation
• Simulation-based inference for neutrino interaction model parameter tuning
• CP4SBI: Local Conformal Calibration of Credible Sets in Simulation-Based Inference
• Quantitative and Predictive Folding Models from Limited Single-Molecule Data Using Simulation-Based Inference

Related

• SBI -- A toolkit for simulation-based inference
• Simulation-Based Inference: A Practical Guide
• Simulation-based inference with the Python Package sbijax
• An Introduction to Bayesian and Frequentist Simulation-Based Inference with Machine Learning
• Reducing Calls to the Simulator in Simulation Based Inference (SBI)