evidential_ce_loss

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

Evidential cross-entropy loss from [SKK18].

Implements the evidential cross-entropy 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 cross-entropy loss averaged over the batch.