更好的海森矩阵近似能提升影响函数的精度。
Better Hessians Matter: Studying the Impact of Curvature Approximations in Influence Functions
- 通过对比不同海森矩阵近似方法,验证其对数据溯源的影响
- 更精确的近似可使影响得分质量显著提升
- 揭示了K-FAC与GGN间特征值差异是主要误差来源
影响函数为追溯模型预测来源提供了理论框架,但在深度学习中受限于需求解大型病态海森矩阵。已有方法如广义高斯-牛顿(GGN)和克罗内克分解近似曲率(K-FAC)被提出以降低计算复杂度,但这些近似与真实海森矩阵的偏离如何影响数据归因性能尚不明确。本文在控制的分类设置下,系统研究海森矩阵近似质量对影响函数归因效果的影响。实验表明,更优的海森近似能持续提升影响得分质量,支持了近期改进近似的努力。进一步分解主流近似方法的步骤,发现各环节对归因准确性的贡献不同;其中,K-FAC与GGN/EK-FAC特征值的不匹配是导致误差和影响损失的主要原因。该结果指明关键改进方向,有助于在计算效率与归因精度间取得平衡。
原文摘要 · Abstract (English)
Influence functions offer a principled way to trace model predictions back to training data, but their use in deep learning is hampered by the need to invert a large, ill-conditioned Hessian matrix. Approximations such as Generalised Gauss-Newton (GGN) and Kronecker-Factored Approximate Curvature (K-FAC) have been proposed to make influence computation tractable, yet it remains unclear how the departure from exactness impacts data attribution performance. Critically, given the restricted regime in which influence functions are derived, it is not necessarily clear better Hessian approximations should even lead to better data attribution performance. In this paper, we investigate the effect of Hessian approximation quality on influence-function attributions in a controlled classification setting. Our experiments show that better Hessian approximations consistently yield better influence score quality, offering justification for recent research efforts towards that end. We further decompose the approximation steps for recent Hessian approximation methods and evaluate each step's influence on attribution accuracy. Notably, the mismatch between K-FAC eigenvalues and GGN/EK-FAC eigenvalues accounts for the majority of the error and influence loss. These findings highlight which approximations are most critical, guiding future efforts to balance computational tractability and attribution accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。