Skip to content

Commit e2e55a5

Browse files
committed
Add EmbeddingCollapseMetric: detect representational collapse in medical imaging embeddings
Closes #8808 Signed-off-by: ekansh-arora0 <ekansh.arora0@gmail.com>
1 parent c434607 commit e2e55a5

File tree

4 files changed

+904
-0
lines changed

4 files changed

+904
-0
lines changed

docs/source/metrics.rst

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,13 @@ Metrics
158158

159159
`Fréchet Inception Distance`
160160
------------------------------
161+
`Embedding Collapse`
162+
------------------------------
163+
.. autofunction:: compute_embedding_collapse
164+
165+
.. autoclass:: EmbeddingCollapseMetric
166+
:members:
167+
161168
.. autofunction:: compute_frechet_distance
162169

163170
.. autoclass:: FIDMetric

monai/metrics/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from .calibration import CalibrationErrorMetric, CalibrationReduction, calibration_binning
1717
from .confusion_matrix import ConfusionMatrixMetric, compute_confusion_matrix_metric, get_confusion_matrix
1818
from .cumulative_average import CumulativeAverage
19+
from .embedding_collapse import EmbeddingCollapseMetric, compute_embedding_collapse
1920
from .f_beta_score import FBetaScore
2021
from .fid import FIDMetric, compute_frechet_distance
2122
from .froc import compute_fp_tp_probs, compute_fp_tp_probs_nd, compute_froc_curve_data, compute_froc_score

0 commit comments

Comments
 (0)