Ding, M., Lin, Z., Lee, C. H., Tan, C. H., & Huang, W. (2023). A Multi-Scale Channel Attention Network for Prostate Segmentation. IEEE Transactions on Circuits and Systems II: Express Briefs, 70(5), 1754–1758. https://doi.org/10.1109/tcsii.2023.3257728
Abstract:
Prostate cancer is one of the most common malignant tumors in men. Magnetic resonance imaging (MRI) has evolved to an important tool for the diagnosis of prostate cancer. Targeted biopsy is required for accurate diagnosis. This often requires MRI-ultrasound (MRI-US) fusion, as the biopsy is usually performed using transrectal ultrasound. Accurate prostate segmentation on MRI is essential for MRI-US fusion biopsy. However, the variation in prostate shape, appearance, and size makes the automatic segmentation challenging, given the limit of the annotated data. In this paper, we propose a method using multi-scale and Channel-wise Self-Attention (CSA) to recalibrate the feature maps from multiple layers. By embedding the multi-scale CSA on the skip-connection in a UNet structure, called as UCAnet, we show the consistent improvement of the prostate segmentation in Dice, IoU and ASSD. For comparison, we also investigate the single-scale CSA in the networks, and incorporate the vision transformer to test if a transformer would boost the performance. Experiments on a public dataset with 204 prostate MRI scans show that UCAnet achieves the best performance and outperforms the state-of-the-art methods for prostate segmentation such as ENet, UNet, USE-Net and TransUNet.
License type:
Publisher Copyright
Funding Info:
We wish to acknowledge the funding support for this project
from Nanyang Technological University under the URECA
Undergraduate Research Programme.