Improve model card with metadata and links
Browse filesHi! I'm Niels, part of the community science team at Hugging Face.
This PR aims to improve the documentation of this repository by:
- Adding the `pipeline_tag: image-segmentation` to the metadata to help users find these artifacts.
- Linking the model card to the corresponding research paper on the Hugging Face Hub.
- Adding a link to the official GitHub repository.
Please review and merge if this looks good!
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 |
+
```
|