from captum.log import log_usage
from captum._utils.common import _select_targets
from captum._utils.typing import (
BaselineType,
TargetType,
TensorOrTupleOfTensorsGeneric,
)
from torch import Tensor
from typing import Any, Callable, Tuple, Union
from .base import _base_metric
def _mse(
logits_original: Tensor, logits_pert: Tensor, target: Tensor
) -> Tensor:
return (
(
_select_targets(logits_original, target).reshape(
len(logits_original), -1
)
- _select_targets(logits_pert, target).reshape(
len(logits_pert), -1
)
)
** 2
).sum(-1, keepdim=True)
[docs]@log_usage()
def mse(
forward_func: Callable,
inputs: TensorOrTupleOfTensorsGeneric,
attributions: TensorOrTupleOfTensorsGeneric,
baselines: BaselineType = None,
additional_forward_args: Any = None,
target: TargetType = None,
n_samples: int = 1,
n_samples_batch_size: int = None,
stdevs: Union[float, Tuple[float, ...]] = 0.0,
draw_baseline_from_distrib: bool = False,
topk: float = 0.2,
mask_largest: bool = True,
weight_fn: Callable[
[Tuple[Tensor, ...], Tuple[Tensor, ...]], Tensor
] = None,
) -> float:
"""
Mean square error.
This metric measures the mean square error between the outputs of
the model using the original inputs and perturbed inputs by removing
or only keeping the topk most important features. Lower is better.
Args:
forward_func (callable): The forward function of the model or any
modification of it.
inputs (tensor or tuple of tensors): Input for which occlusion
attributions are computed. If forward_func takes a single
tensor as input, a single input tensor should be provided.
If forward_func takes multiple tensors as input, a tuple
of the input tensors should be provided. It is assumed
that for all given input tensors, dimension 0 corresponds
to the number of examples (aka batch size), and if
multiple input tensors are provided, the examples must
be aligned appropriately.
attributions (tensor or tuple of tensors):
The attributions with respect to each input feature.
Attributions will always be
the same size as the provided inputs, with each value
providing the attribution of the corresponding input index.
If a single tensor is provided as inputs, a single tensor
is returned. If a tuple is provided for inputs, a tuple of
corresponding sized tensors is returned.
baselines (scalar, tensor, tuple of scalars or tensors, optional):
Baselines define the starting point from which integral
is computed and can be provided as:
- a single tensor, if inputs is a single tensor, with
exactly the same dimensions as inputs or the first
dimension is one and the remaining dimensions match
with inputs.
- a single scalar, if inputs is a single tensor, which will
be broadcasted for each input value in input tensor.
- a tuple of tensors or scalars, the baseline corresponding
to each tensor in the inputs' tuple can be:
- either a tensor with matching dimensions to
corresponding tensor in the inputs' tuple
or the first dimension is one and the remaining
dimensions match with the corresponding
input tensor.
- or a scalar, corresponding to a tensor in the
inputs' tuple. This scalar value is broadcasted
for corresponding input tensor.
In the cases when `baselines` is not provided, we internally
use zero scalar corresponding to each input tensor.
Default: None
additional_forward_args (any, optional): If the forward function
requires additional arguments other than the inputs for
which attributions should not be computed, this argument
can be provided. It must be either a single additional
argument of a Tensor or arbitrary (non-tuple) type or a
tuple containing multiple additional arguments including
tensors or any arbitrary python types. These arguments
are provided to forward_func in order following the
arguments in inputs.
For a tensor, the first dimension of the tensor must
correspond to the number of examples. It will be
repeated for each of `n_steps` along the integrated
path. For all other types, the given argument is used
for all forward evaluations.
Note that attributions are not computed with respect
to these arguments.
Default: None
target (int, tuple, tensor or list, optional): Output indices for
which gradients are computed (for classification cases,
this is usually the target class).
If the network returns a scalar value per example,
no target index is necessary.
For general 2D outputs, targets can be either:
- a single integer or a tensor containing a single
integer, which is applied to all input examples
- a list of integers or a 1D tensor, with length matching
the number of examples in inputs (dim 0). Each integer
is applied as the target for the corresponding example.
For outputs with > 2 dimensions, targets can be either:
- A single tuple, which contains #output_dims - 1
elements. This target index is applied to all examples.
- A list of tuples with length equal to the number of
examples in inputs (dim 0), and each tuple containing
#output_dims - 1 elements. Each tuple is applied as the
target for the corresponding example.
Default: None
n_samples (int, optional): The number of randomly generated examples
per sample in the input batch. Random examples are
generated by adding gaussian random noise to each sample.
Default: 1
n_samples_batch_size (int, optional): The number of the `n_samples`
that will be processed together. With the help
of this parameter we can avoid out of memory situation and
reduce the number of randomly generated examples per sample
in each batch.
Default: None if `n_samples_batch_size` is not provided. In
this case all `n_samples` will be processed together.
stdevs (float, or a tuple of floats optional): The standard deviation
of gaussian noise with zero mean that is added to each
input in the batch. If `stdevs` is a single float value
then that same value is used for all inputs. If it is
a tuple, then it must have the same length as the inputs
tuple. In this case, each stdev value in the stdevs tuple
corresponds to the input with the same index in the inputs
tuple.
Default: 0.0
draw_baseline_from_distrib (bool, optional): Indicates whether to
randomly draw baseline samples from the `baselines`
distribution provided as an input tensor.
Default: False
topk: Proportion of input to be dropped. Must be between 0 and 1.
Default: 0.2
mask_largest: Whether to mask the topk attribution or to only keep
the topk attribution.
Default: True
weight_fn (Callable): Function to compute metrics weighting using
original inputs and pertubed inputs. None if note provided.
Default: None
Returns:
(float): The mean square error metric.
Examples:
>>> import torch as th
>>> from captum.attr import Saliency
>>> from tint.metrics import mse
>>> from tint.models import MLP
<BLANKLINE>
>>> inputs = th.rand(8, 7, 5)
>>> mlp = MLP([5, 3, 1])
<BLANKLINE>
>>> explainer = Saliency(mlp)
>>> attr = explainer.attribute(inputs, target=0)
<BLANKLINE>
>>> mae_ = mse(mlp, inputs, attr, target=0)
"""
return _base_metric(
metric=_mse,
forward_func=forward_func,
inputs=inputs,
attributions=attributions,
baselines=baselines,
additional_forward_args=additional_forward_args,
target=target,
n_samples=n_samples,
n_samples_batch_size=n_samples_batch_size,
stdevs=stdevs,
draw_baseline_from_distrib=draw_baseline_from_distrib,
topk=topk,
largest=mask_largest,
weight_fn=weight_fn,
classification=False,
)