Malignancy Detection in Prostate Multi-Parametric MR Images Using U-net with Attention

Abstract

Multiparametric magnetic resonance (mpMR) images are increasingly being used for diagnosis and monitoring of prostate cancer. Detection of malignancy from prostate mpMR images requires expertise, is time consuming and prone to human error. The recent developments of U-net have demonstrated promising detection results in many medical applications. Straightforward use of U-net tends to result in over-detection in mpMR images. The recently developed attention mechanism can help retain only features relevant for malignancy detection, thus improving the detection accuracy. In this work, we propose a U-net architecture that is enhanced by the attention mechanism to detect malignancy in prostate mpMR images. This approach resulted in improved performance in terms of higher Dice score and reduced over-detection when compared to U-net in detecting malignancy.