arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.13402stat.MLcs.LG

使用可微覆盖概率校准神经仿真推断

Calibrating Neural Simulation-Based Inference with Differentiable Coverage Probability

University of Geneva(日内瓦大学);HES-SO/HEG Genève(瑞士西部高等专业学院/日内瓦管理学院);The University of Tokyo(东京大学);RIKEN(理化学研究所);University of Liège(列日大学)

Maciej Falkiewicz, Naoya Takeishi, Imahn Shekhzadeh, Antoine Wehenkel, Arnaud Delaunoy, Gilles Louppe, Alexandros Kalousis

AI总结:

针对仿真推断中现有算法可能产生过度自信后验的问题,提出一种将可微覆盖概率校准项直接纳入神经模型训练目标的方法,实现端到端反向传播,在六个基准上取得更优的覆盖率和后验密度。

摘要:

贝叶斯推断允许在给定先验信息和证据似然的情况下,表达概率模型下后验信念的不确定性。通常,似然函数仅由仿真器隐式建立,从而产生了对仿真推断(SBI)的需求。然而,如果不确定性量化不准确,现有算法可能产生过度自信的后验(Hermans等,2022),这违背了可信性的整体目的。我们提出在选定的摊销SBI技术中,将校准项直接纳入神经模型的训练目标。通过引入经典校准误差公式的松弛形式,我们实现了端到端的反向传播。所提出的方法不绑定于任何特定神经模型,且与其带来的收益相比,仅产生中等的计算开销。它可直接应用于现有计算流程,实现可靠的黑盒后验推断。我们在六个基准问题上实证表明,在覆盖率和期望后验密度方面,所提出的方法取得了比现有方法更具竞争力或更好的结果。

英文摘要:

Bayesian inference allows expressing the uncertainty of posterior belief under a probabilistic model given prior information and the likelihood of the evidence. Predominantly, the likelihood function is only implicitly established by a simulator posing the need for simulation-based inference (SBI). However, the existing algorithms can yield overconfident posteriors (Hermans *et al.*, 2022) defeating the whole purpose of credibility if the uncertainty quantification is inaccurate. We propose to include a calibration term directly into the training objective of the neural model in selected amortized SBI techniques. By introducing a relaxation of the classical formulation of calibration error we enable end-to-end backpropagation. The proposed method is not tied to any particular neural model and brings moderate computational overhead compared to the profits it introduces. It is directly applicable to existing computational pipelines allowing reliable black-box posterior inference. We empirically show on six benchmark problems that the proposed method achieves competitive or better results in terms of coverage and expected posterior density than the previously existing approaches.

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

↑