Spatiotemporal Mammography-based Deep Learning Model for Improved Breast Cancer Risk Prediction.

Abstract

Breast cancer is one of the most prevalent cancers among women. It is the second leading cause of death in cancer-related deaths. Early detection and personalized risk assessment can reduce the mortality rate and improve survival rates. Classical risk prediction models which rely on traditional risk factors produce inconsistent results among the different populations. Thus, they are not routinely used in screening programs. Deep learning was proven to improve the results of breast cancer risk prediction. CNNs can detect risk cues from screening mammograms. However, the deep learning models utilize the spatial information of each screening mammogram independently. This study aims to further improve the risk prediction models by exploiting the spatiotemporal information in multiple screening time points. We implemented a Siamese neural network for spatiotemporal risk prediction and compared the results against CNN trained using two different time points (T1 and T2) independently. We tested our results on 191 cases, 61 of which were diagnosed with cancer. The Siamese model showed a superior AUC of 0.81 against 0.75 and 0.77 at T1 and T2 respectively. The Siamese network also exhibited higher accuracy and F1-score with values of 0.78 and 0.61 while CNNs have the same accuracy of 0.76 with an F1-score of 0.54 at T1, and 0.59 at T2. The results suggest that spatiotemporal risk prediction can be a more reliable risk assessment tool.