arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.12697cs.CV
收藏

校准的多样集成熵最小化用于前列腺癌检测中的鲁棒测试时自适应

Calibrated Diverse Ensemble Entropy Minimization for Robust Test-Time Adaptation in Prostate Cancer Detection

University of British Columbia(不列颠哥伦比亚大学);Exact Imaging(埃克斯特成像公司)

Mahdi Gilany, Mohamed Harmanani, Paul Wilson, Minh Nguyen Nhat To, Amoon Jamzad, Fahimeh Fooladgar, Brian Wodlinger, Purang Abolmaesumi, Parvin Mousavi

AI总结:

针对超声前列腺癌检测中分布偏移导致的性能下降,提出多样集成熵最小化(DEnEM)方法,通过校准集成熵最小化提升测试时自适应鲁棒性,AUROC较现有方法提高5%-7%。

摘要:

高分辨率微型超声在实时前列腺癌检测中展现出前景,深度学习已成为学习超声上反映的复杂组织特性的重要工具。然而,先前工作常忽视的一个现实部署的重大障碍仍然存在:当模型应用于来自不同临床中心的数据时,由于数据分布的变化,模型性能会下降。这种分布偏移显著影响模型的鲁棒性,对临床部署构成重大挑战。域自适应,特别是其测试时自适应(TTA)变体,为应对这一挑战提供了有前景的解决方案。在旨在反映真实世界条件的设置中,我们将现有方法与为癌症检测采用的最先进的TTA方法进行比较,证明了前者对分布偏移缺乏鲁棒性。然后,我们提出了多样集成熵最小化(DEnEM),质疑当前TTA方法在超声数据上的有效性。我们表明,这些方法虽然优于基线,但由于依赖神经网络输出概率(可能未校准)或依赖数据增强(在超声数据上不易定义),因此是次优的。我们的结果显示,与现有方法相比,AUROC显著提高了5%至7%,与TTA方法相比提高了3%至5%,证明了DEnEM在解决分布偏移方面的优势。

英文摘要:

High resolution micro-ultrasound has demonstrated promise in real-time prostate cancer detection, with deep learning becoming a prominent tool for learning complex tissue properties reflected on ultrasound. However, a significant roadblock to real-world deployment remains, which prior works often overlook: model performance suffers when applied to data from different clinical centers due to variations in data distribution. This distribution shift significantly impacts the model's robustness, posing major challenge to clinical deployment. Domain adaptation and specifically its test-time adaption (TTA) variant offer a promising solution to address this challenge. In a setting designed to reflect real-world conditions, we compare existing methods to state-of-the-art TTA approaches adopted for cancer detection, demonstrating the lack of robustness to distribution shifts in the former. We then propose Diverse Ensemble Entropy Minimization (DEnEM), questioning the effectiveness of current TTA methods on ultrasound data. We show that these methods, although outperforming baselines, are suboptimal due to relying on neural networks output probabilities, which could be uncalibrated, or relying on data augmentation, which is not straightforward to define on ultrasound data. Our results show a significant improvement of $5\%$ to $7\%$ in AUROC over the existing methods and $3\%$ to $5\%$ over TTA methods, demonstrating the advantage of DEnEM in addressing distribution shift. \keywords{Ultrasound Imaging \and Prostate Cancer \and Computer-aided Diagnosis \and Distribution Shift Robustness \and Test-time Adaptation.}

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑