CLN Segmenter MSD Fold0

--- license: cc-by-sa-4.0 tags: - nnunet - nnunetv2 - medical-imaging - segmentation - 3d-segmentation - ct - lung - lung-cancer - tumor-segmentation library_name: nnunetv2 pipeline_tag: image-segmentation datasets: - MSD-Task06-Lung language: - en --- # CLN-Segmenter — MSD Task06 Lung Tumor Segmentation (fold 0) A 3D U-Net (nnU-Net v2 `3d_fullres`) trained on the **Medical Segmentation Decathlon Task06: Lung Tumor** dataset, fold 0 of 5-fold cross-validation. Released as part of the CLN-Segmenter project at the Rasool Lab, Moffitt Cancer Center. This is a single-fold pretrain checkpoint, intended as a starting point for downstream lung-lesion segmentation work — not a clinical-grade tool. ## Quick stats | | | |--|--| | **Architecture** | nnU-Net v2 `3d_fullres` (PlainConvUNet, 6 stages, features `[32, 64, 128, 256, 320, 320]`) | | **Training data** | MSD Task06 Lung — 63 cases (50 train / 13 val for fold 0) | | **Loss** | Dice + Cross-Entropy (nnU-Net default), `batch_dice=True` | | **Schedule** | 1000 epochs, polynomial LR decay 0.01 → 0, batch size 2, patch `[80, 192, 160]` | | **Hardware** | 1× NVIDIA H100 80GB, ~6h wall-time | | **Mean Validation Dice** (per-case, sliding-window) | **0.7161** | | **Best EMA Pseudo Dice** (in-training proxy) | 0.8155 (epoch ~755) | | **Foreground IoU** (per-case avg) | ~0.59 (from `validation_summary.json`) | | **Comparison** | Within published nnU-Net Task06 range (0.69–0.78 across various reports) | ## Files in this repo | File | Role | |------|------| | `checkpoint_best.pth` | Model weights — saved at the EMA Pseudo Dice peak (~epoch 755), *before* the late-epoch overfitting plateau | | `nnUNetPlans.json` | Architecture spec + preprocessing plans. **Required** for inference. | | `dataset.json` | Channel names, label names, file ending (nnU-Net v2 schema). **Required** for inference. | | `dataset_fingerprint.json` | HU intensity stats from training data | | `splits_final.json` | Train/val case ID splits for fold 0 (reproducibility) | | `progress.png` | Training curves: loss, Pseudo Dice, epoch duration, learning rate | ## Training data and provenance This model was trained **only on the publicly available MSD Task06 Lung dataset** (Antonelli et al. 2022, *Nature Communications*, CC-BY-SA 4.0). It contains expert pixel-level lung tumor annotations from 63 diagnostic CT scans. **No patient-identifiable or in...

Source context: 0 downloads · 0 likes · Pipeline image-segmentation · Library nnunetv2 · Repo Lab-Rasool/CLN-Segmenter-MSD-fold0