摘要
广义可加模型 (GAMs) 长期以来一直是用于可理解分析表格数据的强大白盒工具,揭示了每个特征对模型预测的影响。尽管神经网络 (NNs) 在各个领域取得了成功,但与基于树的方法相比,它们在表格数据分析中作为基于 NN 的 GAMs 的应用仍然不够理想,并且 NN-GAMs 中编码器的透明度也阻碍了用户理解网络如何学习函数。在这项工作中,我们提出了一种新的深度表格学习方法,称为原型神经加性模型 (ProtoNAM),该方法将原型引入 GAMs 框架中的神经网络。通过引入基于原型的特征激活,ProtoNAM 可以灵活地对表格特征到输出的非规则映射进行建模,同时保持最终预测的可解释性。我们还提出了一种受梯度提升启发的分层形状函数建模方法,便于发现复杂的特征模式,并使每个网络层的学习过程透明。我们的实证评估表明,ProtoNAM 优于所有现有的基于 NN 的 GAMs,同时还提供了对每个特征学习的形状函数的更多见解。ProtoNAM 的源代码可在 \url{https://github.com/Teddy-XiongGZ/ProtoNAM} 获取。