Skip to content

Commit

Permalink
📚 Add ConflictualLoss reference in documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
alafage committed Sep 4, 2024
1 parent 2dee70b commit d47f0f3
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 3 deletions.
13 changes: 13 additions & 0 deletions docs/source/references.rst
Original file line number Diff line number Diff line change
Expand Up @@ -243,6 +243,19 @@ For Laplace Approximation, consider citing:
* Authors: *Erik Daxberger, Agustinus Kristiadi, Alexander Immer, Runa Eschenhagen, Matthias Bauer, and Philipp Hennig*
* Paper: `NeurIPS 2021 <https://arxiv.org/abs/2106.14806>`__.

Losses
------

Conflictual Loss
^^^^^^^^^^^^^^^^

For the conflictual loss, consider citing:

**On the Calibration of Epistemic Uncertainty: Principles, Paradoxes and Conflictual Loss**

* Authors: *Mohammed Fellaji, Frédéric Pennerath, Brieuc Conan-Guez, and Miguel Couceiro*
* Paper: `ArXiv 2024 <https://arxiv.org/pdf/2407.12211`__.

Metrics
-------

Expand Down
6 changes: 3 additions & 3 deletions torch_uncertainty/losses/classification.py
Original file line number Diff line number Diff line change
Expand Up @@ -235,16 +235,16 @@ def __init__(
reg_weight: float = 1,
reduction: str | None = "mean",
) -> None:
"""The Conflictual Loss.
r"""The Conflictual Loss.
Args:
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. https://arxiv.org/pdf/2407.12211
`Mohammed Fellaji et al. On the Calibration of Epistemic Uncertainty:
Principles, Paradoxes and Conflictual Loss <https://arxiv.org/pdf/2407.12211>`_.
"""
super().__init__()
if reduction is None:
Expand Down

0 comments on commit d47f0f3

Please sign in to comment.