arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1808.09744cs.CLcs.LGstat.ML
收藏

用于训练模型全局解释的规则归纳

Rule induction for global explanation of trained models

Computational Linguistics & Psycholinguistics Research Center(计算语言学与心理语言学研究中心);University of Antwerp(安特卫普大学)

Madhumita Sushil, Simon Šuster, Walter Daelemans

摘要:

了解训练网络的行为并为其输出找到解释对于提高网络的性能和泛化能力以及确保自动化系统的信任非常重要。以前已经提出了几种方法来通过分析训练的网络来识别和可视化最重要的特征。然而,在大多数情况下,不同特征和类之间的关系丢失了。我们提出了一种技术,以诱导集合的if-then-else规则,捕捉这些关系,以全局解释网络的预测。我们首先计算训练好的网络中特征的重要性。然后,我们用这些特征重要性得分来衡量原始输入,简化转换后的输入空间,最后拟合一个规则归纳模型来解释模型预测。我们发现,输出的规则集可以解释的预测神经网络训练的4类文本分类从20个新闻组数据集的宏观平均F分数为0.80。我们在https://github.com/clips/interpret_with_rules上提供代码。

英文摘要:

Understanding the behavior of a trained network and finding explanations for its outputs is important for improving the network's performance and generalization ability, and for ensuring trust in automated systems. Several approaches have previously been proposed to identify and visualize the most important features by analyzing a trained network. However, the relations between different features and classes are lost in most cases. We propose a technique to induce sets of if-then-else rules that capture these relations to globally explain the predictions of a network. We first calculate the importance of the features in the trained network. We then weigh the original inputs with these feature importance scores, simplify the transformed input space, and finally fit a rule induction model to explain the model predictions. We find that the output rule-sets can explain the predictions of a neural network trained for 4-class text classification from the 20 newsgroups dataset to a macro-averaged F-score of 0.80. We make the code available at https://github.com/clips/interpret_with_rules.

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

↑