Premium
Masked convolutional neural network for supervised learning problems
Author(s) -
Liu Leo YuFeng,
Liu Yufeng,
Zhu Hongtu
Publication year - 2020
Publication title -
stat
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.61
H-Index - 18
ISSN - 2049-1573
DOI - 10.1002/sta4.290
Subject(s) - interpretability , artificial intelligence , computer science , convolutional neural network , machine learning , binary classification , deep learning , artificial neural network , key (lock) , set (abstract data type) , binary number , supervised learning , support vector machine , computer security , arithmetic , mathematics , programming language
Convolutional neural networks (CNNs) have exhibited superior performance in various types of classification and prediction tasks, but their interpretability remains to be low despite years of research effort. It is crucial to improve the ability of existing models to interpret deep neural networks from both theoretical and practical perspectives and to develop new neural network models with interpretable representations. The aim of this paper is to propose a set of novel masked CNN (MCNN) models with better ability to interpret networks and more accurate prediction. The key ideas behind MCNNs are to introduce a latent binary network to extract informative regions of interest that contain important signals for prediction and to integrate the latent binary network with CNNs to achieve better prediction in various supervised learning problems. Extensive numerical studies demonstrate the competitive performance of the proposed MCNN models.