Improve model card with metadata and links
#1
by
nielsr
HF Staff
- opened
README.md
CHANGED
|
@@ -1,5 +1,57 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
|
|
|
| 3 |
---
|
| 4 |
|
| 5 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
+
pipeline_tag: image-segmentation
|
| 4 |
---
|
| 5 |
|
| 6 |
+
# nnActive: Toothfairy2 Queries and Analysis Results
|
| 7 |
+
|
| 8 |
+
This repository contains the queries and analysis results for the **Toothfairy2** dataset, as presented in the papers:
|
| 9 |
+
|
| 10 |
+
* **Finally Outshining the Random Baseline: A Simple and Effective Solution for Active Learning in 3D Biomedical Imaging** (TMLR 2026). [[Paper]](https://huggingface.co/papers/2601.13677)
|
| 11 |
+
* **nnActive: A Framework for Evaluation of Active Learning in 3D Biomedical Segmentation** (TMLR 2025). [[Paper]](https://arxiv.org/abs/2511.19183)
|
| 12 |
+
|
| 13 |
+
Official Code: [GitHub - MIC-DKFZ/nnActive](https://github.com/MIC-DKFZ/nnActive/tree/nnActive_v2)
|
| 14 |
+
|
| 15 |
+
## Summary
|
| 16 |
+
Active learning (AL) aims to reduce annotation costs in 3D biomedical image segmentation by selecting the most informative samples. However, standard uncertainty-based AL methods often struggle to outperform random baselines due to class imbalance and query redundancy.
|
| 17 |
+
|
| 18 |
+
This work introduces **ClaSP PE** (Class-stratified Scheduled Power Predictive Entropy), a simple and effective query strategy that addresses these limitations through:
|
| 19 |
+
1. **Class-stratified querying**: Ensures coverage of underrepresented structures.
|
| 20 |
+
2. **Scheduled Power Noising**: Enforces query diversity in early-stage AL and encourages exploitation later.
|
| 21 |
+
|
| 22 |
+
The results in this repository were generated using the **nnActive** framework, an open-source evaluation framework for 3D medical AL built on top of nnU-Net.
|
| 23 |
+
|
| 24 |
+
## Artifact Information
|
| 25 |
+
The repository includes:
|
| 26 |
+
- `loop_XXX.json`: Lists of specific 3D patches selected for annotation in each Active Learning cycle.
|
| 27 |
+
- `dataset.json`: Metadata and label configuration for the Toothfairy2 experiment.
|
| 28 |
+
- `auc.json` & `summary.json`: Detailed performance metrics and Dice score statistics.
|
| 29 |
+
- `config.json`: Experimental parameters for reproducibility.
|
| 30 |
+
|
| 31 |
+
## Citation
|
| 32 |
+
|
| 33 |
+
If you use the **nnActive** framework or these results, please cite:
|
| 34 |
+
|
| 35 |
+
```bibtex
|
| 36 |
+
@article{luth2025nnactive,
|
| 37 |
+
title={nnActive: A Framework for Evaluation of Active Learning in 3D Biomedical Segmentation},
|
| 38 |
+
author={Carsten T. L{\"u}th and Jeremias Traub and Kim-Celine Kahl and Till J. Bungert and Lukas Klein and Lars Kr{\"a}mer and Paul F Jaeger and Fabian Isensee and Klaus Maier-Hein},
|
| 39 |
+
journal={Transactions on Machine Learning Research},
|
| 40 |
+
issn={2835-8856},
|
| 41 |
+
year={2025},
|
| 42 |
+
url={https://openreview.net/forum?id=AJAnmRLJjJ},
|
| 43 |
+
}
|
| 44 |
+
```
|
| 45 |
+
|
| 46 |
+
If you use **ClaSP PE**, please cite:
|
| 47 |
+
|
| 48 |
+
```bibtex
|
| 49 |
+
@article{luth2026finally,
|
| 50 |
+
title={Finally Outshining the Random Baseline: A Simple and Effective Solution for Active Learning in 3D Biomedical Imaging},
|
| 51 |
+
author={Carsten T. L{\"u}th and Jeremias Traub and Kim-Celine Kahl and Till J. Bungert and Lukas Klein and Lars Kr{\"a}mer and Paul F Jaeger and Klaus Maier-Hein and Fabian Isensee},
|
| 52 |
+
journal={Transactions on Machine Learning Research},
|
| 53 |
+
issn={2835-8856},
|
| 54 |
+
year={2026},
|
| 55 |
+
url={https://openreview.net/forum?id=UamXueEaYW},
|
| 56 |
+
}
|
| 57 |
+
```
|