用多阶段学习提升妇科放疗器官分割精度,显著优于现有方法。
A Large Convolutional Neural Network for Clinical Target and Multi-organ Segmentation in Gynecologic Brachytherapy with Multi-stage Learning
- 分三阶段训练:自监督预训练+多器官微调+临床任务精调
- 宫颈肿瘤分割DSC达0.837,膀胱、直肠等关键器官分割效果优秀
- 适合医学影像分割研究者及放射治疗临床应用开发者
目的:准确分割临床靶区(CTV)和危及器官对优化妇科腔内放疗(GYN-BT)计划至关重要。但解剖变异、CT图像软组织对比度低及标注数据有限带来挑战。本文提出GynBTNet,一种基于多阶段学习的新型框架,通过自监督预训练与层级微调提升分割性能。方法:采用三阶段策略:(1)在大规模CT数据集上使用稀疏子流形卷积进行自监督预训练,提取鲁棒解剖特征;(2)在多器官分割数据集上进行有监督微调以优化特征提取;(3)在专用GYN-BT数据集上进行任务特定微调,提升临床适用性。模型通过Dice相似系数(DSC)、95% Hausdorff距离(HD95)和平均表面距离(ASD)评估。结果:GynBTNet显著优于nnU-Net和Swin-UNETR,CTV DSC为0.837±0.068,膀胱0.940±0.052,直肠0.842±0.070,子宫0.871±0.047,且HD95与ASD更低。自监督预训练显著提升复杂边界结构分割性能。但乙状结肠分割仍具挑战,或因解剖模糊与个体差异。统计分析确认改进具有显著性。
原文摘要 · Abstract (English)
Purpose: Accurate segmentation of clinical target volumes (CTV) and organs-at-risk is crucial for optimizing gynecologic brachytherapy (GYN-BT) treatment planning. However, anatomical variability, low soft-tissue contrast in CT imaging, and limited annotated datasets pose significant challenges. This study presents GynBTNet, a novel multi-stage learning framework designed to enhance segmentation performance through self-supervised pretraining and hierarchical fine-tuning strategies. Methods: GynBTNet employs a three-stage training strategy: (1) self-supervised pretraining on large-scale CT datasets using sparse submanifold convolution to capture robust anatomical representations, (2) supervised fine-tuning on a comprehensive multi-organ segmentation dataset to refine feature extraction, and (3) task-specific fine-tuning on a dedicated GYN-BT dataset to optimize segmentation performance for clinical applications. The model was evaluated against state-of-the-art methods using the Dice Similarity Coefficient (DSC), 95th percentile Hausdorff Distance (HD95), and Average Surface Distance (ASD). Results: Our GynBTNet achieved superior segmentation performance, significantly outperforming nnU-Net and Swin-UNETR. Notably, it yielded a DSC of 0.837 +/- 0.068 for CTV, 0.940 +/- 0.052 for the bladder, 0.842 +/- 0.070 for the rectum, and 0.871 +/- 0.047 for the uterus, with reduced HD95 and ASD compared to baseline models. Self-supervised pretraining led to consistent performance improvements, particularly for structures with complex boundaries. However, segmentation of the sigmoid colon remained challenging, likely due to anatomical ambiguities and inter-patient variability. Statistical significance analysis confirmed that GynBTNet's improvements were significant compared to baseline models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。