IDGSA-DRIU-Net: Internal dilated guided self-attention renal mass segmentation model based on dilated residual inception U-Net.
Authors
Affiliations (2)
Affiliations (2)
- Department of Computer Science and Systems Engineering, Andhra University College of Engineering (A), Andhra University, Vishakhapatnam, Andhra Pradesh 530003, India. Electronic address: [email protected].
- Department of Computer Science and Systems Engineering, Andhra University College of Engineering (A), Andhra University, Vishakhapatnam, Andhra Pradesh 530003, India.
Abstract
Computed tomography (CT) is essential for finding and diagnosing kidney tumors and cysts because good lesion segmentation enables accurate diagnosis, appropriate therapy planning, and disease monitoring. Renal mass (Tumor and cyst) shapes and sizes are diverse and complex, particularly around diffuse borders, and low-intensity contrast and heterogeneous morphology make effective segmentation difficult. To overcome these challenges, introduced internal dilated guided self-attention model based on dilated residual inception U-Net (IDGSA-DRIU-Net), a deep learning model for segmenting renal tumors and cysts. The architecture features a newly built Internal Dilated Guided Self-Attention (IDGSA) module, which combines Dilated Multiscale Position Attention and Channel Attention introducing an attention mechanism that leverages internal guidance to capture spatial dependencies across multiple dilation scales and emphasizes key feature channels, allowing for more effective utilization of local and global contextual information for better multi-scale feature aggregation. A Dilated Residual Inception (DRI) module improves multiscale contextual feature recovery while preserving structural features. In post-processing, a Conditional Random Field (CRF) is utilized to eliminate false positives and refine bounds. The proposed model was evaluated using the KiTS19, KiTS21, and KiTS23 datasets and obtained a Dice Similarity Coefficient (DSC) of 98.27% for the tumor on KiTS19, 96.31% and 94.82% for the tumor and cyst on KiTS21, and 94.98% and 93.64% for the tumor and cyst on KiTS23. The results suggest that IDGSA-DRIU-Net with CRF outperforms the popular state-of-the-art models, indicating that successful in kidney tumor and cyst segmentation.