probly.train.evidential.torch.evidential_mse_loss¶

probly.train.evidential.torch.evidential_mse_loss(alphas: Tensor, targets: Tensor) → Tensor[source]¶

Evidential Mean Squared Error loss for classification uncertainty estimation.

Implements the evidential MSE loss proposed by Sensoy et al. (2018), combining prediction error and predictive variance under a Dirichlet distribution.

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 mean squared error loss averaged over the batch.