Focal loss 3D nnU-Net for ICH/PHE/IVH CT segmentation (TICH-2)
2026-01-24https://doi.org/10.1148/atlas.1769273309892
10
Overview
Schema Version
https://atlas.rsna.org/schemas/2025-11/model.json
Name
Focal loss 3D nnU-Net for ICH/PHE/IVH CT segmentation (TICH-2)
Link
https://dx.doi.org/10.1148/ryai.220096
Indexing
Keywords: Head/Neck, Brain/Brain Stem, Hemorrhage, Segmentation, Quantification, Convolutional Neural Network (CNN), Deep Learning Algorithms, Machine Learning Algorithms, intracerebral hemorrhage, intraventricular hemorrhage, perihematomal edema, nnU-Net, U-Net
Content: CT, NR
RadLex: RID10321
Author(s)
Yong En Kok
Stefan Pszczolkowski
Zhe Kang Law
Azlinawati Ali
Kailash Krishnan
Philip M. Bath
Nikola Sprigg
Robert A. Dineen
Andrew P. French
Organization(s)
University of Nottingham
NIHR Nottingham Biomedical Research Centre
National University of Malaysia
Universiti Sultan Zainal Abidin
Nottingham University Hospitals NHS Trust
Version
1.0
License
Text: © 2022 by the Radiological Society of North America, Inc.
Funding
This study included data from the TICH-2 trial funded in part by the UK National Institute for Health Research Health Technology Assessment Programme (project code 11_129_109).
Ethical review
Ethical approval was granted from the UK Health Research Authority and relevant national or local institutional review boards; written informed consent was obtained before enrollment.
Date
Published: 2022-09-28
References
[1] Kok YE, Pszczolkowski S, Law ZK, et al.. "Semantic Segmentation of Spontaneous Intracerebral Hemorrhage, Intraventricular Hemorrhage, and Associated Edema on CT Images Using Deep Learning". Radiology: Artificial Intelligence. 2022 Nov;4(6):e220096.. 2022-11-01. doi:10.1148/ryai.220096. PMID: 36523645. PMCID: PMC9745441.
Model
Architecture
Three-dimensional nnU-Net (U-Net–based convolutional neural network); comparisons included 2D nnU-Net, ensemble (2D+3D), U-Net (MONAI BasicUNet), BLAST-CT (DeepMedic-based), and DeepLabv3+.
Availability
Implementations used: nnU-Net (https://github.com/MIC-DKFZ/nnUNet); BLAST-CT (https://github.com/biomedia-mira/blast-ct); MONAI BasicUNet; DeepLabv3+ implementation used by authors (https://github.com/janetkok/MONAI-DeepLabV3plus); loss function code (https://github.com/JunMa11/SegLoss.git). Data sharing on request to the TICH-2 Chief Investigator as per article.
Clinical benefit
Automated segmentation and quantification of ICH, PHE, and IVH to provide quantitative outcome measures for clinical trials and potentially accelerate studies in large cohorts of spontaneous ICH.
Clinical workflow phase
Clinical research/clinical trials biomarker quantification; potential decision support for outcome prediction studies.
Degree of automation
Fully automated image segmentation of target lesions on noncontrast head CT.
Indications for use
Segmentation and volumetric quantification of intracerebral hemorrhage, perihematomal edema, and intraventricular hemorrhage on baseline noncontrast CT in adults with spontaneous ICH (within 8 hours of symptom onset in the TICH-2 cohort).
Input
Baseline noncontrast head CT scans from patients with spontaneous intracerebral hemorrhage.
Instructions
Models trained for 1800 epochs using default pipeline parameters of respective frameworks; three-dimensional nnU-Net refined with Focal/Tversky/FocalTversky/DiceTopK loss functions to address class imbalance.
Limitations
Lower performance and smoothed boundaries for PHE segmentation; single train/test split (no k-fold cross-validation); DSC can be inflated by cases with absent lesions; ground truth contains some segmentation errors; class imbalance with approximately 30% IVH prevalence; multicenter CT variability.
Output
CDEs: RDE1955, RDE1775, RDE842, RDE1776
Description: Semantic segmentation masks for ICH, PHE, and IVH with corresponding lesion volumes and agreement metrics.
Recommendation
U-Net–based networks, particularly 3D nnU-Net with Focal loss, are recommended to address class imbalance and improve IVH segmentation; performance satisfactory for ICH and moderate for PHE.
Reproducibility
Open-source frameworks were used (nnU-Net, MONAI, BLAST-CT); models trained with 1800 epochs; dataset split 90/10; specifics detailed in the supplement (Appendix E1/Table E1).
Use
Intended: Image segmentation
User
Intended: Radiologist, Researcher