中文

基于 Hessian 的物理中可靠且可解释机器学习的工具箱

量子物理 2021-11-25 v1 无序系统与神经网络

摘要

应用于量子多体物理的机器学习(ML)技术已成为一个新的研究领域。尽管该方法的数值能力无可否认,但最具表达力的 ML 算法(如神经网络)是黑箱:用户既不知道模型预测背后的逻辑,也不知道模型预测的不确定性。在本工作中,我们提出一个与模型架构无关的可解释性与可靠性工具箱。具体而言,它提供给定测试点上输入数据对预测影响程度的度量、模型预测不确定性的估计,以及模型预测的外推分数。此类工具箱仅需对训练损失函数计算一次 Hessian。我们的工作为在应用于物理及更广义科学的 ML 中系统性使用可解释性与可靠性方法开辟了道路。

关键词

引用

@article{arxiv.2108.02154,
  title  = {Hessian-based toolbox for reliable and interpretable machine learning in physics},
  author = {Anna Dawid and Patrick Huembeli and Michał Tomza and Maciej Lewenstein and Alexandre Dauphin},
  journal= {arXiv preprint arXiv:2108.02154},
  year   = {2021}
}

备注

17 pages, 7 figures, example code is available at https://github.com/Shmoo137/Hessian-Based-Toolbox