probly.train.evidential.torch.evidential_ce_loss¶
- probly.train.evidential.torch.evidential_ce_loss(alphas: Tensor, targets: Tensor) Tensor[source]¶
Evidential Cross Entropy Loss for classification uncertainty estimation.
Implements the evidential cross-entropy loss proposed by Sensoy et al. (2018) for Evidential Deep Learning.
- Reference:
Sensoy et al., “Evidential Deep Learning to Quantify Classification Uncertainty”, NeurIPS 2018. https://arxiv.org/abs/1806.01768
- 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.