SEALS-Net: hybrid loss-guided segmentation and uncertainty based bayesian GRU classification for liver tumor
摘要
Recently, liver tumor segmentation gained significant attention in medical imaging by serving as key step for accurate disease assessment, and precise treatment planning. However, segmentation remained challenging for weakly defined tumors with irregular boundaries that blend in surrounding tissues like muscles and fat, often leading to inaccuracies. To address this issue, Squeeze–Excitation Attention-based Liver Segmentation Network (SEALS-Net) is proposed, by integrating the attention mechanisms to assign precise importance scores to feature maps for improved tumor delineation. The network is trained by using hybrid loss function combining Jaccard and Dice losses, which effectively captures overlapping regions and enhancing segmentation accuracy and then optimized with AdamW optimizer for stable convergence. For classification, Uncertainty based Bayesian Gated Recurrent Unit (UBGRU) model is employed, by utilizing GRU’s ability to capture spatial dependencies in lesion features while Bayesian modeling introduces uncertainty estimation, improving robustness and handling noisy data which are critical for reliable medical decision-making. The experimental results demonstrate the superiority of proposed approach, by achieving a Dice Similarity Coefficient of 96.85% and a Jaccard score of 93.48%, outperforming existing modified SegNet in liver tumor segmentation tasks.