evidential_log_loss

probly.losses.torch.evidential_log_loss(alphas: Tensor, targets: Tensor) Tensor[source]

Evidential log loss from [SKK18].

Implements the evidential log loss for classification uncertainty estimation in Evidential Deep Learning.

Parameters:
  • alphas – Dirichlet concentration parameters, shape (B, C).

  • targets – Ground-truth class labels, shape (B,).

Returns:

Scalar evidential log loss averaged over the batch.

Examples using probly.losses.torch.evidential_log_loss

Evidential on MNIST

Evidential on MNIST