MRI-based Head and Neck Tumor Segmentation Using nnU-Net with 15-fold Cross-Validation Ensemble

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Mol, Frank N., van der Hoek, Luuk, Ma, Baoqiang, Nagam, Bharath Chowdhary, Sijtsema, Nanna M., van Dijk, Lisanne V., Bunte, Kerstin, Vlijm, Rifka, van Ooijen, Peter M. A.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912149504262144
author Mol, Frank N.
van der Hoek, Luuk
Ma, Baoqiang
Nagam, Bharath Chowdhary
Sijtsema, Nanna M.
van Dijk, Lisanne V.
Bunte, Kerstin
Vlijm, Rifka
van Ooijen, Peter M. A.
author_facet Mol, Frank N.
van der Hoek, Luuk
Ma, Baoqiang
Nagam, Bharath Chowdhary
Sijtsema, Nanna M.
van Dijk, Lisanne V.
Bunte, Kerstin
Vlijm, Rifka
van Ooijen, Peter M. A.
contents The superior soft tissue differentiation provided by MRI may enable more accurate tumor segmentation compared to CT and PET, potentially enhancing adaptive radiotherapy treatment planning. The Head and Neck Tumor Segmentation for MR-Guided Applications challenge (HNTSMRG-24) comprises two tasks: segmentation of primary gross tumor volume (GTVp) and metastatic lymph nodes (GTVn) on T2-weighted MRI volumes obtained at (1) pre-radiotherapy (pre-RT) and (2) mid-radiotherapy (mid-RT). The training dataset consists of data from 150 patients, including MRI volumes of pre-RT, mid-RT, and pre-RT registered to the corresponding mid-RT volumes. Each MRI volume is accompanied by a label mask, generated by merging independent annotations from a minimum of three experts. For both tasks, we propose adopting the nnU-Net V2 framework by the use of a 15-fold cross-validation ensemble instead of the standard number of 5 folds for increased robustness and variability. For pre-RT segmentation, we augmented the initial training data (150 pre-RT volumes and masks) with the corresponding mid-RT data. For mid-RT segmentation, we opted for a three-channel input, which, in addition to the mid-RT MRI volume, comprises the registered pre-RT MRI volume and the corresponding mask. The mean of the aggregated Dice Similarity Coefficient for GTVp and GTVn is computed on a blind test set and determines the quality of the proposed methods. These metrics determine the final ranking of methods for both tasks separately. The final blind testing (50 patients) of the methods proposed by our team, RUG_UMCG, resulted in an aggregated Dice Similarity Coefficient of 0.81 (0.77 for GTVp and 0.85 for GTVn) for Task 1 and 0.70 (0.54 for GTVp and 0.86 for GTVn) for Task 2.
format Preprint
id arxiv_https___arxiv_org_abs_2412_06610
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MRI-based Head and Neck Tumor Segmentation Using nnU-Net with 15-fold Cross-Validation Ensemble
Mol, Frank N.
van der Hoek, Luuk
Ma, Baoqiang
Nagam, Bharath Chowdhary
Sijtsema, Nanna M.
van Dijk, Lisanne V.
Bunte, Kerstin
Vlijm, Rifka
van Ooijen, Peter M. A.
Medical Physics
The superior soft tissue differentiation provided by MRI may enable more accurate tumor segmentation compared to CT and PET, potentially enhancing adaptive radiotherapy treatment planning. The Head and Neck Tumor Segmentation for MR-Guided Applications challenge (HNTSMRG-24) comprises two tasks: segmentation of primary gross tumor volume (GTVp) and metastatic lymph nodes (GTVn) on T2-weighted MRI volumes obtained at (1) pre-radiotherapy (pre-RT) and (2) mid-radiotherapy (mid-RT). The training dataset consists of data from 150 patients, including MRI volumes of pre-RT, mid-RT, and pre-RT registered to the corresponding mid-RT volumes. Each MRI volume is accompanied by a label mask, generated by merging independent annotations from a minimum of three experts. For both tasks, we propose adopting the nnU-Net V2 framework by the use of a 15-fold cross-validation ensemble instead of the standard number of 5 folds for increased robustness and variability. For pre-RT segmentation, we augmented the initial training data (150 pre-RT volumes and masks) with the corresponding mid-RT data. For mid-RT segmentation, we opted for a three-channel input, which, in addition to the mid-RT MRI volume, comprises the registered pre-RT MRI volume and the corresponding mask. The mean of the aggregated Dice Similarity Coefficient for GTVp and GTVn is computed on a blind test set and determines the quality of the proposed methods. These metrics determine the final ranking of methods for both tasks separately. The final blind testing (50 patients) of the methods proposed by our team, RUG_UMCG, resulted in an aggregated Dice Similarity Coefficient of 0.81 (0.77 for GTVp and 0.85 for GTVn) for Task 1 and 0.70 (0.54 for GTVp and 0.86 for GTVn) for Task 2.
title MRI-based Head and Neck Tumor Segmentation Using nnU-Net with 15-fold Cross-Validation Ensemble
topic Medical Physics
url https://arxiv.org/abs/2412.06610