通过测试时解剖结构引导,提升超声图像癌症检测的跨中心适应能力
Learning Prostate Anatomy at Test Time for Cancer Detection in Micro-Ultrasound

- 测试时用前列腺分割任务引导模型适应新设备图像
- 在两个中心数据上分别提升0.029和0.036的AUC值
- 适合临床部署中需跨设备部署的癌症检测系统
不同医疗机构使用不同成像设备或采集协议导致的领域偏移,仍是深度学习用于前列腺癌检测的主要障碍。现有测试时自适应(TTA)方法通过熵最小化或基于增强的自监督纠正图像外观的统计差异,但忽略目标域的解剖结构。本文提出ANT框架,通过在测试时求解辅助前列腺分割任务,利用冻结的预训练分割网络生成伪掩码,引导癌症检测编码器适配目标域。该方法通过对齐目标域中的前列腺解剖结构,纠正领域特异性特征漂移,同时保留癌症判别结构。模型在693名患者(早期微超声扫描仪)数据上训练,评估于118名患者(新型扫描仪,两中心)数据。在留一中心排除协议下,与无自适应相比,ANT在活检核心和患者级别分别提升平均AUC 2.9%和3.6%,优于所有基线方法。代码已开源。
原文摘要 · Abstract (English)
Domain shift across clinical centers using different imaging hardware or acquisition protocols remains a fundamental barrier to deploying deep learning models for prostate cancer (PCa) detection. Existing test-time adaptation (TTA) methods address distribution shift through entropy minimization or augmentation-based self-supervision, correcting for statistical differences in image appearance but ignoring the anatomical structure of the target domain. We propose ANT, a segmentation-guided TTA framework that adapts a pretrained cancer detection encoder to the target domain by solving an auxiliary prostate segmentation task at test time, supervised by pseudo-masks from a frozen pretrained segmentation network. By aligning encoder representations to prostate anatomy in the target domain, ANT corrects domain-specific feature drift while preserving cancer-discriminative structure. The model was trained on 693 patients imaged with an earlier-generation micro-ultrasound scanner in a multi-center clinical trial, and evaluated on 118 patients acquired with a newer-generation system across two centers in another clinical trial. Under a leave-one-center-out protocol with identical evaluation conditions across all methods, ANT improves mean AUC by 2.9% and 3.6% at the biopsy-core and patient levels, respectively, over no adaptation, outperforming TTA baselines. Code is available at: https://github.com/ObedDzik/ant.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。