Skip to content

Commit 5384817

Browse files
committed
Add average calibration losses
Signed-off-by: Theo Barfoot <theo.barfoot@gmail.com>
1 parent 9ea04d4 commit 5384817

6 files changed

Lines changed: 672 additions & 6 deletions

File tree

docs/source/losses.rst

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,16 @@ Segmentation Losses
2222
.. autoclass:: dice
2323
:members:
2424

25+
`HardL1ACELoss`
26+
~~~~~~~~~~~~~~~~
27+
.. autoclass:: HardL1ACELoss
28+
:members:
29+
30+
`SoftL1ACELoss`
31+
~~~~~~~~~~~~~~~~
32+
.. autoclass:: SoftL1ACELoss
33+
:members:
34+
2535
`MaskedDiceLoss`
2636
~~~~~~~~~~~~~~~~
2737
.. autoclass:: MaskedDiceLoss

monai/handlers/calibration.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,8 +59,8 @@ class CalibrationError(IgniteMetricHandler):
5959
- Guo, C., et al. "On Calibration of Modern Neural Networks." ICML 2017.
6060
https://proceedings.mlr.press/v70/guo17a.html
6161
- Barfoot, T., et al. "Average Calibration Losses for Reliable Uncertainty in
62-
Medical Image Segmentation." arXiv:2506.03942v3, 2025.
63-
https://arxiv.org/abs/2506.03942v3
62+
Medical Image Segmentation." IEEE Transactions on Medical Imaging, 2026.
63+
https://doi.org/10.1109/TMI.2026.3673118
6464
6565
See Also:
6666
- :py:class:`~monai.metrics.CalibrationErrorMetric`: The underlying metric class.

monai/losses/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from .aucm_loss import AUCMLoss
1616
from .barlow_twins import BarlowTwinsLoss
1717
from .boundary_loss import BoundaryLoss
18+
from .calibration import HardL1ACELoss, SoftL1ACELoss
1819
from .cldice import SoftclDiceLoss, SoftDiceclDiceLoss
1920
from .contrastive import ContrastiveLoss
2021
from .deform import BendingEnergyLoss, DiffusionLoss

0 commit comments

Comments
 (0)