Neural Partially Linear Additive Model

Liangxuan Zhu,Han Li,Xuelin Zhang,Lingjuan Wu,Hong Chen
DOI: https://doi.org/10.1007/s11704-023-2662-3
IF: 2.6688
2023-01-01
Frontiers of Computer Science
Abstract:Interpretability has drawn increasing attention in machine learning. Most works focus on post-hoc explanations rather than building a self-explaining model. So, we propose a Neural Partially Linear Additive Model (NPLAM), which automatically distinguishes insignificant, linear, and nonlinear features in neural networks. On the one hand, neural network construction fits data better than spline function under the same parameter amount; on the other hand, learnable gate design and sparsity regular-term maintain the ability of feature selection and structure discovery. We theoretically establish the generalization error bounds of the proposed method with Rademacher complexity. Experiments based on both simulations and real-world datasets verify its good performance and interpretability.
What problem does this paper attempt to address?