Enhancing Renal Tumor Malignancy Prediction: Deep Learning with Automatic 3D CT Organ Focused Attention
Authors: Zhengkang Fan, Chengkun Sun, Russell Terry, Jie Xu, Longin Jan Latecki
Organizations: Department of Health Outcomes & Biomedical Informatics, University of Florida, Gainesville, FL, USA · Department of Computer & Information Sciences, Temple University, Philadelphia, PA, USA · Department of Urology, University of Florida, Gainesville, FL, USA
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.
Figures & tables
Figure 1: Overview of the 3D Organ-Focused Attention (OFA) framework
Institute of Computer Science, University of Tartu, Tartu, Estonia · Institute of Tissue Medicine and Pathology, University of Bern, Bern, Switzerland · Department of Digital Medicine, University of Bern, Bern, Switzerland +4
Charité - Universitätsmedizin Berlin, Department of Radiology, Berlin, Germany · Radboudumc, Diagnostic Image Analysis Group, Nijmegen, The Netherlands · Klinikum rechts der Isar, TUM University Hospital, Technical University of Munich, Munich, Germany +2
Research Ireland Centre for Research Training in Machine Learning · School of Computer Science, University College Dublin, Dublin, Ireland · School of Veterinary Medicine, University College Dublin, Dublin, Ireland