Detection of Local Prostate Cancer Recurrence from PET/CT Scans Using Deep Learning

TL;DR

This study uses deep learning to detect prostate cancer recurrence from PET scans, achieving 77% accuracy but highlighting the need for further improvements.

Contribution

The study introduces a deep learning model for local prostate cancer recurrence detection using PSMA PET/CT scans and shares pre-trained models for future research.

Findings

Model accuracy reached 77% for post-prostatectomy and non-operated patients.

Restricting the analysis to the bladder area improved accuracy to 71%.

All models showed overfitting with nearly 100% training accuracy but lower validation performance.

Abstract

Prostate cancer is a leading cause of cancer-related deaths in men around the world. A type of imaging technique called positron emission tomography (PET), which uses a special scan to detect cancer, has shown great promise in identifying recurring prostate cancer and spread to other parts of the body. In this study, we created a computer-based model that uses PET scan images to predict if prostate cancer has come back after treatment. To improve the model’s performance, we tried different methods, such as focusing on different parts of the image, adding extra information from the patient’s medical history, and including details about whether the patient had prior surgery to remove the prostate. These efforts led to an accuracy of 77% in predicting cancer recurrence. While this accuracy was lower than the desired 90%, the model still showed significant improvement. Many approaches were tested, each helping to improve the model. The results of this study are an important step forward in developing tools that can reliably detect cancer recurrence in the prostate area. However, more research is required to further improve the model’s accuracy and make it more useful for doctors in real-life situations.

Background: Prostate cancer (PC) is a leading cause of cancer-related deaths in men worldwide. PSMA-directed positron emission tomography (PET) has shown promising results in detecting recurrent PC and metastasis, improving the accuracy of diagnosis and treatment planning. To evaluate an artificial intelligence (AI) model based on [18F]-prostate specific membrane antigen (PSMA)-1007 PET datasets for the detection of local recurrence in patients with prostate cancer. Methods: We retrospectively analyzed 1404 [18F]-PSMA-1007 PET/CTs from patients with histologically confirmed prostate cancer. Artificial neural networks were trained to recognize the presence of local recurrence based on the PET data. First, the hyperparameters were optimized for an initial model (model A). Subsequently, the bladder was localized using an already published model and a model (model B) was trained only on a 20 cm cube around the bladder. Finally, two separate models were trained on the same section depending on the prostatectomy status (model C (post-prostatectomy) and model D (non-operated)). Results: Model A achieved an accuracy of 56% on the validation data. By restricting the region to the area around the bladder, Model B achieved a validation accuracy of 71%. When validating the specialized models according to prostatectomy status, model C achieved an accuracy of 77% and model D an accuracy of 77%. All models achieved accuracies of almost 100% on the training data, indicating overfitting. Conclusions: For the presented task, 1404 examinations were insufficient to reach an accuracy of over 90% even when employing data augmentation, including additional metadata and performing automated hyperparameter optimization. The low F1-score and AUC values indicate that none of the presented models produce reliable results. However, we will facilitate future research and the development of better models by openly sharing our source code and all pre-trained models for transfer learning.

Figures