Attention-enhanced hybrid U-Net for prostate cancer grading and explainability.
Authors
Affiliations (6)
Affiliations (6)
- BOYA International College, Jiangxi University of Technology, 115 Ziyang Avenue, Nanchang, 330022, Jiangxi, China. [email protected].
- Department of Computer Science, COMSATS University Islamabad, Sahiwal Campus, Sahiwal, Pakistan.
- BOYA International College, Jiangxi University of Technology, 115 Ziyang Avenue, Nanchang, 330022, Jiangxi, China.
- College of Finance and Economics, Jiangxi University of Technology, 115 Ziyang Avenue, Nanchang, 330022, Jiangxi, China.
- Department of Information Science, College of Humanities and Social Sciences, King Saud University, Riyadh, Saudi Arabia.
- Department of Information Technology, College of Computers and Information Technology, Taif University, P.O. Box 11099, 21944, Taif, Saudi Arabia.
Abstract
Prostate cancer remains a leading cause of mortality, necessitating precise histopathological segmentation for accurate Gleason Grade assessment. However, existing deep learning-based segmentation models lack contextual awareness and explainability, leading to inconsistent performance across heterogeneous tissue structures. Conventional U-Net architectures and CNN-based approaches struggle with capturing long-range dependencies and fine-grained histopathological patterns, resulting in suboptimal boundary delineation and model generalizability. To address these limitations, we propose a transformer-attention hybrid U-Net (TAH U-Net), integrating hybrid CNN-transformer encoding, attention-guided skip connections, and a multi-stage guided loss mechanism for enhanced segmentation accuracy and model interpretability. The ResNet50-based convolutional layers efficiently capture local spatial features, while Vision Transformer (ViT) blocks model global contextual dependencies, improving segmentation consistency. Attention mechanisms are incorporated into skip connections and decoder pathways, refining feature propagation by suppressing irrelevant tissue noise while enhancing diagnostically significant regions. A novel hierarchical guided loss function optimizes segmentation masks at multiple decoder stages, improving boundary refinement and gradient stability. Additionally, Explainable AI (XAI) techniques such as LIME, Occlusion Sensitivity, and Partial Dependence Analysis (PDP), validate the model's decision-making transparency, ensuring clinical reliability. The experimental evaluation on the SICAPv2 dataset demonstrates state-of-the-art performance, surpassing traditional U-Net architectures with a 4.6% increase in Dice Score, 5.1% gain in IoU, along with notable improvements in Precision (+ 4.2%) and Recall (+ 3.8%). This research significantly advances AI-driven prostate cancer diagnostics by providing an interpretable and highly accurate segmentation framework, enhancing clinical trust in histopathology-based grading within medical imaging and computational pathology.