iPCa-Former: A Multi-Task Transformer Framework for Perceiving Incidental Prostate Cancer

Xianwei Pan,Simiao Wang,Yunan Liu,Lijie Wen,Mingyu Lu
DOI: https://doi.org/10.1109/lsp.2024.3372787
2024-03-22
IEEE Signal Processing Letters
Abstract:Despite significant progress in medical image analysis using deep learning, predicting incidental prostate cancer (iPCa) remains challenging due to subtle differences in multiparametric magnetic resonance imaging (mpMRI) and a lower incidence rate. To address these challenges, we propose iPCa-Former, a transformer-based framework designed to enhance iPCa prediction within prostate mpMRI slices. Firstly, built on an encoder-decoder architecture, our iPCa-Former facilitates the simultaneous optimization of two tasks through mutual learning: prostate transition zone segmentation and iPCa prediction. Secondly, we introduce a joint optimization function that combines focal loss and boundary-based mutual information (BMI) loss, effectively addressing the imbalance of positive and negative samples in classification and the challenge posed by a small proportion of the foreground region in segmentation. Moreover, we construct an iPCa mpMRI dataset comprising 10,276 prostate mpMRI slices from 485 patients clinically diagnosed with benign prostatic hyperplasia; however, 27 out of these patients are identified as iPCa. When evaluated on this benchmark dataset, our iPCa-Former outperforms state-of-the-art methods, demonstrating the superior performance of our approach.
engineering, electrical & electronic
What problem does this paper attempt to address?