Back to Search Start Over

Neural partially linear additive model.

Authors :
Zhu, Liangxuan
Li, Han
Zhang, Xuelin
Wu, Lingjuan
Chen, Hong
Source :
Frontiers of Computer Science; Dec2024, Vol. 18 Issue 6, p1-17, 17p
Publication Year :
2024

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. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
20952228
Volume :
18
Issue :
6
Database :
Complementary Index
Journal :
Frontiers of Computer Science
Publication Type :
Academic Journal
Accession number :
174517815
Full Text :
https://doi.org/10.1007/s11704-023-2662-3