MRI-based Prostate Cancer Detection Using Cross-Shaped Windows Transformer

Yuheng Li,Jacob Wynne,Jing Wang,Richard L. J. Qiu,Justin Roper,Shaoyan Pan,Ashesh B. Jani,Tian Liu,Pretesh R. Patel,Hui Mao,Xiaofeng Yang
DOI: https://doi.org/10.1117/12.3006812
2024-01-01
Abstract:Prostate multiparametric magnetic resonance imaging (mpMRI) has demonstrated promising results in prostate cancer (PCa) detection using deep learning models using convolutional neural networks (CNNs). Recently, transformers have achieved competitive performance compared to CNNs in computer vision. Large-scale transformers benefit from training with large-scale annotated data, which are expensive and labor-intensive to obtain in medical imaging. Self-supervised learning can effectively leverage unlabeled data to extract useful semantic representations with no additional annotation cost. This can improve model performance on downstream tasks with limited labelled data and increase model robustness to external data. We present a novel end-to-end cross-shaped transformer model CSwin UNet to detect clinically significant prostate cancer (csPCa) in prostate bi-parametric MR imaging (bpMRI). Using a large prostate bpMRI dataset with 1500 patients, our Cswin UNet achieves 0.880 +/- 0.013 AUC and 0.790 +/- 0.033 pAUC, significantly outperforming state-of-the-art CNN and transformer models.
What problem does this paper attempt to address?