Anatomy-aware fusion attention network for high-precision prostate MRI segmentation.
Authors
Affiliations (4)
Affiliations (4)
- College of Information, Mechanical and Electrical Engineering, Shanghai Engineering Research Center of Intelligent Education and Bigdata, Shanghai Normal University, Shanghai, China.
- School of Information Science and Engineering, East China University of Science and Technology, Shanghai, China.
- School of Computer Science and Engineering, Southeast University, Nanjing, Jiangsu, China.
- Department of Radiology, Shanghai Pudong New Area Gongli Hospital, Shanghai, China.
Abstract
Accurate prostate segmentation in Magnetic Resonance Imaging (MRI) is crucial for clinical diagnosis and treatment. However, it remains a challenging task due to low soft-tissue contrast and structural ambiguity in the gland's appearance. While recent deep learning methods have focused on refining network structures, they often neglect the inherent anatomical consistency of the prostate. To address this limitation, a novel segmentation framework integrating an Anatomy-Aware Fusion Attention (AAFA) module is proposed. By leveraging anatomical templates, our approach establishes a multilevel feature cross-attention mechanism that enhances global contextual modeling of prostate regions. Additionally, we introduce a Phased Learning strategy that progressively trains the model to mitigate the adverse effects of invalid or noisy samples commonly found in clinical MRI data. Extensive ablation studies on the PROMISE12 dataset validate the contribution of each component to overall performance. Comparison experiments on both PROMISE12 and MSD Prostate datasets show that our method consistently outperforms existing approaches across key metrics, such as Dice similarity coefficient (DSC), Intersection over Union (IoU), Precision and 95% Hausdorff distance (HD95). These results confirm the robustness and strong generalization capability of the proposed framework in challenging clinical segmentation tasks.