用解剖信息提升淋巴瘤病灶检测准确率
Anatomy-Aware Lymphoma Lesion Detection in Whole-Body PET/CT
- 在深度学习模型中加入器官分割图作为解剖先验
- 基于CNN的模型检测性能显著提升,而Transformer效果不变
- 解剖上下文对传统CNN更关键,适合医学影像检测研究者
早期癌症检测对改善患者预后至关重要,18F FDG PET/CT通过结合代谢与解剖信息发挥重要作用。由于需识别大小不一的多个病灶,准确检测仍具挑战。本研究探讨在基于深度学习的病灶检测模型中引入解剖先验信息的效果。具体地,将TotalSegmentator工具生成的器官分割掩码作为辅助输入,提供给nnDetection(当前最优病灶检测框架)和Swin Transformer。后者采用两阶段训练:自监督预训练加有监督微调。方法在AutoPET和Karolinska淋巴瘤数据集上测试。结果表明,解剖先验显著提升了nnDetection框架的检测性能,但对Swin Transformer几乎无影响。此外,Swin Transformer并未明显优于nnDetection中使用的传统卷积神经网络编码器。这些发现突显了解剖上下文在癌症病灶检测中的关键作用,尤其在基于CNN的模型中。
原文摘要 · Abstract (English)
Early cancer detection is crucial for improving patient outcomes, and 18F FDG PET/CT imaging plays a vital role by combining metabolic and anatomical information. Accurate lesion detection remains challenging due to the need to identify multiple lesions of varying sizes. In this study, we investigate the effect of adding anatomy prior information to deep learning-based lesion detection models. In particular, we add organ segmentation masks from the TotalSegmentator tool as auxiliary inputs to provide anatomical context to nnDetection, which is the state-of-the-art for lesion detection, and Swin Transformer. The latter is trained in two stages that combine self-supervised pre-training and supervised fine-tuning. The method is tested in the AutoPET and Karolinska lymphoma datasets. The results indicate that the inclusion of anatomical priors substantially improves the detection performance within the nnDetection framework, while it has almost no impact on the performance of the vision transformer. Moreover, we observe that Swin Transformer does not offer clear advantages over conventional convolutional neural network (CNN) encoders used in nnDetection. These findings highlight the critical role of the anatomical context in cancer lesion detection, especially in CNN-based models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。