STU-Net Large — BraTS 2026 Brain Metastasis Segmentation
STU-Net Large checkpoint trained on the BraTS 2026 Brain Metastasis Segmentation training set (fold 0, checkpoint_best.pth). Part of a three-model ensemble; see the full pipeline at the GitHub repository.
Model Details
- Architecture: Swin Transformer-based U-Net (large variant)
- Input: 4 MRI modalities — T1n, T1c, T2w, T2f
- Output: 5-class segmentation (background, NETC, SNFH, ET, RC)
- Trainer:
STUNetTrainerLarge - Checkpoint:
checkpoint_best.pth(fold 0)
Label Convention
| Label | Region |
|---|---|
| 0 | Background |
| 1 | NETC (Non-Enhancing Tumor Core) |
| 2 | SNFH (Surrounding Non-Enhancing FLAIR Hyperintensity) |
| 3 | ET (Enhancing Tumor) |
| 4 | RC (Resection Cavity) |
Usage
Inference code and full ensemble pipeline: GitHub repository
Install STU-Net following the instructions at uni-medical/STU-Net.
Related Models
Citation
@article{TODO}