Losses registry#
Loss functions assembled by build_criteria(). See Add a model or loss.
Build any of these from a config with criteria=[dict(type=...)], using the type string in the first column. Generated from the live registry - 14 classes (14 names including aliases), grouped by role.
Classification#
Cross-entropy and focal variants for class logits.
Name |
Summary |
|---|---|
Binary focal loss for class-imbalanced binary targets. |
|
Cross-entropy for an extra categorical per-query head. |
|
Standard multi-class cross-entropy over per-point class logits. |
|
Multi-class focal loss over per-point class logits. |
|
Label-smoothed cross-entropy with NaN-robust averaging. |
Binary focal loss for class-imbalanced binary targets. |
|
Cross-entropy for an extra categorical per-query head. |
|
Standard multi-class cross-entropy over per-point class logits. |
|
Multi-class focal loss over per-point class logits. |
|
Label-smoothed cross-entropy with NaN-robust averaging. |
Segmentation#
Region-overlap losses for semantic segmentation.
Name |
Summary |
|---|---|
Soft multi-class Dice loss over per-point class probabilities. |
|
Lovasz loss: a direct surrogate for the IoU (Jaccard) metric. |
Soft multi-class Dice loss over per-point class probabilities. |
|
Lovasz loss: a direct surrogate for the IoU (Jaccard) metric. |
Regression#
Continuous-target losses (momentum, vertex, …).
Name |
Summary |
|---|---|
Mean-absolute-error (L1) regression loss for continuous targets. |
|
Mean-squared-error (L2) regression loss for continuous targets. |
|
Smooth-L1 (Huber) regression loss for continuous targets. |
Mean-absolute-error (L1) regression loss for continuous targets. |
|
Mean-squared-error (L2) regression loss for continuous targets. |
|
Smooth-L1 (Huber) regression loss for continuous targets. |
Instance segmentation#
Set-based mask + class (+ regression) losses with Hungarian matching.
Name |
Summary |
|---|---|
Mask2Former-style instance segmentation loss (cached, vectorized). |
|
Instance segmentation loss plus configurable per-instance regressions. |
|
Combine instance mask/class losses with configured query-head losses. |
|
DEPRECATED alias for |
|
Mask2Former-style instance segmentation loss (cached, vectorized). |
|
Instance segmentation loss plus configurable per-instance regressions. |
|
Combine instance mask/class losses with configured query-head losses. |
DEPRECATED alias for |