CP4SBI: Local Conformal Calibration of Credible Sets in Simulation-Based Inference

2508.17077v1 stat.ML, cs.LG 2025-08-27
Авторы:

Luben M. C. Cabezas, Vagner S. Santos, Thiago R. Ramos, Pedro L. C. Rodrigues, Rafael Izbicki

Резюме на русском

## Контекст Симуляционная основанная инференция (SBI) широко используется в экспериментальных науках для инверсии сложных нелинейных моделей с неопределимой вероятностью. Однако, полученные при помощи SBI постериорные оценки часто оказываются малокалиброванными, что приводит к тому, что кредибилити-регионы недооценивают настоящие параметры. Это приводит к недостатку надежности в оценке неуверенности, что является критическим для принятия решений в таких областях, как медицина, экология и физика. Наша мотивация заключается в разработке метода, который сможет улучшить калибровку постериорных регионов в SBI, увеличив надежность и точность оценок неуверенности. ## Метод Мы предлагаем $\texttt{CP4SBI}$, модели-агностическую рамформу, основанную на конформальной калибровке, которая строит кредибилити-регионы с локальной байесовской калибровкой. Наш фреймворк имеет две версии: локальная калибровка с использованием регрессионных деревьев и калибровка на основе оценок CDF. Основной идеей является то, чтобы убедиться в локальной байесовской калибровке на основе локальных оценок вероятности, которые могут быть применены к любой скоринг-функции, включая ХПД, симметричные и квантильные регионы. Функциональность $\texttt{CP4SBI}$ позволяет ему адаптироваться к различным SBI-методам, включая нормализующие потоки и модели скорости сдвига. ## Результаты Мы проводили эксперименты на различных SBI-benchmarks, включая симуляции в управляемых и неуправляемых методах SBI. Мы сравнивали наш подход с другими методами калибровки на таких метриках, как локальная байесовская калибровка, лог-дивергенция и точность оценки. Наши результаты показывают, что $\texttt{CP4SBI}$ существенно улучшает калибровку постериорных регионов, даже при небольших выборках. Мы также показали, что наш метод работает хорошо с разными SBI-подходами, включая нормализующие потоки и модели скорости сдвига, что демонстрирует его модели-агностичность. ## Значимость Наш подход имеет широкую область применения в экспериментальных науках, где SBI используется для инверсии нелинейных моделей с неопределимой вероятностью. Он может быть применен в таких областях, как биология, физика, экономика и машинное обучение. Основные преимущества $\texttt{CP4SBI}$ заключаются в его модели-агностичности, точности и улучшении калибровки. Это может привести к более надежным оценкам неуверенности в принятии решений и моделировании в этих областях. ## Выводы Мы представляем $\texttt{CP4SBI}$, новую модели-агностическую систему калибровки для кредибилити-регионов в SBI. На

Abstract

Current experimental scientists have been increasingly relying on simulation-based inference (SBI) to invert complex non-linear models with intractable likelihoods. However, posterior approximations obtained with SBI are often miscalibrated, causing credible regions to undercover true parameters. We develop $\texttt{CP4SBI}$, a model-agnostic conformal calibration framework that constructs credible sets with local Bayesian coverage. Our two proposed variants, namely local calibration via regression trees and CDF-based calibration, enable finite-sample local coverage guarantees for any scoring function, including HPD, symmetric, and quantile-based regions. Experiments on widely used SBI benchmarks demonstrate that our approach improves the quality of uncertainty quantification for neural posterior estimators using both normalizing flows and score-diffusion modeling.

Ссылки и действия