diff --git a/.gitignore b/.gitignore index 252aa91d4..ee76df517 100644 --- a/.gitignore +++ b/.gitignore @@ -113,3 +113,4 @@ ENV/ *.model +!documentation/assets/scribble_example.png \ No newline at end of file diff --git a/documentation/assets/scribble_example.png b/documentation/assets/scribble_example.png new file mode 100644 index 000000000..62e775cfd Binary files /dev/null and b/documentation/assets/scribble_example.png differ diff --git a/documentation/ignore_label.md b/documentation/ignore_label.md new file mode 100644 index 000000000..17336bba4 --- /dev/null +++ b/documentation/ignore_label.md @@ -0,0 +1,91 @@ +# Ignore Label + +The _ignore label_ can be used to mark regions that should be ignored by nnU-Net. When the _ignore label_ is used nnU-Net switches internally to a partial variant of its loss function (See our [paper](https://arxiv.org/abs/2403.12834) for more information). Consequently, all pixels marked with the ignore label are ignored during the loss computation and gradients are not propagated from these pixels. Furthermore, these pixels will also be excluded from model evaluation. The most common use-cases for the ignore label are: + +- Sparse annotation + - Annotation of all or a subset of slices with scribbles (Scribble Supervision) + - Dense annotation of a subset of slices + - Dense annotation of chosen patches/cubes within an image +- Coarsly masking out faulty segmentations in the reference segmentations +- Masking areas for other reasons + +If you are using nnU-Net's ignore label, please cite the following paper in addition to the original nnU-net paper: + +``` +Gotkowski, K., Lüth, C., Jäger, P. F., Ziegler, S., Krämer, L., Denner, S., Xiao, S., Disch, N., H., K., & Isensee, F. (2024). Embarrassingly Simple Scribble Supervision for 3D Medical Segmentation. ArXiv. /abs/2403.12834 +``` + +## Usecases + +### Scribble Supervision + +Scribbles are free-form drawings to coarsly annotate an image. As we have demonstrated in our recent [paper](https://arxiv.org/abs/2403.12834), nnU-Net's partial loss implementation enables state-of-the-art learning from partially annotated data and even surpasses many purpose-built methods for learning from scribbles. As a starting point, for each image slice and each class (including background), an interior and a border scribble should be generated: + +- Interior Scribble: A scribble placed randomly within the class interior of a class instance +- Border Scribble: A scribble roughly delineating a small part of the class border of a class instance + +An example of such scribble annotations is depicted in Figure 1 and an animation in Animation 1. +Depending on the availability of data and their variability it is also possible to only annotated a subset of selected slices. + +

+ +

Figure 1: Examples of segmentation types with (A) depicting a dense segmentation and (B) a scribble segmentation.
+ +

+ +

+ + +

Animation 1: Depiction of a dense segmentation and a scribble annotation. Background scribbles have been excluded for better visualization.
+

+ +### Dense annotation of a subset of slices + +Another form of sparse annotation is the dense annotation of a subset of slices. These slices should be selected by the user either randomly, based on visual class variation between slices or in an active learning setting. An example with only 10% of slices annotated is depicted in Figure 2. + +

+ + +

Figure 2: Examples of a dense annotation of a subset of slices. The ignored areas are shown in red.
+ +

+ + +## Usage within nnU-Net + +Usage of the ignore label in nnU-Net is straightforward and only requires the definition of an _ignore_ label in the _dataset.json_. +This ignore label MUST be the highest integer label value in the segmentation. Exemplary, given the classes background and two foreground classes, then the ignore label must have the integer 3. The ignore label must be named _ignore_ in the _dataset.json_. Given the BraTS dataset as an example the labels dict of the _dataset.json_ must look like this: + +```python +... +"labels": { + "background": 0, + "edema": 1, + "non_enhancing_and_necrosis": 2, + "enhancing_tumor": 3, + "ignore": 4 +}, +... +``` + +Of course, the ignore label is compatible with [region-based training](region_based_training.md): + +```python +... +"labels": { + "background": 0, + "whole_tumor": (1, 2, 3), + "tumor_core": (2, 3), + "enhancing_tumor": 3, # or (3, ) + "ignore": 4 +}, +"regions_class_order": (1, 2, 3), # don't declare ignore label here! It is not predicted +... +``` + +Then use the dataset as you would any other. + +Remember that nnU-Net runs a cross-validation. Thus, it will also evaluate on your partially annotated data. This +will of course work! If you wish to compare different sparse annotation strategies (through simulations for example), +we recommend evaluating on densely annotated images by running inference and then using `nnUNetv2_evaluate_folder` or +`nnUNetv2_evaluate_simple`. \ No newline at end of file