.png){kind=icon}
Return
A self-interpretable module for deep image classification on small data
.png){kind=icon}
OA
AI
DOI：10.1007/s10489-022-03886-6.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Deep neural networks are the driving force of the recent explosion of machine learning applications in everyday life. However, they usually require a lot of training data to work well, and they act as black-boxes, making predictions without any explanation about them. This paper presents Memory Wrap, a module (i.e, a set of layers) that can be added to deep learning models to improve their performance and interpretability in settings where few data are available. Memory Wrap adopts a sparse content-attention mechanism between the input and some memories of past training samples. We show that adding Memory Wrap to standard deep neural networks improves their performance when they learn from a limited set of data, and allows them to reach comparable performance when they learn from the full dataset. We discuss how the analysis of its structure and content-attention weights helps to get insights about its decision process and makes their predictions more interpretable, compared to the same networks without Memory Wrap. We test our approach on image classification tasks using several networks on three different datasets, namely CIFAR10, SVHN, and CINIC10.
Keywords:
Interpretable deep learning
eXplainable Artificial Intelligence
Memory augmented neural networks
Small data
Journal
IF:
3.5
Papers:
7.6K
Citations:
1.7W
Organization
Cited Papers
.png){kind=icon}
Desempenho de crianças do ensino fundamental na solução de problemas aritméticos<A NAME="top"></A>OAAI
OAAIShareSaveShareSaveInfluences of acid treatments of active carbons on NO reduction over carbon-supported copper oxides
PREAI
ShareSaveShareSave
View more