论文标题

分配变化中的准确且强大的特征重要性估计

Accurate and Robust Feature Importance Estimation under Distribution Shifts

论文作者

Thiagarajan, Jayaraman J., Narayanaswamy, Vivek, Anirudh, Rushil, Bremer, Peer-Timo, Spanias, Andreas

论文摘要

随着在关键应用程序中越来越依赖黑框模型的结果,不需要访问模型内​​部设备的事后解释性工具通常被用来使人类理解和信任这些模型。特别是,我们专注于可以揭示输入特征对预测输出的影响的方法类别。尽管采用了广泛的采用,但已知现有的方法遭受以下挑战的困扰:计算复杂性,大型不确定性以及最重要的是,无法处理现实世界中的域转移。在本文中,我们提出了概况,这是一种应对所有这些挑战的新功能重要性估计方法。通过使用与预测模型共同训练的损失估计器和因果目标,概况即使在复杂的分布变化下也可以准确估计特征的重要性得分,而无需任何其他重新训练。为此,我们还制定了训练损失估算器的学习策略,即对比度和辍学校准,并发现它可以有效地检测分布变化。利用几个基准图像和非图像数据的经验研究,就忠诚度和鲁棒性而言,我们对最先进的方法显示出显着改善。

With increasing reliance on the outcomes of black-box models in critical applications, post-hoc explainability tools that do not require access to the model internals are often used to enable humans understand and trust these models. In particular, we focus on the class of methods that can reveal the influence of input features on the predicted outputs. Despite their wide-spread adoption, existing methods are known to suffer from one or more of the following challenges: computational complexities, large uncertainties and most importantly, inability to handle real-world domain shifts. In this paper, we propose PRoFILE, a novel feature importance estimation method that addresses all these challenges. Through the use of a loss estimator jointly trained with the predictive model and a causal objective, PRoFILE can accurately estimate the feature importance scores even under complex distribution shifts, without any additional re-training. To this end, we also develop learning strategies for training the loss estimator, namely contrastive and dropout calibration, and find that it can effectively detect distribution shifts. Using empirical studies on several benchmark image and non-image data, we show significant improvements over state-of-the-art approaches, both in terms of fidelity and robustness.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源