Improve model card with metadata and links

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +53 -1
README.md CHANGED
@@ -1,5 +1,57 @@
1
  ---
2
  license: apache-2.0
 
3
  ---
4
 
5
- Queries and analysis results for the Toothfairy2 dataset, as presented in the papers [https://arxiv.org/abs/2511.19183](https://arxiv.org/abs/2511.19183) and [https://arxiv.org/abs/2601.13677](https://arxiv.org/abs/2601.13677).
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ ```