用深度学习自动分割头颈癌放疗中肿瘤体积,提升效率与一致性。
Deep Learning for Longitudinal Gross Tumor Volume Segmentation in MRI-Guided Adaptive Radiotherapy for Head and Neck Cancer
- 基于SegResNet设计纵向分割模型,融合预治与中程期MRI数据。
- 最终模型在内部测试集上达0.794的平均骰子系数,部分任务第一。
- 适合放射科医生和医学影像算法研发者参考使用。
准确分割大体肿瘤体积(GTV)对头颈癌磁共振引导自适应放疗(MRgART)至关重要。然而,治疗过程中人工勾画GTV耗时且易受观察者差异影响。深度学习有望通过自动分割克服这些挑战。本研究由团队UW LAIR针对放疗前(任务1)和中程放疗期(任务2)的GTV分割提出解决方案。采用以带深层监督的SegResNet为骨干的多模型架构。任务1中,结合放疗前与中程期MRI数据训练,较仅用放疗前数据的模型显著提升聚合骰子系数(DSCagg)。任务2中引入掩码感知注意力模块,使放疗前的GTV掩码影响中程期特征学习,相较简单拼接输入的方法略有提升。最终测试中,10个放疗前模型集成达到平均DSCagg 0.794,其中原发肿瘤(GTVp)为0.745,转移淋巴结(GTVn)为0.844;10个中程期模型集成平均达DSCagg 0.733,GTVp为0.607,GTVn为0.859,获得第一名。结果表明,所提方法可有效支持MRgART中的自动化肿瘤分割,优化放疗流程。代码与权重已开源。
原文摘要 · Abstract (English)
Accurate segmentation of gross tumor volume (GTV) is essential for effective MRI-guided adaptive radiotherapy (MRgART) in head and neck cancer. However, manual segmentation of the GTV over the course of therapy is time-consuming and prone to interobserver variability. Deep learning (DL) has the potential to overcome these challenges by automatically delineating GTVs. In this study, our team, $\textit{UW LAIR}$, tackled the challenges of both pre-radiotherapy (pre-RT) (Task 1) and mid-radiotherapy (mid-RT) (Task 2) tumor volume segmentation. To this end, we developed a series of DL models for longitudinal GTV segmentation. The backbone of our models for both tasks was SegResNet with deep supervision. For Task 1, we trained the model using a combined dataset of pre-RT and mid-RT MRI data, which resulted in the improved aggregated Dice similarity coefficient (DSCagg) on an internal testing set compared to models trained solely on pre-RT MRI data. In Task 2, we introduced mask-aware attention modules, enabling pre-RT GTV masks to influence intermediate features learned from mid-RT data. This attention-based approach yielded slight improvements over the baseline method, which concatenated mid-RT MRI with pre-RT GTV masks as input. In the final testing phase, the ensemble of 10 pre-RT segmentation models achieved an average DSCagg of 0.794, with 0.745 for primary GTV (GTVp) and 0.844 for metastatic lymph nodes (GTVn) in Task 1. For Task 2, the ensemble of 10 mid-RT segmentation models attained an average DSCagg of 0.733, with 0.607 for GTVp and 0.859 for GTVn, leading us to $\textbf{achieve 1st place}$. In summary, we presented a collection of DL models that could facilitate GTV segmentation in MRgART, offering the potential to streamline radiation oncology workflows. Our code and model weights are available at https://github.com/xtie97/HNTS-MRG24-UWLAIR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。