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

最小距离摘要用于鲁棒神经后验估计

Minimum Distance Summaries for Robust Neural Posterior Estimation

University of Cambridge(剑桥大学)

Sherman Khoo, Dennis Prangle, Song Liu, Mark Beaumont

AI总结:

提出最小距离摘要方法,通过最大均值差异(MMD)在测试时自适应调整摘要统计量,在不修改预训练神经后验估计器的情况下实现鲁棒推断,理论保证鲁棒性并实验验证。

摘要:

基于模拟的推断(SBI)通过首先在先验-模拟器对上训练神经后验估计器(NPE),通常使用低维摘要统计量,实现摊销贝叶斯推断,然后可以在新测试观测上查询以廉价地重复用于快速推断。由于NPE是在训练数据分布下估计的,当观测偏离训练分布时,它容易受到误指定的影响。许多鲁棒SBI方法通过修改NPE训练或引入误差模型来解决这个问题,将鲁棒性与推断网络耦合,损害了摊销和模块化。我们引入了最小距离摘要,一种即插即用的鲁棒NPE方法,独立于预训练NPE自适应调整测试时的摘要统计量。利用最大均值差异(MMD)作为观测数据与摘要条件预测分布之间的距离,自适应摘要从MMD继承了强鲁棒性属性。我们证明该算法可以通过随机傅里叶特征近似高效实现,产生轻量级、无模型的测试时自适应过程。我们为算法的鲁棒性提供了理论保证,并在各种合成和真实世界任务上进行了实证评估,表明在最小额外开销下实现了显著的鲁棒性提升。

英文摘要:

Simulation-based inference (SBI) enables amortized Bayesian inference by first training a neural posterior estimator (NPE) on prior-simulator pairs, typically through low-dimensional summary statistics, which can then be cheaply reused for fast inference by querying it on new test observations. Because NPE is estimated under the training data distribution, it is susceptible to misspecification when observations deviate from the training distribution. Many robust SBI approaches address this by modifying NPE training or introducing error models, coupling robustness to the inference network and compromising amortization and modularity. We introduce minimum-distance summaries, a plug-in robust NPE method that adapts queried test-time summaries independently of the pretrained NPE. Leveraging the maximum mean discrepancy (MMD) as a distance between observed data and a summary-conditional predictive distribution, the adapted summary inherits strong robustness properties from the MMD. We demonstrate that the algorithm can be implemented efficiently with random Fourier feature approximations, yielding a lightweight, model-free test-time adaptation procedure. We provide theoretical guarantees for the robustness of our algorithm and empirically evaluate it on a range of synthetic and real-world tasks, demonstrating substantial robustness gains with minimal additional overhead.

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

↑