arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.02413stat.MLastro-ph.COastro-ph.IMcs.LG

基于分位数回归的仿真推断

Simulation-Based Inference with Quantile Regression

Princeton University(普林斯顿大学)

He Jia

AI总结:

本文提出神经分位数估计(NQE),通过条件分位数回归与自回归建模实现仿真推断,并引入基于局部CDF的快速可信区域及后处理校准,在多个基准上取得领先性能。

摘要:

我们提出神经分位数估计(NQE),一种基于条件分位数回归的新型仿真推断(SBI)方法。NQE 以自回归方式,在数据和先前后验维度的条件下,为每个后验维度学习单独的一维分位数。后验样本通过使用单调三次 Hermite 样条插值预测的分位数获得,并对尾部行为和多模态分布进行特殊处理。我们引入了一种使用局部累积密度函数(CDF)的贝叶斯可信区域的替代定义,其评估速度显著快于传统的最高后验密度区域(HPDR)。在仿真预算有限和/或已知模型设定错误的情况下,可以将后处理校准步骤集成到 NQE 中,以确保后验估计的无偏性,且额外的计算成本可忽略不计。我们证明 NQE 在各种基准问题上达到了最先进的性能。

英文摘要:

We present Neural Quantile Estimation (NQE), a novel Simulation-Based Inference (SBI) method based on conditional quantile regression. NQE autoregressively learns individual one dimensional quantiles for each posterior dimension, conditioned on the data and previous posterior dimensions. Posterior samples are obtained by interpolating the predicted quantiles using monotonic cubic Hermite spline, with specific treatment for the tail behavior and multi-modal distributions. We introduce an alternative definition for the Bayesian credible region using the local Cumulative Density Function (CDF), offering substantially faster evaluation than the traditional Highest Posterior Density Region (HPDR). In case of limited simulation budget and/or known model misspecification, a post-processing calibration step can be integrated into NQE to ensure the unbiasedness of the posterior estimation with negligible additional computational cost. We demonstrate that NQE achieves state-of-the-art performance on a variety of benchmark problems.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑