GAMI-Net: An explainable neural network based on generalized additive models with structured interactions

作者:

Highlights:

• A novel explainable neural network is proposed for modeling main effects and structured interactions.

• The GAMI-Net is a disentangled feedforward network with multiple additive subnetworks.

• GAMI-Net takes into account three interpretability constraints: sparsity, heredity, marginal clarity.

• An adaptive training algorithm is developed for training GAMI-Net efficiently.

• GAMI-Net enjoys superior interpretability and outperforms benchmark methods.

摘要

•A novel explainable neural network is proposed for modeling main effects and structured interactions.•The GAMI-Net is a disentangled feedforward network with multiple additive subnetworks.•GAMI-Net takes into account three interpretability constraints: sparsity, heredity, marginal clarity.•An adaptive training algorithm is developed for training GAMI-Net efficiently.•GAMI-Net enjoys superior interpretability and outperforms benchmark methods.

论文关键词:Explainable neural network,Generalized additive model,Pairwise interaction,Interpretability constraints

论文评审过程:Received 4 November 2020, Revised 28 June 2021, Accepted 4 July 2021, Available online 20 July 2021, Version of Record 28 July 2021.

论文官网地址:https://doi.org/10.1016/j.patcog.2021.108192