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

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