In the last decade, it has been demonstrated in an impressive way how efficiently and successfully neural networks can analyze and understand enormous amounts of data. They can recognize patterns and associations and transfer this knowledge to new data points with remarkable accuracy. Moreover, their flexibility eliminates the feature engineering step that was often necessary before and allows them to work directly with raw data. Nevertheless, these associations and internal findings are hidden somewhere in the black box and it is unclear to the user what the crucial aspects of the prediction are. One way to open the black box is through so-called feature attribution methods. These are local methods that — based on a single data point (image, tabular instance,…) — assign a relevance of a previously defined output class or node to each input variable. In general, only a normal forward pass and a method-specific backward pass are required, making the implementation much faster compared to perturbation- or optimization-based methods like LIME or Occlusion. Figure 1 illustrates the basic approach of the feature attribution methods.
Of course, we are not the first to provide several feature attribution methods for neural networks in one package. For example, there are several packages for Python, such as iNNvestigate, captum and zennit. Due to the great and extremely efficient deep learning libraries Keras/TensorFlow and PyTorch, it is only reasonable that these are all Python-exclusive. However, in recent years these libraries have been integrated more and more successfully into the R programming language. We fill this lack of feature attribution methods for neural networks in R with our package innsight.
In addition to the availability in R, the package is also outstanding for the following aspects:
Deep-learning-library-agnostic: To be as flexible as
possible and available to a range of users, we do not limit ourselves to
models from a particular deep learning library, as is the case with all
Python variants. Using the Converter
, each passed model
(from keras, torch or
neuralnet) is first converted into a list with all
relevant information about the model. Then, a
torch-model is created from this list, which has the
available feature attribution methods pre-implemented for each layer. If
our package does not support your favorite library, there is also the
option to do the converting step by yourself and pass a list
directly.
No Python dependency: In R, there are currently two major deep learning libraries, namely keras/tensorflow and torch. However, keras/tensorflow, accesses the corresponding Python methods via the package reticulate. We use the fast and efficient torch package for all computations, which runs without Python and directly accesses the C++ variant of PyTorch called LibTorch (see Fig. 2).
Unified framework: It does not matter which model and method you choose, it is always the same three steps that lead to a visual illustration of the results (see the next section for details):
Converter
$\xrightarrow{\text{Step 2}}$ method $\xrightarrow{\text{Step 3}}$
plot()
or plot_global()
/boxplot()
Visualization tools: Our package innsight offers several visualization methods for individual or summarized results regardless of whether it is tabular, 1D signal, 2D image data or a mix of these. Additionally, interactive plots can be created based on the plotly package.
The following is more of a high-level overview that only explains
some of the details of the three steps. In case you are looking for a
more detailed overview of all configuration options, we refer you to the
vignette
“In-depth explanation” (same as
vignette("detailed_overview", package = "innsight")
). The
three steps for explaining individual predictions with the provided
methods are unified in this package and follow a strict scheme. This
will hopefully allow any user a smooth and easy introduction to the
possibilities of this package. The steps are:
# Step 0: Model creation
model <- ... # this step is left to the user
# Step 1: Convert the model
converter <- convert(model)
converter <- Converter$new(model) # the same but without helper function
# Step 2: Apply selected method to your data
result <- run_method(converter, data)
result <- Method$new(converter, data) # the same but without helper function
# Step 3: Show and plot the results
get_result(result) # get the result as an `array`, `data.frame` or `torch_tensor`
plot(result) # for individual results (local)
plot_global(result) # for summarized results (global)
boxplot(result) # alias for `plot_global` for tabular and signal data
The innsight package aims to be as flexible as
possible and independent of any particular deep learning package in
which the passed network was learned or defined. For this reason, there
are several ways in this package to pass a neural network to the
Converter
object, but the call is always the same:
# Using the helper function `convert`
converter <- convert(model, ...)
# It simply passes all arguments to the initialization function of
# the corresponding R6 class, i.e., it is equivalent to
converter <- Converter$new(model, ...)
Except for a neuralnet model, no names of inputs or
outputs are stored in the given model. If no further arguments are set
for the Converter
instance or convert()
function, default labels are generated for the input
(e.g. 'X1'
, 'X2'
, …) and output names
('Y1'
, 'Y2'
, … ). In the converter, however,
there is the possibility with the optional arguments
input_names
and output_names
to pass the
names, which will then be used in all results and plots created by this
object.
Currently, only models created by torch::nn_sequential
are accepted. However, the most popular standard layers and activation
functions are available (see the detailed
vignette for details).
📝 Note
If you want to create an instance of the classConverter
with a torch model that meets the above conditions, you have to specify the shape of the inputs with the argumentinput_dim
because this information is not stored in every given torch model.
library(torch)
library(innsight)
torch_manual_seed(123)
# Create model
model <- nn_sequential(
nn_linear(3, 10),
nn_relu(),
nn_linear(10, 2, bias = FALSE),
nn_softmax(2)
)
# Convert the model
conv_dense <- convert(model, input_dim = c(3))
# Convert model with input and output names
conv_dense_with_names <-
convert(model,
input_dim = c(3),
input_names = list(c("Price", "Weight", "Height")),
output_names = list(c("Buy it!", "Don't buy it!"))
)
Models created by keras_model_sequential
or keras_model
with the keras package are accepted. Within these
functions, the most popular layers and activation functions are accepted
(see the in-depth
vignette for details).
library(keras)
# Create model
model <- keras_model_sequential()
model <- model %>%
layer_conv_2d(4, c(5, 4), input_shape = c(10, 10, 3), activation = "softplus") %>%
layer_max_pooling_2d(c(2, 2), strides = c(1, 1)) %>%
layer_conv_2d(6, c(3, 3), activation = "relu", padding = "same") %>%
layer_max_pooling_2d(c(2, 2)) %>%
layer_conv_2d(4, c(2, 2), strides = c(2, 1), activation = "relu") %>%
layer_flatten() %>%
layer_dense(5, activation = "softmax")
# Convert the model
conv_cnn <- convert(model)
The usage with nets from the package neuralnet is
very simple and straightforward, because the package offers much fewer
options than torch or keras. The only
thing to note is that no custom activation function can be used.
However, the package saves the names of the inputs and outputs, which
can, of course, be overwritten with the arguments
input_names
and output_names
when creating the
converter object.
If you have not trained your net with keras, torch or neuralnet, you can also pass your model as a list, i.e., you write your own wrapper for your library. But you have to consider a few points, which are explained in detail in the in-depth vignette.
model <- list(
input_dim = 2,
input_names = list(c("X1", "Feat2")),
input_nodes = 1,
output_nodes = 2,
layers = list(
list(
type = "Dense", weight = matrix(rnorm(10), 5, 2), bias = rnorm(5),
activation_name = "relu", input_layers = 0, output_layers = 2
),
list(
type = "Dense", weight = matrix(rnorm(5), 1, 5), bias = rnorm(1),
activation_name = "sigmoid", input_layers = 1, output_layers = -1
)
)
)
converter <- convert(model)
After an instance of the Converter
class has been
created, the base print()
method can be used to output the
most important components of the object in summary form:
The innsight package provides several tools for
analyzing black box neural networks based on dense or convolution
layers. For the sake of uniform usage, all implemented methods inherit
from the InterpretingMethod
super class (see
?InterpretingMethod
for details) and differ in each case
only by method-specific arguments and settings. Therefore, each method
has the following initialization structure:
method <- Method$new(converter, data, # required arguments
channels_first = TRUE, # optional settings
output_idx = NULL, # .
ignore_last_act = TRUE, # .
... # other args and method-specific args
)
However, you can also use the helper functions (e.g.,
run_grad()
, run_deeplift()
, etc.) for
initializing a new object:
method <- run_method(converter, data, # required arguments
channels_first = TRUE, # optional settings
output_idx = NULL, # .
ignore_last_act = TRUE, # .
... # other args and method-specific args
)
The most important arguments are explained below. For a complete and
detailed explanation, however, we refer to the R documentation (see
?InterpretingMethod
) or the vignette “In-depth
explanation”
(vignette("detailed_overview", package = "innsight")
).
converter
: This is the converter object created in
the first
step.
data
: The data to which the method is to be applied.
These must have the same format as the input data of the passed model to
the converter object. This means either an array
,
data.frame
, torch_tensor
or array-like format
of size (batchsize, input_dim), if
e.g., the model has only one input layer, or a list
of the
respective input sizes for each of the input layers.
channels_first
: The channel position of the given
data (argument data
). If TRUE
, the channel
axis is placed at the second position between the batch size and the
remaining input axes, e.g., c(10,3,32,32)
for a batch of
ten images with three channels and a height and width of 32 pixels.
Otherwise (FALSE
), the channel axis is at the last
position, i.e., c(10,32,32,3)
. If the data has no channel
axis, use the default value TRUE
.
output_idx
: These indices specify the output nodes
or classes for which the method is to be applied. If the model has only
one output layer, the values correspond to the indices of the output
nodes, e.g., c(1,3,4)
for the first, third and fourth
output node. If your model has more than one output layer, you can pass
the respective output nodes in a list which is described in detail in
the R documentation (see ?InterpretingMethod
) or in the in-depth
vignette
output_label
: These values specify the output nodes
for which the method is to be applied and can be used as an alternative
to the argument output_idx
. Only values that were
previously passed with the argument output_names
in the
converter
can be used.
ignore_last_act
: Set this logical value to include
the last activation functions for each output layer, or not (default:
TRUE
)
The package innsight now offers the following
methods for interpreting your model. To use them, simply replace the
name "Method"
with one of the method’s names below. There
are also method-specific arguments, but these are explained in detail
along with the methods in the R documentation (e.g.,
?Gradient
or ?LRP
) or in the in-depth
vignette. Let x ∈ ℝp the input
instance, i is the feature
index of the input and c the
index of the output node or class to be explained:
Gradient
: Calculation of the model
output Gradients with respect to the model inputs including the
attribution method Gradient×Input:
# Apply method 'Gradient' for the dense network
grad_dense <- Gradient$new(conv_dense, iris[-c(1, 2, 5)])
# You can also use the helper function `run_grad`
grad_dense <- run_grad(conv_dense, iris[-c(1, 2, 5)])
# Apply method 'Gradient x Input' for CNN
x <- torch_randn(c(10, 3, 10, 10))
grad_cnn <- run_grad(conv_cnn, x, times_input = TRUE)
SmoothGrad
: Calculation of the
smoothed model output gradients (SmoothGrad) with
respect to the model inputs by averaging the gradients over number of
inputs with added noise (including SmoothGrad×Input): $$
\text{SmoothGrad}(x)_i^c = \mathbb{E}_{\varepsilon \sim \mathcal{N}(0,
\sigma^2)}\left[\frac{\partial f(x + \varepsilon)_c}{\partial (x +
\varepsilon)_i}\right] \approx \frac{1}{n} \sum_{k=1}^n \frac{\partial
f(x + \varepsilon_k)_c}{\partial (x + \varepsilon_k)_i}
$$ with ε1, …εn ∼ 𝒩(0, σ2).
IntegratedGradient
: Calculation of the
integrated gradients (Sundararajan et al. (2017))
with respect to a reference input x̃: $$
\text{IntGrad}(x)_i^c = (x - \tilde{x}) \int_{\alpha = 0}^1
\frac{\partial f(\tilde{x} + \alpha (x - \tilde{x}))}{\partial x}
d\alpha.
$$
# Apply method 'IntegratedGradient' for the dense network
intgrad_dense <- run_intgrad(conv_dense, iris[-c(1, 2, 5)])
# Apply method 'IntegratedGradient' for CNN with the average baseline
x <- torch_randn(c(10, 3, 10, 10))
x_ref <- x$mean(1, keepdim = TRUE)
intgrad_cnn <- run_intgrad(conv_cnn, x, x_ref = x_ref)
ExpectedGradient
: Calculation of the
integrated gradients (Erion et al.,
2021) with respect to a whole reference dataset X̃ ∼ x̃: $$
\text{ExpGrad}(x)_i^c = \mathbb{E}_{\tilde{x}\sim \tilde{X}, \alpha \sim
U(0,1)} \left[(x - \tilde{x}) \times \frac{\partial f(\tilde{x} + \alpha
(x - \tilde{x}))}{\partial x} \right]
$$
# Apply method 'ExpectedGradient' for the dense network
expgrad_dense <- run_expgrad(conv_dense, iris[-c(1, 2, 5)],
data_ref = iris[-c(1, 2, 5)])
# Apply method 'ExpectedGradient' for CNN
x <- torch_randn(c(10, 3, 10, 10))
data_ref <- torch_randn(c(20, 3, 10, 10))
expgrad_cnn <- run_expgrad(conv_cnn, x, data_ref = data_ref)
LRP
: Back-propagating the model output
to the model input neurons to obtain relevance scores for the model
prediction which is known as Layer-wise
Relevance Propagation: $$
f(x)_c \approx \sum_{i=1}^d R_i
$$ with Ri relevance
score for input neuron i.
DeepLift
: Calculation of a
decomposition of the model output with respect to the model inputs and a
reference input which is known as Deep Learning Important
Features (DeepLift): $$
\Delta y_c = f(x)_c - f(x_\text{ref})_c = \sum_{i=1}^d C_i
$$ with Ci contribution
score for input neuron i to
the difference-from-reference model output Δyc.
# Define reference value
x_ref <- array(colMeans(iris[-c(1, 2, 5)]), dim = c(1, 2))
# Apply method 'DeepLift' for the dense network
deeplift_dense <- run_deeplift(conv_dense, iris[-c(1, 2, 5)], x_ref = x_ref)
# Apply method 'DeepLift' for CNN (default is a zero baseline)
x <- torch_randn(c(10, 3, 10, 10))
deeplift_cnn <- run_deeplift(conv_cnn, x)
ConnectionWeights
: This is a naive and
old approach by calculating the product of all weights from an input to
an output neuron and then adding them up (see Connection
Weights).
DeepSHAP
and the
model-agnostic methods LIME
and
SHAP
are implemented (by the functions
run_deepshap()
, run_lime()
and
run_shap()
). For details, we refer to our vignette
“In-depth explanation”.
📝 Notes
By default, the last activation function is not taken into account for all data-based methods. Because often, this is a sigmoid/logistic or softmax function, which has increasingly smaller gradients with a growing distance from 0, which leads to the so-called saturation problem. But if you still want to consider the last activation function, use the argument
ignore_last_act = FALSE
.For data with channels, it is impossible to determine exactly on which axis the channels are located. Internally, all data and the converted model are in the data format “channels first”, i.e., directly after the batch dimension (batchsize, channels, input_dim). In case you want to pass data with “channels last” (e.g., for MNIST-data (batchsize, 28, 28, 3)), you have to indicate that with argument
channels_first
in the applied method.It can happen with very large and deep neural networks that the calculation for all outputs requires the entire memory and takes a very long time. But often, the results are needed only for certain output nodes. By default, only the results for the first 10 outputs are calculated, which can be adjusted individually with the argument
output_idx
by passing the relevant output indices.
Similar to the instances of the Converter
class, the
default print()
function for R6 classes was also overridden
for each method object, so that all important contents of the
corresponding method are displayed:
Once a method object has been created, the results can be returned as
an array
, data.frame
, or
torch_tensor
, and can be further processed as desired. In
addition, for each of the three sizes of the inputs (tabular, 1D signals
or 2D images) suitable plot and boxplot functions based on ggplot2 are
implemented. Due to the complexity of higher dimensional inputs, these
plots and boxplots can also be displayed as an interactive plotly plots by using
the argument as_plotly
.
Each instance of the interpretability methods has the class method
get_result()
, which is used to return the results. You can
choose between the data formats array
,
data.frame
or torch_tensor
by passing the name
as an character for argument type
. This method is also
implemented as a S3 method. For a deeper view in this method look this
section in the in-depth vignette.
# Get the result with the class method
method$get_result(type = "array")
# or use the S3 function
get_result(method, type = "array")
array
(default)
data.frame
The package innsight also provides methods for
visualizing the results. By default a ggplot2-plot is
created, but it can also be rendered as an interactive
plotly plot with the as_plotly
argument.
You can use the argument output_idx
to select the indices
of the output nodes for the plot. In addition, if the results have
channels, the aggr_channels
argument can be used to
determine how the channels are aggregated. All arguments are explained
in detail in the R documentation (see ?InterpretingMethod
)
or here
for plot()
and here
for plot_global()
.
# Create a plot for single data points
plot(method,
data_idx = 1, # the data point to be plotted
output_idx = NULL, # the indices of the output nodes/classes to be plotted
output_label = NULL, # the class labels to be plotted
aggr_channels = "sum",
as_plotly = FALSE, # create an interactive plot
... # other arguments
)
# Create a plot with summarized results
plot_global(method,
output_idx = NULL, # the indices of the output nodes/classes to be plotted
output_label = NULL, # the class labels to be plotted
ref_data_idx = NULL, # the index of an reference data point to be plotted
aggr_channels = "sum",
as_plotly = FALSE, # create an interactive plot
... # other arguments
)
# Alias for `plot_global` for tabular and signal data
boxplot(...)
Examples:📝 Note
The argumentoutput_idx
can be either a vector of indices or a list of vectors of indices but must be a subset of the indices for which the results were calculated, i.e., a subset of the argumentoutput_idx
passed to the respective method previously. By default (NULL
), the smallest index of all computed output nodes and output layers is used.
plot()
function (ggplot2)
# Plot the result of the first data point (default) for the output classes '1', '2' and '3'
plot(smooth_dense, output_idx = 1:3)
# You can plot several data points at once
plot(smooth_dense, data_idx = c(1, 144), output_idx = 1:3)
# Plot result for the first data point and first and fourth output classes
# and aggregate the channels by taking the Euclidean norm
plot(lrp_cnn, aggr_channels = "norm", output_idx = c(1, 4))
plot()
function (plotly)
plot_global()
function (ggplot2)