Source code for detectors.methods.msp

import torch
from torch import Tensor


[docs]@torch.no_grad()
def msp(input: Tensor, model: torch.nn.Module, **kwargs) -> Tensor:
    """Maximum Softmax Response OOD detector.

    Args:
        input (Tensor): input tensor.
        model (nn.Module): classifier.

    Returns:
        Tensor: OOD scores for each input.

    References:
        [1] https://arxiv.org/abs/1610.02136
    """
    model.eval()
    logits = model(input)
    probs = torch.softmax(logits, dim=1)
    return torch.max(probs, dim=1)[0]