Cancer3.AIAI w onkologiiModele AI › nnU-Net
Framework Otwarte wagi Nie dotyczy radiology

nnU-Net v2

Samokonfigurujący się framework segmentacji: dla etykietowanego zbioru dobiera preprocessing, topologię sieci i harmonogram treningu automatycznie; pozostaje punktem odniesienia w większości konkursów segmentacji medycznej, także guzów.

W skrócie

TwórcaGerman Cancer Research Center (DKFZ), Division of Medical Image Computing
Wersjav2
Wydanie2020-12-07
Licencjaapache-2.0
DostępnośćOtwarte wagi
RodzajFramework
Status regulacyjnyNie dotyczy

Co robi

Trenuje mocny 2D/3D U-Net na własnych zaadnotowanych wolumenach TK/MRI/PET bez projektowania architektury; używany do segmentacji guzów, narządów krytycznych i węzłów chłonnych w radioterapii i radiomice.

Zadania, typy danych i nowotwory

Typ danychTKMRIPET
NowotwórWiele nowotworów
WejścieNIfTI volumes + label maps
Wyjścielabel maps

Architektura

RodzinaU-Net (2D, 3D full-res, 3D cascade), auto-configured
Pretreningnone — trained from scratch per dataset

Dane treningowe

Brak stałego korpusu treningowego. nnU-Net nie udostępnia wag wytrenowanych wcześniej: konfiguruje się sam i trenuje od zera na zbiorze etykietowanym dostarczonym przez użytkownika. W pracy źródłowej zastosowano go bez żadnych ręcznych korekt między zbiorami do Medical Segmentation Decathlon (10 zadań) oraz kolejnych konkursów segmentacji biomedycznej.

Powiązane zbiory danych

Ewaluacja

Benchmark / zbiórMiaraWartośćWalidacja zewnętrznaŹródło
Medical Segmentation Decathlon + 23 public challenges (paper) Dice first place or top tier on the majority tak Źródło

Jak uruchomić

pip install nnunetv2
nnUNetv2_plan_and_preprocess -d 001 --verify_dataset_integrity
nnUNetv2_train 001 3d_fullres 0
nnUNetv2_predict -i imagesTs/ -o preds/ -d 001 -c 3d_fullres
Bibliotekapip package nnunetv2
SprzętGPU with ≥10 GB for 3D full-resolution training

Status regulacyjny i przeznaczenie

Status regulacyjnyNie dotyczy
PrzeznaczenieFramework badawczy; produkty zbudowane na nim wymagają własnego dopuszczenia.

Źródło →

Status regulacyjny cytujemy ze wskazanego źródła i może się zmienić. Modele „tylko do badań” nie służą do decyzji klinicznych.

Ograniczenia i błędy systematyczne

  • Wymaga etykiet wokselowych; brak skrótu przez pretrening.
  • Długi trening; wariant kaskadowy jest ciężki.

Źródła

  1. Isensee F et al. nnU-Net: a self-configuring method for deep learning-based biomedical image segmentation. Nat Methods 2021
  2. Isensee F et al. nnU-Net: Self-adapting Framework for U-Net-Based Medical Image Segmentation (preprint)
  3. GitHub — MIC-DKFZ/nnUNet

Ta strona ma charakter edukacyjny — nie stanowi porady medycznej i nie zastępuje konsultacji z onkologiem. Decyzje diagnostyczne i terapeutyczne podejmuje wyłącznie lekarz specjalista.