arXiv:2412.14846eess.IVcs.AI2024-12被引 1

提升头颈癌放疗中影像肿瘤分割精度,融合预训练与双流网络。

Head and Neck Tumor Segmentation of MRI from Pre- and Mid-radiotherapy with Pre-training, Data Augmentation and Dual Flow UNet

  • 用预训练权重和MixUp增强数据,提升放疗前图像分割效果。
  • 放疗中图像采用双流结构,融合前后期影像实现72.53% DSC。
  • 模型适用于临床放疗动态监测,对医学影像研究者有参考价值。

头颈部肿瘤及转移性淋巴结对治疗规划和预后分析至关重要。精确分割需像素级标注,自动化分割技术对头颈癌诊疗不可或缺。本研究考察多种策略对放疗前(pre-RT)与放疗中(mid-RT)MRI图像分割的影响。针对pre-RT图像,采用完全监督学习,并结合预训练权重与MixUp数据增强;针对mid-RT图像,提出一种计算友好的双流网络架构,分别处理mid-RT图像与配准后的pre-RT图像及其标签。中流分支在前向传播中逐步整合pre-RT图像与标签信息。每折选取最优模型,通过集成平均进行推理。最终测试中,模型在HiLab数据集上取得pre-RT 82.38%、mid-RT 72.53%的综合Dice相似系数(DSC)。代码已公开于https://github.com/WltyBY/HNTS-MRG2024_train_code。

原文摘要 · Abstract (English)

Head and neck tumors and metastatic lymph nodes are crucial for treatment planning and prognostic analysis. Accurate segmentation and quantitative analysis of these structures require pixel-level annotation, making automated segmentation techniques essential for the diagnosis and treatment of head and neck cancer. In this study, we investigated the effects of multiple strategies on the segmentation of pre-radiotherapy (pre-RT) and mid-radiotherapy (mid-RT) images. For the segmentation of pre-RT images, we utilized: 1) a fully supervised learning approach, and 2) the same approach enhanced with pre-trained weights and the MixUp data augmentation technique. For mid-RT images, we introduced a novel computational-friendly network architecture that features separate encoders for mid-RT images and registered pre-RT images with their labels. The mid-RT encoder branch integrates information from pre-RT images and labels progressively during the forward propagation. We selected the highest-performing model from each fold and used their predictions to create an ensemble average for inference. In the final test, our models achieved a segmentation performance of 82.38% for pre-RT and 72.53% for mid-RT on aggregated Dice Similarity Coefficient (DSC) as HiLab. Our code is available at https://github.com/WltyBY/HNTS-MRG2024_train_code.

肿瘤分割医学影像深度学习放疗监测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。