对比nnUNet与MedNeXt在头颈癌放疗影像分割中的表现,实现高效自动勾画。
Comparative Analysis of nnUNet and MedNeXt for Head and Neck Tumor Segmentation in MRI-guided Radiotherapy
- 用注册前后扫描数据预训练,再微调原图提升分割精度。
- 任务1达0.8254的平均骰子系数,排名第一;任务2得0.7005,排名第八。
- 适合医学图像分割研究者和放疗自动化系统开发者参考。
放射治疗(RT)是头颈癌(HNC)的重要手段,磁共振成像(MRI)引导的放疗能提供更优的软组织对比度和功能成像。然而,手动肿瘤勾画耗时且复杂,仍是挑战。本研究中,我们以团队TUMOR身份参与了聚焦于放疗前及放疗中期MRI图像中主要原发肿瘤体积(GTVp)和转移淋巴结肿瘤体积(GTVn)自动分割的HNTS-MRG24 MICCAI挑战赛。采用包含150例患者数据的HNTS-MRG2024数据集,包含原始与配准后的放疗前及放疗中期的T2加权图像及其对应的GTVp和GTVn标注。使用深度学习领域两个前沿模型:nnUNet和MedNeXt。针对任务1,先在配准后的放疗前与放疗中期图像上进行预训练,再在原始放疗前图像上微调。针对任务2,将配准后的放疗前图像、对应分割掩码以及放疗中期数据作为多通道输入进行训练。最终,任务1解决方案在测试阶段取得0.8254的聚合骰子系数,位列第一;任务2方案得分0.7005,排名第八。所提方法已公开于GitHub仓库。
原文摘要 · Abstract (English)
Radiation therapy (RT) is essential in treating head and neck cancer (HNC), with magnetic resonance imaging(MRI)-guided RT offering superior soft tissue contrast and functional imaging. However, manual tumor segmentation is time-consuming and complex, and therfore remains a challenge. In this study, we present our solution as team TUMOR to the HNTS-MRG24 MICCAI Challenge which is focused on automated segmentation of primary gross tumor volumes (GTVp) and metastatic lymph node gross tumor volume (GTVn) in pre-RT and mid-RT MRI images. We utilized the HNTS-MRG2024 dataset, which consists of 150 MRI scans from patients diagnosed with HNC, including original and registered pre-RT and mid-RT T2-weighted images with corresponding segmentation masks for GTVp and GTVn. We employed two state-of-the-art models in deep learning, nnUNet and MedNeXt. For Task 1, we pretrained models on pre-RT registered and mid-RT images, followed by fine-tuning on original pre-RT images. For Task 2, we combined registered pre-RT images, registered pre-RT segmentation masks, and mid-RT data as a multi-channel input for training. Our solution for Task 1 achieved 1st place in the final test phase with an aggregated Dice Similarity Coefficient of 0.8254, and our solution for Task 2 ranked 8th with a score of 0.7005. The proposed solution is publicly available at Github Repository.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。