中文

揭示 Hessian 与决策边界的关联

机器学习 2023-06-13 v1 无序系统与神经网络 机器学习

摘要

理解具有良好泛化性的极小值的性质是深度学习研究的核心。一方面,神经网络的泛化性已被关联到决策边界的复杂度,而这在高维输入空间中难以研究。另一方面,极小值的平坦性已成为一个有争议的泛化代理指标。在本工作中,我们提供了这两种方法之间缺失的纽带,并表明 Hessian 的顶层特征向量刻画了神经网络所学到的决策边界。值得注意的是,Hessian 谱中离群值的数量与决策边界的复杂度成正比。基于这一发现,我们提供了一种研究高维决策边界复杂度的新颖且直接的方法;表明这一联系自然地启发了一种新的泛化度量;最后,我们开发了一种新颖的边界估计技术,该技术结合泛化度量,可精确识别具有简单宽边界的极小值。总体而言,该分析建立了 Hessian 与决策边界之间的联系,并提供了一种识别具有简单宽边界决策边界的极小值的新方法。

关键词

引用

@article{arxiv.2306.07104,
  title  = {Unveiling the Hessian's Connection to the Decision Boundary},
  author = {Mahalakshmi Sabanayagam and Freya Behrens and Urte Adomaityte and Anna Dawid},
  journal= {arXiv preprint arXiv:2306.07104},
  year   = {2023}
}

备注

14 pages, 6 figures + 18-page appendices with 19 figures. Any feedback is very welcome! Code is available at https://github.com/Shmoo137/Hessian-and-Decision-Boundary