训练实例影响的高效估计
机器学习
2021-11-22 v2 计算与语言
计算机视觉与模式识别
摘要
理解训练实例对神经网络模型的影响有助于提高可解释性。然而,评估这种影响(即如果某个训练实例未被使用,模型的预测将如何变化)是困难且低效的。在本文中,我们提出了一种高效的影响估计方法。我们的方法受dropout启发,dropout将子网络置零并阻止该子网络学习每个训练实例。通过在dropout掩码之间切换,我们可以利用学习了或未学习每个训练实例的子网络来估计其影响。通过在分类数据集上使用BERT和VGGNet进行实验,我们证明了所提出的方法能够捕捉训练影响,增强错误预测的可解释性,并清理训练数据集以提高泛化能力。
引用
@article{arxiv.2012.04207,
title = {Efficient Estimation of Influence of a Training Instance},
author = {Sosuke Kobayashi and Sho Yokoi and Jun Suzuki and Kentaro Inui},
journal= {arXiv preprint arXiv:2012.04207},
year = {2021}
}
备注
This is an extended version of the paper presented at SustaiNLP 2020