Improve model card metadata and content
Browse filesHi, I'm Niels from the Hugging Face community team. This PR improves the model card by adding relevant metadata and linking it to the associated research paper. It also adds a link to the official code repository and provides sample usage instructions based on the GitHub README to improve the artifact's discoverability and usability.
README.md
CHANGED
|
@@ -1,5 +1,55 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
|
|
|
| 3 |
---
|
| 4 |
|
| 5 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
+
pipeline_tag: image-segmentation
|
| 4 |
---
|
| 5 |
|
| 6 |
+
# WORD Dataset Queries and Analysis Results
|
| 7 |
+
|
| 8 |
+
This repository contains queries and analysis results for the **WORD** 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** ([Arxiv](https://arxiv.org/abs/2601.13677))
|
| 11 |
+
- **nnActive: A Framework for Evaluation of Active Learning in 3D Biomedical Segmentation** ([Arxiv](https://arxiv.org/abs/2511.19183))
|
| 12 |
+
|
| 13 |
+
Official Code: [GitHub - MIC-DKFZ/nnActive](https://github.com/MIC-DKFZ/nnActive)
|
| 14 |
+
|
| 15 |
+
## Summary
|
| 16 |
+
Active learning (AL) has the potential to drastically reduce annotation costs in 3D biomedical image segmentation. This work introduces **Class-stratified Scheduled Power Predictive Entropy (ClaSP PE)**, a query strategy that addresses class imbalance and redundancy. Within the **nnActive** framework, the authors demonstrate that ClaSP PE consistently outperforms improved random baselines in terms of both performance and annotation efficiency in realistic scenarios.
|
| 17 |
+
|
| 18 |
+
## Sample Usage
|
| 19 |
+
|
| 20 |
+
To setup and run experiments using the `nnactive` framework, you can use the following CLI commands:
|
| 21 |
+
|
| 22 |
+
```bash
|
| 23 |
+
# Setup an experiment
|
| 24 |
+
nnactive setup_experiment --experiment word__tr-nnActiveTrainer_500epochs__patch-29_74_87__sb-random-label2-all-classes__sbs-800__qs-800__unc-random__seed-12348
|
| 25 |
+
|
| 26 |
+
# Run the experiment
|
| 27 |
+
nnactive run_experiment --experiment word__tr-nnActiveTrainer_500epochs__patch-29_74_87__sb-random-label2-all-classes__sbs-800__qs-800__unc-random__seed-12348
|
| 28 |
+
```
|
| 29 |
+
|
| 30 |
+
To analyze existing experiments:
|
| 31 |
+
|
| 32 |
+
```bash
|
| 33 |
+
nnactive analyze_experiments --base_path $nnActive_results --raw_path $nnActive_data --output_path {OUTPUT_PATH}
|
| 34 |
+
```
|
| 35 |
+
|
| 36 |
+
## Citation
|
| 37 |
+
If you use these results or the nnActive framework, please cite the following work:
|
| 38 |
+
|
| 39 |
+
```bibtex
|
| 40 |
+
@article{luth2026finally,
|
| 41 |
+
title={Finally Outshining the Random Baseline: A Simple and Effective Solution for Active Learning in 3D Biomedical Imaging},
|
| 42 |
+
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},
|
| 43 |
+
journal={Transactions on Machine Learning Research},
|
| 44 |
+
year={2026},
|
| 45 |
+
url={https://openreview.net/forum?id=UamXueEaYW},
|
| 46 |
+
}
|
| 47 |
+
|
| 48 |
+
@article{luth2025nnactive,
|
| 49 |
+
title={nnActive: A Framework for Evaluation of Active Learning in 3D Biomedical Segmentation},
|
| 50 |
+
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},
|
| 51 |
+
journal={Transactions on Machine Learning Research},
|
| 52 |
+
year={2025},
|
| 53 |
+
url={https://openreview.net/forum?id=AJAnmRLJjJ},
|
| 54 |
+
}
|
| 55 |
+
```
|