通过局部保真正则化提升可解释神经网络的可靠性与预测性能
Improved Predictive Performance and Interpretability for Mesomorphic Neural Networks Using Local Fidelity Regularization

- 引入局部保真正则化,防止线性层权重集中于单一特征
- 在OpenML上实现比原模型更高的AUROC,兼顾准确率与可解释性
- 适合需要可信解释且不牺牲性能的机器学习应用
可解释的中等结构神经网络(IMNs)结合了深度网络的预测能力与线性模型的可解释性。然而,原始方法缺乏保障机制,导致模型可能将全部解释方差集中到输出层的单个权重上,虽有强预测表现但解释毫无意义。更糟的是,用于鼓励稀疏解的L1正则化反而加剧了这种退化解。为此,本文提出局部保真正则化(LFR),通过将线性输出权重与局部数据变化对齐,阻止权重坍缩,确保解释真实可信。实证结果表明,LFR在OpenML基准套件上不仅未牺牲准确性,反而提升了AUROC。其性能与当前最先进的黑盒模型相当,实现了可靠解释与优异预测的双重优势。源代码与使用说明见https://github.com/hugohammer/LFR-IMN.git。
原文摘要 · Abstract (English)
Interpretable Mesomorphic Neural Networks (IMNs) offer a promising framework that combines the predictive power of deep neural networks with the interpretability of linear models. However, the original formulation lacks safeguards to ensure that the learned interpretations are in fact reliable. In particular, the network is free to concentrate all explanatory variance into a single weight of the linear output layer, achieving strong predictive performance while producing interpretations that are largely meaningless. Paradoxically, the L1 penalty proposed to encourage sparse solutions exacerbates this problem by further incentivizing such degenerate configurations. To address this vulnerability, we introduce Local Fidelity Regularization (LFR), a novel penalty term that prevents degenerate weight collapse by aligning the linear output weights with local data variations. This structural constraint guarantees faithful explanations and substantially improves the reliability of model interpretations. Furthermore, empirical evaluations across the OpenML benchmark suite demonstrate that LFR does not compromise accuracy for explainability; rather, it achieved improved AUROC over the unregularized IMN. By yielding results highly competitive with state-of-the-art black-box models, LFR provides the dual benefit of reliable interpretability and superior predictive performance. Source code and usage instructions are available at https://github.com/hugohammer/LFR-IMN.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。