解决氟代脱氧葡萄糖到前列腺特异性膜抗原的3D病灶检测跨示踪剂迁移问题。
Unsupervised Adaptation from FDG to PSMA PET/CT for 3D Lesion Detection under Label Shift
- 通过自训练动态调整检测框尺度,适应目标域小病灶特征。
- 按病灶体积分桶分配伪标签数量,缓解目标域病灶大小分布差异。
- 在自动标注数据集上显著提升准确率和检出率,适合医学影像跨模态应用。
本文提出一种用于3D体积分割病灶检测的无监督域适应(UDA)框架,将基于标记的氟代脱氧葡萄糖(FDG)PET/CT训练的检测器适配至未标记的前列腺特异性膜抗原(PSMA)PET/CT。除协变量偏移外,跨示踪剂适配还存在病灶大小分布与每例患者病灶数量的标签偏移。我们引入两种机制显式建模并补偿该标签偏移:首先,通过选定伪标签重新估计目标域边界框尺度,并使用指数移动平均更新锚框,提升对小病灶的正锚覆盖并稳定回归;其次,不采用固定置信度阈值选择伪标签,而是根据估计的目标域病灶体积直方图,按体积分桶分配伪标签名额。自训练交替进行基于先验引导的伪标签化PSMA监督学习与标注的FDG监督学习。在AutoPET 2024数据集上,从501例标记的FDG研究适配至369例$^{18}$F-PSMA研究,所提方法在平均精度(AP)和受试者工作特征曲线下面积(FROC)上均优于源模型基线及无标签偏移缓解的常规自训练,表明建模目标域病灶流行率与大小构成是实现鲁棒跨示踪剂检测的有效路径。
原文摘要 · Abstract (English)
In this work, we propose an unsupervised domain adaptation (UDA) framework for 3D volumetric lesion detection that adapts a detector trained on labeled FDG PET/CT to unlabeled PSMA PET/CT. Beyond covariate shift, cross tracer adaptation also exhibits label shift in both lesion size composition and the number of lesions per subject. We introduce self-training with two mechanisms that explicitly model and compensate for this label shift. First, we adaptively adjust the detection anchor shapes by re-estimating target domain box scales from selected pseudo labels and updating anchors with an exponential moving average. This increases positive anchor coverage for small PSMA lesions and stabilizes box regression. Second, instead of a fixed confidence threshold for pseudo-label selection, we allocate size bin-wise quotas according to the estimated target domain histogram over lesion volumes. The self-training alternates between supervised learning with prior-guided pseudo labeling on PSMA and supervised learning on labeled FDG. On AutoPET 2024, adapting from 501 labeled FDG studies to 369 $^{18}$F-PSMA studies, the proposed method improves both AP and FROC over the source-only baseline and conventional self-training without label-shift mitigation, indicating that modeling target lesion prevalence and size composition is an effective path to robust cross-tracer detection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。