无需手动分割,用注意力机制提升肾肿瘤恶性预测准确率。
Enhancing Renal Tumor Malignancy Prediction: Deep Learning with Automatic 3D CT Organ Focused Attention
- 引入器官聚焦注意力损失,自动让图像块专注器官区域
- 在私有数据集AUC达0.685,公开数据集AUC达0.760
- 适合临床医生快速辅助诊断,减少人工标注负担
准确预测肾肿瘤恶性程度对指导临床决策和优化治疗策略至关重要。然而,现有影像技术难以在手术前可靠预测恶性程度。深度学习虽在3D CT图像上展现潜力,但传统方法依赖人工分割肿瘤区域以降低噪声,提升预测性能。人工分割耗时耗力且依赖专家经验。本研究提出一种基于器官聚焦注意力(OFA)损失函数的深度学习框架,通过调整图像块注意力,使器官区域仅关注其他器官区域,从而无需部署时进行3D肾部CT图像分割。该框架在佛罗里达大学综合数据仓库(UF IDR)私有数据集上达到AUC 0.685、F1-score 0.872;在公开的KiTS21数据集上达到AUC 0.760、F1-score 0.852。结果优于依赖分割裁剪的传统模型,证明该方法可在不依赖显式分割的情况下提升预测准确性,为肾癌诊断提供更高效可靠的临床决策支持。
原文摘要 · Abstract (English)
Accurate prediction of malignancy in renal tumors is crucial for informing clinical decisions and optimizing treatment strategies. However, existing imaging modalities lack the necessary accuracy to reliably predict malignancy before surgical intervention. While deep learning has shown promise in malignancy prediction using 3D CT images, traditional approaches often rely on manual segmentation to isolate the tumor region and reduce noise, which enhances predictive performance. Manual segmentation, however, is labor-intensive, costly, and dependent on expert knowledge. In this study, a deep learning framework was developed utilizing an Organ Focused Attention (OFA) loss function to modify the attention of image patches so that organ patches attend only to other organ patches. Hence, no segmentation of 3D renal CT images is required at deployment time for malignancy prediction. The proposed framework achieved an AUC of 0.685 and an F1-score of 0.872 on a private dataset from the UF Integrated Data Repository (IDR), and an AUC of 0.760 and an F1-score of 0.852 on the publicly available KiTS21 dataset. These results surpass the performance of conventional models that rely on segmentation-based cropping for noise reduction, demonstrating the frameworks ability to enhance predictive accuracy without explicit segmentation input. The findings suggest that this approach offers a more efficient and reliable method for malignancy prediction, thereby enhancing clinical decision-making in renal cancer diagnosis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。