Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

nnU-Net: Self-adapting Framework for U-Net-Based Medical Image Segmentation

About

The U-Net was presented in 2015. With its straight-forward and successful architecture it quickly evolved to a commonly used benchmark in medical image segmentation. The adaptation of the U-Net to novel problems, however, comprises several degrees of freedom regarding the exact architecture, preprocessing, training and inference. These choices are not independent of each other and substantially impact the overall performance. The present paper introduces the nnU-Net ('no-new-Net'), which refers to a robust and self-adapting framework on the basis of 2D and 3D vanilla U-Nets. We argue the strong case for taking away superfluous bells and whistles of many proposed network designs and instead focus on the remaining aspects that make out the performance and generalizability of a method. We evaluate the nnU-Net in the context of the Medical Segmentation Decathlon challenge, which measures segmentation performance in ten disciplines comprising distinct entities, image modalities, image geometries and dataset sizes, with no manual adjustments between datasets allowed. At the time of manuscript submission, nnU-Net achieves the highest mean dice scores across all classes and seven phase 1 tasks (except class 1 in BrainTumour) in the online leaderboard of the challenge.

Fabian Isensee, Jens Petersen, Andre Klein, David Zimmerer, Paul F. Jaeger, Simon Kohl, Jakob Wasserthal, Gregor Koehler, Tobias Norajitra, Sebastian Wirkert, Klaus H. Maier-Hein• 2018

Related benchmarks

TaskDatasetResultRank
Medical Image SegmentationISIC
DICE81.9
64
Medical Image SegmentationMedical Segmentation Decathlon (MSD) (test)
Mean Dice Score75
27
Medical Image SegmentationMSD Pancreas (test)--
25
Prostate SegmentationProstate
DSC (Avg)59
21
Medical Image SegmentationCHAOS
DSC89.4
14
Medical Image SegmentationNuclei
Dice Score60.3
12
Medical Image SegmentationMSD Heart (5-fold CV)--
12
Medical Image SegmentationMSD Hippocampus (test)--
12
Image SegmentationKiTS19 kidney (organ) (val)
DSC96.9
11
Image SegmentationKiTS19 kidney (tumor) (val)
DSC87.6
11
Showing 10 of 34 rows

Other info

Code

Follow for update