BIF-Net: Boundary information fusion network for abdominal aortic aneurysm segmentation

Comput Biol Med. 2024 Oct 10:183:109191. doi: 10.1016/j.compbiomed.2024.109191. Online ahead of print.

Abstract

The accurate abdominal aortic aneurysm (AAA) segmentation is significant for assisting clinicians in diagnosis and treatment planning. However, existing segmentation methods exhibit a low utilization rate for the semantic information of vessel boundaries, which is disadvantageous for segmenting AAA with significant scale variability of vessel diameter (diameter ranges from 4 mm to 85 mm). To tackle this problem, we introduce a boundary information fusion network (BIF-Net) specially designed for AAA segmentation. BIF-Net initially constructs convolutional kernel functions based on Gabor and Sobel operators, enriching the global semantic features and localization information through the Gabor and Sobel dilated convolution (GSDC) module. Additionally, BIF-Net supplements lost boundary feature information during the sampling process through the guided filtering feature supplementation (GFFS) module and the channel-spatial attention module (CSAM), enhancing the ability to capture targets with shape diversity and boundary features. Finally, we introduce a boundary feature loss function to alleviate the impact of the imbalance between positive and negative samples. The results demonstrate that BIF-Net outperforms current state-of-the-art methods across multiple evaluation metrics, achieving the highest Dice similarity coefficient (DSC) accuracies of 93.29 % and 91.01 % on the preoperative and postoperative datasets, respectively. Compared to the state-of-the-art methods, BIF-Net improves DSC by 6.86 % and 3.85 %. Due to the powerful boundary feature extraction ability, the proposed BIF-Net is a competitive AAA segmentation method exhibiting significant potential for application in diagnosis and treatment processes of AAA.

Keywords: AAA segmentation; Boundary feature loss function; Boundary information fusion; Gabor and Sobel operator; Guided filter.