DECLoss

class torch_uncertainty.losses.DECLoss(annealing_step=None, reg_weight=None, loss_type='log', reduction='mean')

The Deep Evidential Classification loss.

Parameters:

• annealing_step (int | None) – Annealing step for the weight of the regularization term. Defaults to None.

• reg_weight (float | None) – Fixed weight of the regularization term. Defaults to None.

• loss_type (str) – Specifies the loss type to apply to the Dirichlet parameters: 'mse' | 'log' | 'digamma'.

• reduction (str | None) – Specifies the reduction to apply to the output:'none' | 'mean' | 'sum'.

References

[1] Sensoy, M., Kaplan, L., & Kandemir, M. (2018). Evidential deep learning to quantify classification uncertainty. NeurIPS 2018.