Login / Signup

Masked convolutional neural network for supervised learning problems.

Leo Yu-Feng LiuYufeng LiuHongtu Zhu
Published in: Stat (2020)
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.
Keyphrases
  • convolutional neural network
  • neural network
  • deep learning
  • machine learning
  • mental health
  • working memory
  • ionic liquid
  • high resolution
  • case control