Fast and Robust Simulation-Based Inference With Optimization Monte Carlo

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Gkolemis, Vasilis, Diou, Christos, Gutmann, Michael U.
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918425966673920
author Gkolemis, Vasilis
Diou, Christos
Gutmann, Michael U.
author_facet Gkolemis, Vasilis
Diou, Christos
Gutmann, Michael U.
contents Bayesian parameter inference for complex stochastic simulators is challenging due to intractable likelihood functions. Existing simulation-based inference methods often require large number of simulations and become costly to use in high-dimensional parameter spaces or in problems with partially uninformative outputs. We propose a new method for differentiable simulators that delivers accurate posterior inference with substantially reduced runtimes. Building on the Optimization Monte Carlo framework, our approach reformulates inference for stochastic simulators in terms of deterministic optimization problems. Gradient-based methods are then applied to efficiently navigate toward high-density posterior regions and avoid wasteful simulations in low-probability areas. A JAX-based implementation further enhances the performance through vectorization of key method components. Extensive experiments, including high-dimensional parameter spaces, uninformative outputs, multiple observations and multimodal posteriors show that our method consistently matches, and often exceeds, the accuracy of state-of-the-art approaches, while reducing the runtime by a substantial margin.
format Preprint
id arxiv_https___arxiv_org_abs_2511_13394
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Fast and Robust Simulation-Based Inference With Optimization Monte Carlo
Gkolemis, Vasilis
Diou, Christos
Gutmann, Michael U.
Machine Learning
Bayesian parameter inference for complex stochastic simulators is challenging due to intractable likelihood functions. Existing simulation-based inference methods often require large number of simulations and become costly to use in high-dimensional parameter spaces or in problems with partially uninformative outputs. We propose a new method for differentiable simulators that delivers accurate posterior inference with substantially reduced runtimes. Building on the Optimization Monte Carlo framework, our approach reformulates inference for stochastic simulators in terms of deterministic optimization problems. Gradient-based methods are then applied to efficiently navigate toward high-density posterior regions and avoid wasteful simulations in low-probability areas. A JAX-based implementation further enhances the performance through vectorization of key method components. Extensive experiments, including high-dimensional parameter spaces, uninformative outputs, multiple observations and multimodal posteriors show that our method consistently matches, and often exceeds, the accuracy of state-of-the-art approaches, while reducing the runtime by a substantial margin.
title Fast and Robust Simulation-Based Inference With Optimization Monte Carlo
topic Machine Learning
url https://arxiv.org/abs/2511.13394