3D U-Net模型实现多中心、多示踪剂下肿瘤病灶自动分割,性能领先挑战赛。
AutoPET III Challenge: Tumor Lesion Segmentation using ResEnc-Model Ensemble
- 基于无新U-Net框架训练3D残差编码器网络,提升跨中心泛化能力。
- 采用总分割器裁剪数据并重采样,提升训练效率与质量。
- 测试时增强与后处理技术结合,Dice得分达0.9627,表现优异。
正电子发射断层扫描(PET)/计算机断层扫描(CT)在癌症诊断、管理和治疗规划中至关重要。在多示踪剂、多中心环境下,开发可靠的深度学习模型以分割PET/CT扫描中的肿瘤病灶,是当前研究重点。不同示踪剂(如氟脱氧葡萄糖FDG、前列腺特异性膜抗原PSMA)具有不同的生理摄取模式,且各中心的采集协议、扫描设备和患者群体存在差异,导致图像质量与病灶可检测性不一,增加了算法设计与泛化难度。为此,本研究在无新U-Net框架下训练了3D残差编码器U-Net,旨在实现全身PET/CT扫描中肿瘤病灶自动分割的跨示踪剂与跨中心泛化。我们探索了多种预处理方法,最终采用总分割器(Total Segmentator)对训练数据进行裁剪,并应用重采样。推理阶段引入测试时增强与其他后处理技术,显著提升分割效果。目前,本团队在Auto-PET III挑战赛中位居榜首,在初步测试集上取得0.9627的Dice分数,优于基准模型。
原文摘要 · Abstract (English)
Positron Emission Tomography (PET) /Computed Tomography (CT) is crucial for diagnosing, managing, and planning treatment for various cancers. Developing reliable deep learning models for the segmentation of tumor lesions in PET/CT scans in a multi-tracer multicenter environment, is a critical area of research. Different tracers, such as Fluorodeoxyglucose (FDG) and Prostate-Specific Membrane Antigen (PSMA), have distinct physiological uptake patterns and data from different centers often vary in terms of acquisition protocols, scanner types, and patient populations. Because of this variability, it becomes more difficult to design reliable segmentation algorithms and generalization techniques due to variations in image quality and lesion detectability. To address this challenge, We trained a 3D Residual encoder U-Net within the no new U-Net framework, aiming to generalize the performance of automatic lesion segmentation of whole body PET/CT scans, across different tracers and clinical sites. Further, We explored several preprocessing techniques and ultimately settled on using the Total Segmentator to crop our training data. Additionally, we applied resampling during this process. During inference, we leveraged test-time augmentations and other post-processing techniques to enhance tumor lesion segmentation. Our team currently hold the top position in the Auto-PET III challenge and outperformed the challenge baseline model in the preliminary test set with Dice score of 0.9627.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。