Deep Learning Model with Attention Mechanism for a 3D Pancreas Segmentation in CT Scans

Accurate segmentation of the pancreas in Computed Tomography (CT) scans is a challenging task, which may be crucial for the diagnosis and treatment planning of pancreatic cancer. The irregular shape of the pancreas, its low contrast relative to surrounding tissues, and its close proximity to other complex anatomical structures make it difficult to accurately delineate its contours. Furthermore, a significant class imbalance between foreground (pancreas) and background voxels in an abdominal CT series represents an additional challenge for deep-learning-based approaches. In this study, we developed a deep learning model for automated pancreas segmentation based on a 3D U-Net architecture enhanced with an attention mechanism to improve the model capability to focus on relevant anatomical features of the pancreas. The model was trained and evaluated on two widely used benchmark datasets for volumetric segmentation, the NIH Healthy Pancreas-dataset and the Medical Segmentation Decathlon (MSD) pancreas dataset. The proposed attention-guided 3D U-Net achieved a Dice score of 80.8 ± 2.1%, ASSD of 2.1 ± 0.3 mm, and HD95 of 8.1 ± 1.6 mm on the NIH dataset, and the values of 78.1 ± 1.1%, 3.3 ± 0.3 mm, and 12.3 ± 1.5 mm for the same metrics on the MSD dataset, demonstrating the value of attention mechanisms in improving segmentation performance in complex and low-contrast anatomical regions.