为可解释神经模型提供可证明最小特征子集的高效算法
Provably Explaining Neural Additive Models
- 针对NAM模型设计新算法,仅需对数级查询即可生成最小解释
- 相比旧方法计算更快,解释规模更小,且保证最优性
- 适合需要可靠解释的高风险场景,如医疗决策
尽管后验解释方法在神经网络中取得进展,但多数仍依赖启发式策略且缺乏可证明保证。关键思路是识别一个基数最小的输入特征子集,该子集足以决定预测结果。然而,对于标准神经网络,这一任务通常计算不可行,因需指数级验证查询,且每次查询均为NP难问题。本文表明,对于一种更可解释的神经网络家族——神经加法模型(NAMs),我们可高效生成具有此类保证的解释。提出一种模型专用算法,通过并行预处理每个一维组件(耗时对数级于精度要求),仅用对数级验证查询即可生成基数最小的解释。该算法不仅使最小解释变得可行,且即便面对更易计算的子集最小化松弛版本,其表现也更优。实验显示,相比现有方法,本方法解释更小、速度更快;同时,所生成的可证明解释优于常规采样技术,后者无法获得相同可靠性。
原文摘要 · Abstract (English)
Despite significant progress in post-hoc explanation methods for neural networks, many remain heuristic and lack provable guarantees. A key approach for obtaining explanations with provable guarantees is by identifying a cardinally-minimal subset of input features which by itself is provably sufficient to determine the prediction. However, for standard neural networks, this task is often computationally infeasible, as it demands a worst-case exponential number of verification queries in the number of input features, each of which is NP-hard. In this work, we show that for Neural Additive Models (NAMs), a recent and more interpretable neural network family, we can efficiently generate explanations with such guarantees. We present a new model-specific algorithm for NAMs that generates provably cardinally-minimal explanations using only a logarithmic number of verification queries in the number of input features, after a parallelized preprocessing step with logarithmic runtime in the required precision is applied to each small univariate NAM component. Our algorithm not only makes the task of obtaining cardinally-minimal explanations feasible, but even outperforms existing algorithms designed to find the relaxed variant of subset-minimal explanations - which may be larger and less informative but easier to compute - despite our algorithm solving a much more difficult task. Our experiments demonstrate that, compared to previous algorithms, our approach provides provably smaller explanations than existing works and substantially reduces the computation time. Moreover, we show that our generated provable explanations offer benefits that are unattainable by standard sampling-based techniques typically used to interpret NAMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。