lop_gpn_loss

probly.train.evidential.torch.lop_gpn_loss(alpha_features: Tensor, mixture_weights: Tensor, y: Tensor, entropy_regularization: Tensor | None = None, entropy_weight: float = 0.0, reduction: str = 'sum') → Tensor[source]

Compute a simple LOP-GPN loss from mixture UCE and optional entropy regularization.

Parameters:

  • alpha_features – Feature-level Dirichlet concentration parameters with shape (N, C).

  • mixture_weights – Dense mixture weights with shape (B, N).

  • y – Ground-truth labels for the mixed nodes with shape (B,).

  • entropy_regularization – Optional per-sample entropy regularizer.

  • entropy_weight – Weight applied to entropy_regularization.

  • reduction – Reduction to apply, either "mean", "sum", or "none".

Returns:

Scalar or per-sample LOP-GPN loss.