New Course: Introduction to Deep Learning and Neural Networks

Learn more

Explainable AI (XAI): A survey of recents methods, applications and frameworks

Ilias Papastratison2021-03-04·16 mins
Machine Learning

Deep learning applications have drawn a lot of attention since they have surpassed humans in many tasks such as image and speech recognition, and recommendation systems. However, these applications lack explainability and reliability.

Deep learning models are usually considered as black boxes that are hard to understand while their underlying mechanism is complex.

They do not justify their decisions and predictions and humans cannot trust them. On the other hand, artificial intelligence algorithms make errors that could be fatal depending on the application.

More specifically, an error in a computer vision system of an autonomous car could lead to a crash, while in the medical area, human lives are depending on these decisions.

ml-black-box Most machine learning models perform as black boxes.

To tackle the aforementioned issues, a plethora of methods have been developed. To this end, eXplainable Artificial Intelligence (XAI) has become a hot research topic in the machine learning community.

These methods aim to provide explanations about machine-deep learning models that are easily understandable by humans.

xai-dl-comparison Comparison of a deep learning and an explainable model.

Categories of Interpretability

Interpretability defines how easily we can understand the cause of a decision that is produced from an algorithm.

The adopted categorization of interpretability methods is based on how explanation information is provided.

In this article, the following categories will be discussed:

  • Visual interpretability methods: visual explanations and plots

  • Textual explanations, given in text form

  • Mathematical or numerical explanations

Visual explanations

Visual explainable methods produce pictures or plots in order to provide information about the model’s decision.

Most methods explain the decision of a model in the form of a saliency map by producing values to reflect the importance and contribution of input components to that decision.

These values can take the form of output probabilities or images like heatmaps. In addition, plot visualization methods produce scatter plots to explain decisions or visualize the data.

Class Activation Mapping (CAM)

One of the first and most popular saliency methods is Class Activation Mapping (CAM) [28]. CAM is able to localize the features of the CNN on the image that are responsible for the classification decision. More specifically, CAM uses a global average pooling layer after the convolutional layers and before the final fully connected layer.

Let fk(x,y)f_{k}(x,y) be the activation unit, wckw_{c}^{k} the weight corresponding to class cc for unit kk. Then, the input to the softmax layer corresponding to class cc for unit kk. Then, the input to the softmax layer corresponding to class c is defined as

Sc=x,ykwckfk(x,y)\mathbf{S}_{c} = \sum_{x,y}\sum_{k}w_{c}^{k}f_{k}(x,y)

Finally, the class activation map McM_c is calculated as :

Mc(x,y)=kwckfk(x,y)M_{c}(x,y) =\sum_{k}w_{c}^{k}f_{k}(x,y)

and shows directly the importance of the activation at spatial point (x,y) to classify it’s class cc .

cam The predicted score is mapped to the last layer in order to generate the activation. The class-important regions are highlighted in CAM. Source: [28]

Gradient-weighted Class Activation Mapping (Grad-CAM)

Later on, Gradient-weighted Class Activation Mapping (Grad-CAM) was introduced. Grad-CAM [22] is an extended work based on CAM, which uses the gradients with respect to the target class cc that flows to the final convolutional layer. Grad-CAM produces a coarse localization map LGradCAMcRv×u\mathbf{L_{Grad-CAM}^{c}} \in \mathbb{R}^{v\times u} of width vv and height uu, which highlights the important pixels for the classification of the image. At first, the gradient ycAk\frac{\partial y^{c}}{\partial \mathbf{A_k}} of the class score ycy^{c} is calculated with respect to the activation maps Ak\mathbf{A_k} of the last convolutional layer. The gradients flow back after being averaged over the activation map's size ZZ and then the neuron's importance weights akca_{k}^{c} are calculated as:

akc=1ZijycAk(i,j)a_{k}^{c}= \frac{1}{Z} \sum_{i} \sum_{j} \frac{\partial y^{c}}{\partial A_{k}(i,j)}

The weighting factor akca_{k}^{c} shows the importance of feature kk for the class cc. Finally, the Grad-CAM heatmaps are produced using the forward propagation activations as:

LGradCAMc=ReLU(kakcAk)\mathbf{L_{Grad-CAM}^{c}}= ReLU(\sum_{k}a_{k}^{c}\mathbf{A_{k}})

grad-cam Overview of Grad-CAM. Source: [22]

Layer-Wise Relevance Propagation (LRP)

Another visual explanation technique that has been adopted is Layer-Wise Relevance Propagation (LRP). LRP [23] is based on the decomposition of the decision and produces relevance scores between the activations x(i)x(i) of neuron ii and its input, and finds the neuron's importance scores Rl(i)R^{l}(i) at the layer ll. More specifically, the relevance scores Rl(i)R^{l}(i) of layer ll are calculated with respect to the layer l+1l+1 as :

Rl(i)=jx(i)w(i,j)ix(i)w(i,j)Rl+1(j)R^{l}(i) = \sum_{j}\frac{x(i)w(i,j)}{\sum_{i}x(i)w(i,j)}R^{l+1}(j)

where w(i,j)w(i,j) is the weight between neuron ii and neuron jj.

The pixel-wise contributions to the classification are displayed as shown below:

lrp LRP visualization. Source: [23]

Subsequently, Peak Response Maps (PRM)

Subsequently, Peak Response Maps (PRM) were introduced for weakly supervised instance segmentation. PRM [29] finds the maximum class activations that specify the class scores in each image location. Then, these activations are back-propagated to the input image to generate the peak response maps. The peak's locations

Pc={(i1,j1),...,(iNc,jNc)}\mathbf{P_{c}} = \{(i_{1},j_{1}), ..., (i_{N^{c}},j_{N^{c}})\}

where NcN^{c} the number of peaks, of the cc-th response map Mc\mathbf{M}_{c}, are extracted from the local maximums inside a window of size 3×33\times 3. The sampling kernel Gx,ycG_{x,y}^{c} is calculated at the forward pass at the point (x,y)(x,y) as:

Gc(x,y)=k=1Ncf(xik,yjk)G^{c} (x,y)= \sum_{k=1}^{N^{c}} f(x-i_{k},y-j_{k})

where x[0,H],y[0,W]x\in[0,H], y\in[0,W] and ff a sampling function that obtains only the features of the peaks. Then, the class confidence score sc\mathbf{s}^{c} is calculated from the convolution of the response map and the sampling kernel as: sc=McGc\mathbf{s}^{c} = \mathbf{M}^{c} * \mathbf{G}^{c}. The gradients that will be back-propagated are

δc=1NcLscGc{\delta}^{c} = \frac{1}{N^{c}}\frac{\partial L}{\partial \mathbf{s}^{c}} \mathbf{G}^{c}

where LL is the classification loss.

prm Peak Response Maps method. Source: [29]

CLass-Enhanced Attentive Response (CLEAR)

CLass-Enhanced Attentive Response (CLEAR) [11] is a similar approach that visualizes the decisions of a deep neural network using the activation values of the network. It used deconvolutions to obtain individual attention maps for each class. After the forward pass, we use deconvolutions to obtain the deconvolved output of layer ll with KK kernels as:

h(l)=k=1Kz(k,l)w(k,l)\mathbf{h}(l) = \sum_{k=1}^K z(k,l) * w(k,l)

where z(l)\mathbf{z}(l) are the feature maps of the layer ll and w(l)\mathbf{w}(l) are the kernel weights. The final response of layer ll is obtained from:

R(l)=h(1)h(2)....h(l)\mathbf{R}(l) = \mathbf{h}(1) \mathbf{h}(2).... \mathbf{h}(l)

The individual attention maps R(x,c)\mathbf{R(x'},c) of class cc and the back-projected input x\mathbf{x'} are computed from all LL layers as :

R(x,c)=h(1)h(2)....h(L)\mathbf{R(x'},c) = \mathbf{h}(1) \mathbf{h}(2).... \mathbf{h}(L)

Then, the dominant class attentive map C(x)\mathbf{C(x')} is constructed as:

C(x)=argmaxcR(x,c)\mathbf{C(x')} = {argmax}_c \mathbf{R(x'},c)

while the dominant response map DC(x)\mathbf{D_C(x')} is constructed of the combination of individual response maps and dominant class attentive maps as:

Dc(x)=R(x,c)\mathbf{D}_c(\mathbf{x'}) = \mathbf{R(x'},c)

The dominant response map shows the attention at each location of the image, while the dominant class-map shows the most important class that was involved in the classification of the image.

Finally, the CLass-Enhanced Attentive Response (CLEAR) map is generated by overlaying the two aforementioned maps as:

M=C(x)+Dc(x).\textbf{M} = \mathbf{C(x')}+ \mathbf{D}_c(\mathbf{x'}).

clear CLEAR method overview. Source: [11]

Visualization of features activations with Deconvolutional Networks

Zeiler et al.[27] tried to visualize the intermediate layers of convolutional neural networks and see what they learn. It was shown that convolutional layers store important information about the images as well as that deeper layers learn more complex patterns. In addition, de-convolutional neural networks were adopted in order to reconstruct the input images from feature maps in reverse order. This inverse operation creates an approximate image showing that CNNs have stored the most information of the image.

deconvolutional-neural-network De-convolutional neural network. Source: [27]

DeepResolve

On the other hand, DeepResolve [12] method uses feature maps from intermediate layers and examines how the network combines those features to classify an input image. DeepResolve computes a class-specific image that is named as feature importance map (FIM):

Hc=argmaxH(Sc(H)λH22)\mathbf{H^{c}} = {argmax}_{\mathbf{H}}( S_{c}(\mathbf{H})-\lambda{||\mathbf{H}||}^{2}_{2})

where cc is the target class, ScS_{c} is the class score obtained from the last layer and HRK×W\mathbf{H} \in \mathbb{R}^{K\times W} contains the feature maps with size WW of all KK neurons from a specific layer. Then, the feature importance score (FIV) Φc=(ϕc1,ϕc2,...,ϕck)\mathbf{\Phi}_c = (\phi_c^1, \phi_c^2, ..., \phi_c^k), where ϕck\phi_{c}^{k} is calculated for each neuron from the global average of FIM as:

ϕck=1Wi=1(Hk(i))c,\phi_{c}^{k} = \frac{1}{W} \sum_{i=1}(H^{k}(i))_c,

where ii is the index of the neuron and kk the index of the channel in a layer . This process is initialized randomly and is repeated TT times with different initial parameters to get several estimations of Hct,Φct\mathbf{H}_c^t, \mathbf{\Phi}_c^t. Afterwards, the weighted variance ILckIL_c^k is calculated as:

ILck=var(ϕct)IL_c^k = var(\phi_c^t)

and is used to obtain the overall neuron importance scores (ONIVs ) Φˉc\mathbf{\bar{\Phi}}_c, to find class similarities and differences. ONIVs show the importance of each class and correlations between them. They are used to construct the similarity matrix S\mathbf{S}.

The Class difference matrix is calculated as:

DCiCj=ΦˉCiΦˉCjD_{C_i C_j} = \bar{\Phi}_{C_i} - \bar{\Phi}_{C_j}

between each pair of classes Ci,CjC_i,C_j.

deep-resolve Illustration of DeepResolve’s working flow. Source: [12]

SCOUTER

A visual explanation method named SCOUTER [13] was recently introduced and is not based on feature maps and gradients to explain decisions. SCOUTER adopts a slot-attention classification layer instead of a fully connected layer.

The output features F\mathbf{F} (from a convolutional neural network) are transformed to a smaller dimension through another convolutional layer, while a position embedding layer models the spatial information. A self-attention mechanism is used to obtain the dot-product attention as :

A(t)=σ(Q(W(t))K(F)),\mathbf{A}^{(t)} = \sigma (Q(\mathbf{W}^{(t)})K(\mathbf{F})),

where Q,KQ, K are fully-connected layers, W(t)\mathbf{W}^{(t)} are the slot weights and σ\sigma is the sigmoid function.

Then, the weighted feature map is calculated as :

U(t)=A(t)F(t)\mathbf{U}^{(t)} = \mathbf{A}^{(t)}\mathbf{F}'^{(t)}

A recurrent GRU layer updates the slot weights as follows:

W(t+1)=GRU(U(t),W(t))\mathbf{W}^{(t+1)} = GRU(\mathbf{U}^{(t)},\mathbf{W}^{(t)})

Each slot produces an interpretable confidence score o=(o1,o2,...,on\mathbf{o}=(o_{1}, o_{2}, ..., o_{n} for all classes as:

o=xSlote(F)=eU(t)1c,\mathbf{o} = xSlot_{e}(\mathbf{F}) = e\cdot \mathbf{U}^{(t)}\mathbf{1_{c}},

where e[1,1]e \in [-1,1] is a tunable hyper-parameter that makes the module to focus on positive and negative explanations, respectively and 1c\mathbf{1_{c}} a vector with ones, 1cRC\mathbf{1_{c}}\in \mathbb{R}^C.

scouter Illustration of Scouter. Source: [13]

Visual feedback

In [19], the authors proposed an interpretable method to identify relevant features for image classification. During training, the most important layers and filters for classification are identified, while in test time visual maps are generated to show the image locations that are responsible for this decision. More specifically, the class jj is predicted by the linear combination wjRm\mathbf{w_{j}} \in \mathbb{R}^{m}, mm the number of neurons, of its responses xj\mathbf{x_{j}}. After storing all the responses XRN×m\mathbf{X} \in \mathbb{R}^{N\times m} for NN images of the training set of the network FF, the following optimization problem is solved:

W=argminWXTWLTF2\mathbf{W}^{*} = argmin_{\mathbf{W}} {||\mathbf{X}^{T}\mathbf{W} -\mathbf{ L}^{T} ||}^{2}_{F}

where LL are the ground-truth labels, to find the most relevant features of each class.

visual-feedbacks Visual explanations using relevant features. Source: [19]

At test time, the internal activations and the learned weights W\mathbf{W} are used to generate the decision after the forward pass of the test image I\mathbf{I}. Then, a class prediction is calculated as y^=F(I)\hat{y} = F(\mathbf{I}) and we store the internal activations xI\mathbf{x_I}. Finally, the response is generated from r=wy^xI\mathbf{r} = \mathbf{w}_{\hat{y}} \mathbf{x_I}, which are used for visualizations that highlight the pixels responsible for this decision.

Plot visualization methods

In this section, we will describe methods that adopt scatter-plots or graph visualizations to generate explanations.

T-distributed stochastic neighbor embedding (t-SNE) is a scatter-plot method that projects high-dimensional data in two or three-dimensional spaces. t-SNE uses conditional probabilities to represent the distances between data points and find similarities. Finally, it uses a similar probability distribution over the points in the two or three-dimensional map and it minimizes the Kullback–Leibler divergence between these distributions.

Visualizing the Hidden Activity of Neural Networks with tSNE

In [20], the authors use t-SNE to visualize the activations of the neurons and the learned representations of the data. It is shown that these projections can provide valuable feedback about the relationships between neurons and classes. hidden-activity-tsne Visualization of hidden activity of neurons on MNIST dataset. Source: [20]

Explain features with PCA

In [3], Principal Component Analysis (PCA) was adopted to explain features from deep neural networks.

Given an input image of an image rθΩr_{\theta} \in \mathbf{\Omega} with index θ[1,Θ]\theta \in [1,\Theta], we obtain the output high-dimensional image representations of the CNN F^L(rθ)\mathbf{\hat{F}}^{L}(r_{\theta}). After centering these vectors by subtracting the mean as:

FL(rθ)=F^L(rθ)1Θt=1ΘF^L(rt)\mathbf{F}^{L}(r_{\theta})=\mathbf{\hat{F}}^{L}(r_{\theta})-\frac{1}{\Theta}\sum_{t=1 }^{\Theta}\mathbf{\hat{F}}^{L}(r_{t})

we compute the eigenvectors by finding the eigenvalues of the covariance matrix:

1Θθ=1Θ(FL(θ))(FL)(θ)T\frac{1}{\Theta}\sum_{\theta =1}^{\Theta}(\mathbf{F}^{L}(\theta)){(\mathbf{F}^{L})(\theta)}^{T}

Then, the embeddings with the largest variance, i.e., the largest eigenvalues, are projected. In addition, the authors assume that the images can be decomposed into linear combinations of scene factors such as the view (position, rotation), colors or lightning and perform again the PCA dimensionality reduction on the decomposed features. Given parameters Θ=Θ1,Θ2,...,ΘN\mathbf{\Theta} = \Theta_1, \Theta_2,..., \Theta_N and a sample image tt with parameters θΘ\mathbf{\theta}\in\mathbf{\Theta}, we obtain the features for a specific scene factor kk as:

FkL(t)=ΘkΘθΘθk=tFL(θ)\mathbf{F}_k^L(t) = \frac{|\Theta_k|}{|\Theta|}\sum_{\theta\in\Theta|\theta_k=t}\mathbf{F}^L(\theta)

In the figure below, image embeddings are projected with respect to different image factors.

features-pca Image embeddings projection. Source: [3]

TreeView

TreeView [25] is a method that tries to partition the feature space and into smaller subspaces where each subspace represents a specific factor. At first, the input data X\mathbf{X} is transformed into features Y\mathbf{Y}. Subsequently, features Y\mathbf{Y} are classified and transformed to label space Z\mathbf{Z}. The aforementioned transformations are denoted as T1:XYT_1 : \mathbf{X} \rightarrow \mathbf{Y} and T2:YZT_2: \mathbf{Y}\rightarrow \mathbf{Z}}.

We partition the feature space of Y\mathbf{Y} into KK partitioned subspaces, which are constructed by clustering similar neurons according to their activations. Each cluster ii describes a specific factor SiS_i . Then, a new KK-dimensional feature is constructed from the cluster labels and a decision tree creates the visualization as shown in Figure . For a layer ll, let us denote the neuron's responses as YlRNl×T\mathbf{Y}_l \in \mathbb{R}^{N_l\times T}, where NlN_l are the filters and TT are the number of data. Yl\mathbf{Y}_l is clustered within KK clusters(factors) with activations FlRK×Nl×T\mathbf{F}_l\in \mathbb{R}^{K\times N_l\times T} according to the similarities of hidden activations. Then, the new interpretable features MRK×T\mathbf{M}\in \mathbb{R}^{K \times T} are constructed using the cluster label. Finally, a classifier PliP^i_l is trained for each factor ii using features M\mathbf{M} and predicts the cluster label.

treeview TreeView explanation. Source: [25]

Textual explanation methods

Some works have focused on textual interpretability. In general, textual explanation methods produce natural language-text to interpret the decisions.

Cell Activation Value

Cell Activation Values [8] is an explainability method for LSTMs. This method adopts character-level language to understand the long-term dependencies of LSTM modules. The input characters are projected into a lower-dimensional space. Subsequently, these vectors are fed to the LSTM at each timestep and projected to word sequences with fully connected layers. The activation values at each timestep model the next character in the sequence and are used to interpret the model.

Interpnet

Recently, Barratt et. al. [4] proposed a deep neural network, named Interpnet, that can be combined with a classification architecture and generate explanations. Let us consider a simple network as follows:

y=softmax(W1relu(W2x+b2)+b1)\mathbf{y} = softmax(\mathbf{W}_{1}relu(\mathbf{W}_{2}\mathbf{x}+\mathbf{b}_{2})+\mathbf{b}_{1})

and the internal activations f1,f2,f3\mathbf{f}_1, \mathbf{f}_2, \mathbf{f}_3

f1=x,f2=relu(W2x+b2),f3=softmax(W1relu(W2x+b2)+b1)\mathbf{f}_1 = \mathbf{x}, \mathbf{f}_2 = relu(\mathbf{W}_{2}\mathbf{x}+\mathbf{b}_{2}), \mathbf{f}_3 = softmax(\mathbf{W}_{1}relu(\mathbf{W}_{2}\mathbf{x}+\mathbf{b}_{2})+\mathbf{b}_{1})

Interpnet uses the concatenated vector r=[f1;f2;f3]\mathbf{r} =[\mathbf{f}_1; \mathbf{f}_2; \mathbf{f}_3] as input to a language model, such as LSTM, to generate explanation captions E(x,y)E(x,y). An example of interpretable text generated from Interpnet is depicted below.

interpnet Interpnet generates explanations for the input images. Source:[4]

Visual Question Answering (VQA)

Here, the authors proposed a Visual Question Answering (VQA) [14] framework that jointly attends the image regions and the words of the question to generate the answer as depicted in Figure. At first, the words of the question Q=(q1,q2,...,qT)\mathbf{Q} = (\mathbf{q}_{1}, \mathbf{q}_{2}, ..., \mathbf{q}_{T}) are projected with an embedding layer into a lower-dimensional space to obtain the word embeddings Qw=(q1w,q2w,...,qTw)\mathbf{Q}^{w} = (\mathbf{q}^{w}_{1}, \mathbf{q}^{w}_{2},..., \mathbf{q}^{w}_{T}). Then, a 1d convolutional layer with kernel size up to 3 and a max-pooling layer are applied to model relationships between neighboring words. Finally, an LSTM module models the long-term dependencies and extracts the hidden representation of the question qth\mathbf{q}_{t}^{h} for each timestep tt as

qth=max(LSTM(conv(qt:t+sh)))\mathbf{q}_{t}^{h} = max(LSTM(conv(\mathbf{q}_{t:t+s}^{h})))

where ss is the receptive field of the 1D convolution layer.

A co-attention mechanism takes as input the image features V=(v1,v2,...,vN)\mathbf{V}=(\mathbf{v_1},\mathbf{v_2},...,\mathbf{v_N}) and the word representations at each hierarchy level r(w,p,s)\mathbf{r} \in (w, p, s), i.e., word (w)(w), phrase (p)(p) and sentence (s)(s), to generate the attended image features vattr\mathbf{v}^{r}_{att} and question features qattr\mathbf{q}^{r}_{att} , respectively. The final answer prediction is based on all the co-attended image and question features, which are modelled from multiple fully-connected layers as:

hw=tanh(Ww(qattw+vattw)\mathbf{h}^w = tanh(\mathbf{W}_w(\mathbf{q}^{w}_{att} + \mathbf{v}^{w}_{att})
hp=tanh(Wp[(qattp+vattp),hw]\mathbf{h}^p = tanh(\mathbf{W}_p[(\mathbf{q}^{p}_{att} + \mathbf{v}^{p}_{att}),\mathbf{h}^w]
hs=tanh(Ws[(qatts+vatts),hp]\mathbf{h}^s = tanh(\mathbf{W}_s[(\mathbf{q}^{s}_{att} + \mathbf{v}^{s}_{att}),\mathbf{h}^p]
p=softmax(Whhs)\mathbf{p} = softmax(\mathbf{W}_h\mathbf{h}^s)

where Ww,Wp,Ws,Wh\mathbf{W}_w, \mathbf{W}_p, \mathbf{W}_s, \mathbf{W}_h are the weights of the fully-connected layers.

vqa Example of questions and answers predicted word-level co-attention maps, phrase-level co-attention maps and question-level co-attention maps. Source: [4]

Semantic information to interpret Neural Networks

In [7], the authors employed semantic information to interpret deep neural networks (DNNs) for video captioning. A sample video-description pair has a video x\mathbf{x} with nn frames and NdN_d target descriptions Y=(y1,y2,...,yNd)\mathbf{Y} = (\mathbf{y}_1, \mathbf{y}_2,..., \mathbf{y}_{N_d}). Each description yY\mathbf{y} \in \mathbf{Y} has NsN_s words. The video encoder extracts video features V=(v1,v2,vn)Rn×Dv\mathbf{V} = (\mathbf{v_1}, \mathbf{v_2}, \mathbf{v_n})\in \mathbb{R}^{n\times D_v}, DuD_u the feature dimension. The video representations are used by an attention decoder to generate the captions. The decoder at timestep tt takes as input the concatenated vector d=[yt1;ϕt(V)]\mathbf{d} = [\mathbf{y}_{t-1};\phi_t(\mathbf{V})], where ϕt(V)=i=1naitvi\phi_t(\mathbf{V}) = \sum_{i=1}^{n}a_i^t\mathbf{v}_i is the weighted sum of the features. The weight aita_i^t is calculated as:

ait=exp(watanh(Uaht1+Tavi+ba))j=1nexp(watanh(Uaht1+Tavj+ba))a_i^t= \frac{exp(\mathbf{w}_a tanh(\mathbf{U}_a \mathbf{h}_{t-1}+\mathbf{T}_a \mathbf{v}_i +\mathbf{b}_a))}{ \sum_{j=1}^n exp(\mathbf{w}_a tanh(\mathbf{U}_a \mathbf{h}_{t-1}+\mathbf{T}_a \mathbf{v}_j +\mathbf{b}_a))}

ba,Ta,Ua,wa\mathbf{b}_a, \mathbf{T}_a, \mathbf{U}_a, \mathbf{w}_a are the parameters of the decoder. Finally, a classifier predicts the next word of the sentence from the probability distribution:

pt=softmax(Wp[ht,ϕt(V),yt1]+bp)\mathbf{p}_t = softmax(\mathbf{W}_p [\mathbf{h}_t, \phi_t (\mathbf{V}), \mathbf{y}_{t-1}] + \mathbf{b}_p)

The system uses descriptions of humans, denoted as s\mathbf{s}, that have information about the data. These descriptions are embedded in the network with a loss function defined as:

LI(v,s)=f(1ni=1nvi)s22L_I(\mathbf{v},\mathbf{s}) = {||f(\frac{1}{n}\sum_{i=1}^n \mathbf{v}_i) - \mathbf{s}||_2^2}

and guide the learning process to learn interpretable features. This guides the neurons of the network to be associated with a specific topic and the whole network can be easily understandable by humans instead of being a black-box model.

semantic-information Interpetable training process of deep neural networks. Source: [7]

Visual dialog

In [6], the authors introduced a new task where an AI agent attempts a conversation with humans about visual content. A human makes questions about an image e.g., what color an object is, and the AI agent tries to answer. More specifically, the AI agent uses an encoder-decoder architecture that embeds the visual content and the history of the dialog to develop the next answer.

visual-dialog Example of visual dialog with an AI agent. Source: [7]

Numerical explanations

Concept Activation Vectors (CAVs)

Concept Activation Vectors (CAVs) [10] aim to explain the high-dimensional internal representations of neural networks. Given user-defined sets PC\mathbf{P}_C of a specific concept CC, we seek vectors in the space of hidden activations flf_{l} that describes it. Then, CAVs are defined as vectors orthogonal to the hyperplane learned after training a binary classifier uClu_{C}^{l} for the concept CC in any layer ll, to separate examples whether they belong or not in the concept. Finally, the sensitivity of the class kk is calculated as:

SC,k,l=hk,l((fl(x))uCl)S_{C,k,l} = \mathbf{h}_{k,l}(\nabla(f_{l}(\mathbf{x}))u_{C}^{l})

where hk,l(x)\mathbf{h}_{k,l}(\mathbf{x}) are the predictions of sample x\mathbf{x}.

Linear classifiers for features inspection

In [1], the authors proposed to train linear classifiers and inspect the features of any layer. A linear classifier is fitted to the intermediate layers to monitor the features and measures how suitable they are for classification.

Given the features hk\mathbf{h}_{k} at layer kk the linear probe is defined as:

fk(hk)=softmax(Whk+b)\mathbf{f}_{k}(\mathbf{h}_{k}) = softmax(\mathbf{Wh}_{k}+\mathbf{b})

The probe learns if the information from layer kk is useful for the classification of the input.

In general, it is proved that the most useful information is carried by the deeper layers of the network.

Local Interpretable Model-Agnostic Explanations (LIME)

Local Interpretable Model-Agnostic Explanations (LIME) [21] is able to interpret the predictions of any model-classifier ff by learning a local explainable model gGg\in G, where GG is a class of interpretable models such as a linear classifier or a decision tree. We also measure the complexity Ω(g)\Omega (g) of the model which is also a significant factor of how easily the explanations are generated. In addition, we calculate the error of gg in approximating ff using a loss or distance function, denoted as L(f,g)L(f,g). Finally, the explanation ξ(g)\xi (g) is calculated from the optimization of

ξ(g)=argminL(f,g)+Ω(g)\xi(g) = argmin L(f,g)+\Omega (g)

Applications

In this section, we will present explainable artificial intelligence methods that have been applied in some real-world tasks, such as autonomous driving and healthcare. These methods develop explainable algorithms to interpret results and improve their decisions or actions according to the task. Recent self-driving systems have adopted interpretation techniques to improve the actions of the autonomous driving system and reduce the risk of a crash. This is also important to increase the trust between humans and AI machines.

Explainable decisions for autonomous cars

In [26], the authors proposed a new explainable self-driving system inspired by the reactions and decisions of humans during driving. The proposed method consists of a CNN to extract features from the input image, while a global module generates the scene context from those features and provides information about the location of the objects. A local branch is employed to select the most important objects of the scene and associate them with the scene context to generate the actions and explanations. Finally, visual explanations are produced for the input image.

explainable-autonomous-cars Example of actions and explanations of a self-driving system. Source: [26]

Similarly in [9], the authors proposed an autonomous driving architecture that is assisted and trained with the help of humans.

The system adopts a visual encoder to segment the objects of the input video stream. A vehicle controller is trained to generate spoken text of the commands, i.e., stops the car because the traffic light is red. In addition, the controller generates attention maps to highlight the important regions and explain their decisions. To further enhance the robustness of the system, an observation generator is employed that summarizes frames of the video and produces general observations that must be considered during driving. These observations are also fed to the vehicle controller to improve its decisions.

autonomous-cars-system-overview System overview. Source: [26]

Explainable medical systems

Artificial intelligence systems have also been implemented for medical applications. Deep learning has shown significant results especially in medical imaging and drug discovery. Recently, researchers have focused towards explainable medical systems to assist medical experts and provide useful explanations so that any expert can understand the predictions of a system. In [5], the authors focused on the detection of coronavirus from x-ray images. They proposed a deep convolutional network to extract features from images and detect if the patient is healthy or diagnosed with pneumonia or coronavirus. Then they use Grad-CAM [26] to provide visual explanations and mark the areas of the x-ray that are affected.

XAI frameworks

explainer ExplAIner pipeline. Source: [24]

In this section, we will highlight some explainable AI frameworks that anyone can start using to interpet a machine learning model.

INNvestigate Neural networks

INNvestigate Neural networks [2] is a python package that has implemented a large variety of visual explanation methods such as LRP, CAM and PatternNet. The library contains examples with explanations of state-of-the-art models and is easy to use. The core and base functions of this framework allow rapid implementation of other methods.

explAIner

explAIner [24] is a unified framework that helps users to understand machine and deep learning models. In addition, the framework contains tools to analyze models using different explainable techniques. Then, these explanations are used in order to monitor and guide the optimization process and build better architectures. The explAIner is able to provide interactive graph visualization of a model, performance metrics and integrate high-level explainable methods to interpret it.

InterpetML

InterpetML [16] is an open-source Python library with many interpretability algorithms, which can be very easily integrated into the code. Then, we can easily understand the behavior of any model and compare different interpretation techniques. interpetml Usage of InterpetML framework. Source: [16]

Conclusion

In this article, we presented the major interpretation techniques and categorized them according to the explanation form. Some methods focus on providing visual explanations in the form of images or plots, while others provide textual or numerical explanations. Then, we described some of the latest explainable applications that are developed in demanding tasks like medical diagnosis and autonomous driving. Finally, we provided some well-known XAI frameworks that can be easily used by researchers for their algorithms.

Cited as:

@article{papastratis2021xai,
title = "Introduction to Explainable Artificial Intelligence (XAI)",
author = "Papastratis, Ilias",
journal = "https://theaisummer.com/",
year = "2021",
url = "https://theaisummer.com/xai/"
}

References

[1] Guillaume Alain and Yoshua Bengio.Understanding intermediate layers using linear classifier probes.arXiv preprintarXiv:1610.01644, 2016.

[2] Maximilian Alber, Sebastian Lapuschkin, Philipp Seegerer, MiriamHägele, Kristof T Schütt, Grégoire Montavon, Wojciech Samek,Klaus-Robert Müller, Sven Dähne, and Pieter-Jan Kindermans. iNNvestigate neural networks! J. Mach. Learn. Res., 20(93):1–8, 2019.

[3] Mathieu Aubry and Bryan C Russell. Understanding deep features with computer-generated imagery. InProceedings of the IEEE International Conference on Computer Vision, pages 2875–2883, 2015.

[4] Shane Barratt. Interpnet: Neural introspection for interpretable deeplearning.arXiv preprint arXiv:1710.09511, 2017.

[5] Luca Brunese, Francesco Mercaldo, Alfonso Reginelli, and Antonella Santone. Explainable deep learning for pulmonary disease and coronavirus covid-19 detection from x-rays. Computer Methods and Programs in Biomedicine, 196:105608, 2020.

[6] A Das, S Kottur, K Gupta, A Singh, D Yadav, S Lee, J Moura,D Parikh, and D Batra. Visual dialog. IEEE transactions on pat-tern analysis and machine intelligence, 2018.

[7] Yinpeng Dong, Hang Su, Jun Zhu, and Bo Zhang. Improving interpretability of deep neural networks with semantic information. InProceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pages 4306–4314, 2017.27

[8] Andrej Karpathy, Justin Johnson, and Li Fei-Fei. Visualizing and understanding recurrent networks. 2016.

[9] Jinkyu Kim, Suhong Moon, Anna Rohrbach, Trevor Darrell, andJohn Canny. Advisable learning for self-driving vehicles by internal-izing observation-to-action rules. InProceedings of the IEEE/CVFConference on Computer Vision and Pattern Recognition, pages 9661–9670, 2020.

[10] Been Kim, Martin Wattenberg, Justin Gilmer, Carrie Cai, JamesWexler, Fernanda Viegas, et al. Interpretability beyond feature at-tribution: Quantitative testing with concept activation vectors (tcav).InInternational conference on machine learning, pages 2668–2677.PMLR, 2018.

[11] Devinder Kumar, Alexander Wong, and Graham W Taylor. Explaining the unexplained: A class-enhanced attentive response (clear) ap-proach to understanding deep neural networks. InProceedings of the IEEE Conference on Computer Vision and Pattern RecognitionWorkshops, pages 36–44, 2017.

[12] Ge Liu and David Gifford. Visualizing feature maps in deep neural networks using deepresolve a genomics case study. InICML Visual-ization Workshop, 2017.

[13] Liangzhi Li, Bowen Wang, Manisha Verma, Yuta Nakashima, Ryo Kawasaki, and Hajime Nagahara. Scouter: Slot attention-based classifier for explainable image recognition. arXiv preprintarXiv:2009.06138, 2020.

[14] Jiasen Lu, Jianwei Yang, Dhruv Batra, and Devi Parikh. Hierarchical question-image co-attention for visual question answering. Advances in neural information processing systems, 29:289–297, 2016.

[15] Laurens van der Maaten and Geoffrey Hinton. Visualizing data using t-sne. Journal of machine learning research, 9(Nov):2579–2605,2008.

[16] Harsha Nori, Samuel Jenkins, Paul Koch, and Rich Caruana. Interpretml: A unified framework for machine learning interpretability.arXiv preprint arXiv:1909.09223, 2019.

[19] José Antonio Oramas Mogrovejo, Kaili Wang, and Tinne Tuyte-laars. Visual explanation by interpretation: Improving visual feedback capabilities of deep neural networks. In https://iclr.cc/Conferences/2019/AcceptedPapersInitial. openReview, 2019.

[20] Paulo E Rauber, Samuel G Fadel, Alexandre X Falcao, and Alexan-dru C Telea. Visualizing the hidden activity of artificial neural networks. IEEE Transactions on Visualization and Computer Graphics,23(1):101–110, 2017.

[21] Marco Tulio Ribeiro, Sameer Singh, and Carlos Guestrin. " Why should i trust you?" explaining the predictions of any classifier. InProceedings of the 22nd ACM SIGKDD international conference on knowledge discovery and data mining, pages 1135–1144, 2016.

[22] Ramprasaath R Selvaraju, Michael Cogswell, Abhishek Das, Ra-makrishna Vedantam, Devi Parikh, and Dhruv Batra. Grad-cam:Visual explanations from deep networks via gradient-based localization. InProceedings of the IEEE international conference on com-puter vision, pages 618–626, 2017.

[23] Wojciech Samek, Grégoire Montavon, Alexander Binder, Sebastian Lapuschkin, and Klaus-Robert Müller. Interpreting the predictions of complex ml models by layer-wise relevance propagation. arXivpreprint arXiv:1611.08191, 2016.

[24] Thilo Spinner, Udo Schlegel, Hanna Schäfer, and Mennatallah El-Assady. explainer: A visual analytics framework for interactive and explainable machine learning. IEEE transactions on visualization and computer graphics, 26(1):1064–1074, 2019.

[25] Jayaraman J Thiagarajan, Bhavya Kailkhura, Prasanna Sattigeri,and Karthikeyan Natesan Ramamurthy. Treeview: Peeking into deep neural networks via feature-space partitioning. arXiv preprintarXiv:1611.07429, 2016.

[26] Yiran Xu, Xiaoyin Yang, Lihang Gong, Hsuan-Chu Lin, Tz-YingWu, Yunsheng Li, and Nuno Vasconcelos. Explainable object-induced action decision for autonomous vehicles. InProceedings of 30BIBLIOGRAPHY the IEEE/CVF Conference on Computer Vision and Pattern Recog-nition, pages 9523–9532, 2020.

[27] Matthew D Zeiler and Rob Fergus. Visualizing and understanding convolutional networks. In European conference on computer vi-sion, pages 818–833. Springer, 2014.

[28] Bolei Zhou, Aditya Khosla, Agata Lapedriza, Aude Oliva, and Anto-nio Torralba. Learning deep features for discriminative localization. In Proceedings of the IEEE conference on computer vision and pat-tern recognition, pages 2921–2929, 2016.

[29] Yanzhao Zhou, Yi Zhu, Qixiang Ye, Qiang Qiu, and Jianbin Jiao. Weakly supervised instance segmentation using class peak response. InProceedings of the IEEE Conference on Computer Vision and Pat-tern Recognition, pages 3791–3800, 2018.

Introduction to Deep Learning & Neural Networks

For a more comprehensive understanding of the fundamental archutectures of Deep Learning, check out our interactive course.

You will learn the basics behind CNNs, LSTMs, Autoencoders, GANs, Transformers and Graph Neural Networks using Pytorch in a 100% text-based way.

Learn more

* Disclosure: Please note that some of the links above might be affiliate links, and at no additional cost to you, we will earn a commission if you decide to make a purchase after clicking through.