PriorGuide: Test-Time Prior Adaptation for Simulation-Based Inference

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yang, Yang, Rissanen, Severi, Chang, Paul E., Loka, Nasrulloh, Huang, Daolang, Solin, Arno, Heinonen, Markus, Acerbi, Luigi
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914492747612160
author Yang, Yang
Rissanen, Severi
Chang, Paul E.
Loka, Nasrulloh
Huang, Daolang
Solin, Arno
Heinonen, Markus
Acerbi, Luigi
author_facet Yang, Yang
Rissanen, Severi
Chang, Paul E.
Loka, Nasrulloh
Huang, Daolang
Solin, Arno
Heinonen, Markus
Acerbi, Luigi
contents Amortized simulator-based inference offers a powerful framework for tackling Bayesian inference in computational fields such as engineering or neuroscience, increasingly leveraging modern generative methods like diffusion models to map observed data to model parameters or future predictions. These approaches yield posterior or posterior-predictive samples for new datasets without requiring further simulator calls after training on simulated parameter-data pairs. However, their applicability is often limited by the prior distribution(s) used to generate model parameters during this training phase. To overcome this constraint, we introduce PriorGuide, a technique specifically designed for diffusion-based amortized inference methods. PriorGuide leverages a novel guidance approximation that enables flexible adaptation of the trained diffusion model to new priors at test time, crucially without costly retraining. This allows users to readily incorporate updated information or expert knowledge post-training, enhancing the versatility of pre-trained inference models.
format Preprint
id arxiv_https___arxiv_org_abs_2510_13763
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle PriorGuide: Test-Time Prior Adaptation for Simulation-Based Inference
Yang, Yang
Rissanen, Severi
Chang, Paul E.
Loka, Nasrulloh
Huang, Daolang
Solin, Arno
Heinonen, Markus
Acerbi, Luigi
Machine Learning
Amortized simulator-based inference offers a powerful framework for tackling Bayesian inference in computational fields such as engineering or neuroscience, increasingly leveraging modern generative methods like diffusion models to map observed data to model parameters or future predictions. These approaches yield posterior or posterior-predictive samples for new datasets without requiring further simulator calls after training on simulated parameter-data pairs. However, their applicability is often limited by the prior distribution(s) used to generate model parameters during this training phase. To overcome this constraint, we introduce PriorGuide, a technique specifically designed for diffusion-based amortized inference methods. PriorGuide leverages a novel guidance approximation that enables flexible adaptation of the trained diffusion model to new priors at test time, crucially without costly retraining. This allows users to readily incorporate updated information or expert knowledge post-training, enhancing the versatility of pre-trained inference models.
title PriorGuide: Test-Time Prior Adaptation for Simulation-Based Inference
topic Machine Learning
url https://arxiv.org/abs/2510.13763