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.
Ссылки и действия
Дополнительные ресурсы: