ConflictualLoss

class torch_uncertainty.losses.ConflictualLoss(reg_weight=1, reduction='mean')

The Conflictual Loss.

Parameters:

• reg_weight (float, optional) – The weight of the regularization term.

• reduction (str, optional) – specifies the reduction to apply to the

• output – 'none' | 'mean' | 'sum'.

Reference:

Mohammed Fellaji et al. On the Calibration of Epistemic Uncertainty: Principles, Paradoxes and Conflictual Loss.

forward(logits, targets)

Compute the conflictual loss.

Parameters:

• logits (Tensor) – The outputs of the model.

• targets (Tensor) – The target values.

Returns:

The conflictual loss.

Return type:

Tensor