Feature Importance Measure of a Multilayer Perceptron Based on the Presingle-Connection Layer

Wenyi Zhang,Xiaohua Shen,Haoran Zhang,Zhaohui Yin,Jiayu Sun,Xisheng Zhang,Lejun Zou
DOI: https://doi.org/10.1007/s10115-023-01959-7
IF: 2.7
2023-01-01
Knowledge and Information Systems
Abstract:In many fields, the interpretability of machine learning models holds equal importance to their prediction accuracy. Highly accurate predictions are possible with a multilayer perceptron (MLP) neural network, but its application in high-risk fields is constrained by its lack of interpretability. To solve this issue, this paper introduces an MLP with a presingle-connection layer (SMLP). The SMLP incorporates a single-to-single connection layer with the ReLU function before the original MLP. By examining the weights of the single-connection layer after training the model, the significance of the input features can be determined. The experimental results demonstrate that this method can accurately measure the feature importance with the MLP. It offers advantages such as a straightforward theory, practical implementation, strong stability, and high reliability when compared with other widely used feature importance algorithms. Moreover, this measure effectively reveals the black box of the MLP, indicates the influence of input features on the prediction, and provides a quantitative standard for feature selection in MLP.
What problem does this paper attempt to address?