torch

Torch training losses.

Functions

cvar_ce_loss	CVaR cross-entropy loss from [WFC+26].
der_loss	Deep Evidential Regression loss from [ASSR20].
disc_vbll_loss	Negative discriminative VBLL ELBO from [HWS24].
elbo_loss	Evidence lower bound loss based on [BCKW15].
evidential_ce_loss	Evidential cross-entropy loss from [SKK18].
evidential_kl_divergence	Evidential KL divergence regularizer from [SKK18].
evidential_log_loss	Evidential log loss from [SKK18].
evidential_mse_loss	Evidential mean squared error loss from [SKK18].
evidential_nignll_loss	Evidential Normal-Inverse-Gamma regression loss from [ASSR20].
evidential_regression_regularization	Evidential regression regularizer from [ASSR20].
focal_loss	Focal Loss based on [LGG+17].
g_vbll_loss	Negative generative VBLL ELBO from [HWS24].
het_vbll_loss	Negative heteroscedastic VBLL ELBO from [HWS24].
intersection_probability_ce_loss	Intersection-probability cross-entropy loss from [WCM+24].
ird_loss	Information Robust Dirichlet loss from [Tsi19].
label_relaxation_loss	Label Relaxation Loss from [LHullermeier21].
lop_gpn_loss	LOP-GPN loss based on [DHullermeier24a].
lp_fn	Lp calibration loss from [Tsi19].
make_in_domain_target_alpha	Construct in-domain Dirichlet targets based on [MG18].
make_ood_target_alpha	Construct OOD Dirichlet targets based on [MG18].
mixture_uce_loss	LOP-GPN mixture uncertainty cross-entropy loss from [DHullermeier24a].
natpn_loss	Natural Posterior Network loss from [CBZugner+22].
normal_wishart_log_prob	Simplified Normal-Wishart log-likelihood based on [MCPG20].
pn_loss	Dirichlet Prior Network loss based on [MG18].
postnet_loss	Posterior Network classification loss from [CZugnerGunnemann20].
regularization_fn	Information Robust Dirichlet regularizer from [Tsi19].
rpn_distillation_loss	Regression Prior Network distillation loss based on [MCPG20].
rpn_loss	Paired ID/OOD Regression Prior Network loss based on [MCPG20].
rpn_ng_kl	Normal-Gamma KL divergence for Regression Prior Networks from [MCPG20].
rpn_prior	Normal-Gamma prior for Regression Prior Networks from [MCPG20].
t_vbll_loss	Negative Student-t VBLL ELBO from [HWS24].