Abstract
Vessel segmentation is crucial for assisting in the diagnosis and treatment of a range of diseases, such as retinal diseases and coronary artery diseases. However, current vessel segmentation methods often face the problem of poor vessel boundary segmentation, making the vessel segmentation task still challenging. This study aims to alleviate this problem, thereby improving the vessel segmentation results. By introducing a boundary detection auxiliary task to the vessel segmentation main task, we propose a boundary perception enhancement network (BPENet) for retinal vessel and coronary angiogram segmentation. Among them, to enhance vessel feature extraction capability, BPENet introduces the feature extraction enhancement (FE2) module. To enhance vessel boundary perception capability, BPENet introduces the boundary enhancement (BE) module. In addition, to fully leverage feature information from different layers, BPENet introduces the deep feature aggregation (DFA) module. Experimental results on retinal vessel datasets (DRIVE, CHASE-DB1, and STARE) and coronary angiogram datasets (DCA1 and XCAD) show that BPENet outperforms the existing mainstream segmentation methods. BPENet is expected to provide a reliable vessel segmentation result for doctors during the diagnosis and treatment of related diseases.
Keywords
Introduction
The vessel system plays a crucial role in nourishing other organs and tissues of the human body. 1 Among them, retinal vessels are the only vessels of the human body that can be directly observed by non-invasive methods. Its morphological characteristics (e.g., branching angle, branching pattern, length, tortuosity, and width) are utilized to diagnose, screen, treat, and evaluate a wide range of ophthalmologic diseases (e.g., glaucoma and diabetic retinopathy, etc.). 2 Coronary arteries are vessels that supply blood to the heart. Its morphological characteristics (e.g., branching angle, diameter, length, and tortuosity) can reveal initial coronary artery disease symptoms.3,4 In traditional diagnosis and treatment processes, vessel segmentation is mainly performed manually by experienced professional doctors.5,6 Due to the complexity of vessel structures, the professional level of the doctors also affects the segmentation results. Therefore, manual segmentation of vessels is a very challenging task. Automated computer segmentation technology can significantly optimize medical processes, not only saving valuable human resources but also helping doctors quickly and accurately complete disease screening, improving overall medical efficiency and quality. Therefore, researching and designing an accurate automated vessel segmentation approach is crucial for assisting doctors in diagnosing and treating related diseases, including retinal and coronary artery diseases.
In recent years, deep learning (DL) has been widely applied to various medical image processing tasks, including breast abnormality recognition, 7 skin lesion recognition, 8 liver tumor segmentation, 9 Alzheimer's disease diagnosis, 10 and COVID-19 lesion segmentation. 11 Convolutional neural networks (CNN) have become one of the mainstream architectures in DL due to their outstanding capabilities in image feature extraction. Among them, FCN 12 has notable contributions to semantic segmentation. It pioneered a CNN-based pixel-level prediction method, providing a basic framework for subsequent semantic segmentation methods. Then emerged U-Net, 13 which uses an encoder-decoder architecture with skip connections and employs traditional square convolutions for feature extraction, achieving significant success in medical image segmentation. Inspired by U-Net, many researchers have improved upon it, thus proposing various models for vessel segmentation. In order to decrease parametric quantities in the original U-Net, M2U-Net 14 mainly employed depthwise separable convolutions for feature extraction. In order to expand the receptive field, AMF-Net 15 mainly employed atrous convolutions for feature extraction. As one of the core parts of a vessel segmentation model, the feature extraction method is crucial for a model. In this paper, the feature extraction enhancement (FE2) module, which mainly comprises four-angle strip convolution (FSC) and adaptive channel selection (ACS), is employed to enhance the model's vessel feature extraction capability, thereby improving the model segmentation performance.
Multi-task learning can enhance model performance through mutual promotion, so this approach is also employed in vessel segmentation studies. Ma et al. 16 designed a network capable of concurrently performing artery/vein classification and vessel segmentation tasks. This network introduced a multi-task output block incorporating the spatial activation mechanism, which leverages the outcome of the relatively easy vessel segmentation task to boost the outcome of the artery/vein classification task. DMAN 17 was a network that integrated quantification and segmentation tasks for main coronary vessels. Among them, quantification can serve as a global restraint to enhance segmentation performance, while segmentation can indicate stenosis lesion locations to gain better quantification accuracy. Furthermore, in the field of building extraction from remote sensing images, some studies have treated boundary detection as one of the tasks in multi-task learning to improve boundary extraction effects. For example, Xu et al. 18 adopted multi-task learning strategies to capture more boundary information by introducing the additional boundary detection decoder. BFL-Net 19 also introduced the additional boundary detection decoder, differing from Xu et al. 18 in that BFL-Net fused the outputs of different decoder branches. Specifically, it concatenated the output feature maps from the segmentation and boundary detection decoder branches and then applied convolution operations to obtain the building extraction results. However, few researchers focus on the problem of poor boundary segmentation in existing vessel segmentation methods. Given the advantages of multi-task learning and the use of boundary detection as an auxiliary task in other fields, this paper introduces a boundary detection auxiliary task to the vessel segmentation main task. In this way, the vessel boundary detection auxiliary task's powerful extraction ability for vessel boundary information can be utilized to enhance the vessel segmentation main task's extraction ability for vessel boundary information, thus alleviating the problem of poor vessel boundary segmentation. In medical image segmentation, attention mechanisms are extensively applied because of their ability to focus on important information. In order to increase sensitivity to information features, DEU-Net 20 introduced the attention skip module with channel attention in U-Net's skip connections. In order to strengthen retinal vessel features in spatial features and highlight salient features transmitted through skip connections, AAU-net 21 introduced gate attention in U-Net's skip connections. Through existing studies, it is seen that the introduction of attention mechanisms in skip connections can improve the model's performance. Given this, this paper innovatively introduces the boundary enhancement (BE) module with boundary attention into the skip connections, which can also alleviate the problem of poor vessel boundary segmentation. Furthermore, unlike BFL-Net, 19 the BE module doesn’t directly fuse the feature maps from the boundary branch with those from the segmentation branch, but instead fuses the boundary detection results into the segmentation task as a boundary attention map. This design allows boundary information to participate in segmentation decisions through attention guidance rather than feature supplementation, so that the segmentation main task has stronger boundary perception capabilities.
Besides attention mechanisms that enhance model performance, the fusion of features from different layers of the U-Net decoder can better utilize multi-scale information, thus improving the model's capability of understanding and segmenting complex images. In order to prevent feature information loss, AMF-Net 15 presented a multi-scale fusion (MSF) module, by upsampling feature maps from each decoder stage to the model input image size using bilinear interpolation and then obtaining the output of this module after concatenation and 1 × 1 convolution (1 × 1 Conv) operations. Although the MSF module can improve model performance, its feature fusion process is so simple that it does not fully leverage feature information from different layers. Given this, this paper designs the deep feature aggregation (DFA) module to further improve model performance.
U-Net is an interesting innovation.
22
Through the aforementioned research work, it can also be noted that the U-Net architecture is very suitable for the vessel segmentation task, but there is still a lot of room for improvement. Given this, we improve on U-Net, thus proposing a boundary perception enhancement network (BPENet) for retinal vessel and coronary angiogram segmentation. Firstly, BPENet introduces the FE2 module to strengthen the model's vessel feature extraction capability. This module mainly comprises FSC and ACS. Among them, FSC primarily employs strip convolutions
23
at four angles (

Comparison of vessel boundary segmentation effects between BPENet and the baseline model. (a1) and (a2) Original image, (b1) and (b2) Ground truth (GT), (c1) and (c2) Baseline model, (d1) and (d2) BPENet (ours).
The main contributions of this paper are as follows:
We propose a boundary perception enhancement network (BPENet) that introduces a boundary detection auxiliary task to the vessel segmentation main task. Experimental results on five datasets show that BPENet is highly competitive in the vessel segmentation task. We introduce the feature extraction enhancement (FE2) module to enhance the model's vessel feature extraction capability, which mainly comprises four-angle strip convolution (FSC) and adaptive channel selection (ACS). We introduce the boundary enhancement (BE) module with boundary attention to enhance the model’s vessel boundary perception capability. We introduce the deep feature aggregation (DFA) module to fully leverage feature information from different layers, which mainly generates richer and more comprehensive feature representations by deeply fusing features from different layers.
Methodology
Overall overview
This paper introduces the FE2, BDAB, BE, and DFA modules to the U-Net, thereby proposing BPENet. As shown in Figure 2, BPENet mainly consists of the encoder, decoder, skip connections, and BDAB. Among them, to enhance vessel feature extraction capability, we employ the FE2 module mainly comprising strip convolutions at four angles (

Network architecture of BPENet. This model uses FE2 to replace U-Net's convolution blocks, thus strengthening the model's vessel feature extraction capability. Next, BDAB is added to the encoder for vessel boundary detection. Then, BE is introduced in the skip connections to strengthen the model's vessel boundary perception capability. Finally, DFA is added to the decoder to fully leverage feature information from different layers.
Feature extraction enhancement (FE2) module
It is crucial for a model to have excellent feature extraction capability. BPENet introduces the FE2 module to enhance the model's vessel feature extraction capability. As shown in Figure 3, the FE2 module mainly comprises FSC and ACS, where four angles refer to the

FE2 module. This module mainly consists of four-angle strip convolution (FSC) and adaptive channel selection (ACS), where FSC mainly comprises strip convolutions at four angles:
ACS is mainly a dual-branch channel attention. Next, F′ passes through ACS, as described in detail:
Finally, F″ and F perform residual connection and get Fout by 1 × 1 Conv, BN, ReLU, and Dropout operations. The specific process is described as follows:
Boundary detection auxiliary branch (BDAB)
Because multi-task learning can improve the model performance through mutual promotion, so combining vessel segmentation and vessel boundary detection for multi-task learning, the vessel boundary detection auxiliary task's powerful extraction ability for vessel boundary information can be utilized to enhance the vessel segmentation main task's extraction ability for vessel boundary information. Based on this, we add the auxiliary branch in the encoder for vessel boundary detection, as shown in Figure 2(a). Firstly, an encoder deep feature map sequentially passes through the FE2 module and bilinear interpolation (upsampling by 4 times), followed by concatenating the feature map obtained from upsampling with the encoder shallow layer feature map along the channel dimension. Finally, the concatenated feature map sequentially goes through the FE2 module, bilinear interpolation (upsampling by 2 times), 1 × 1 Conv, and sigmoid function to get the vessel boundary prediction result.
Boundary enhancement (BE) module
Attention mechanisms can highlight important features while suppressing unimportant features. To enhance the model's vessel boundary perception capability, we introduce the BE module. As shown in Figure 4, the BE module mainly takes the boundary prediction results obtained from BDAB as boundary attention and introduces them to the skip connections, which enables the segmentation main network to focus more on the vessel boundary regions and capture the boundary information more effectively, thereby improving the boundary segmentation effects. Firstly, encoder feature map E is element-wise multiplied with boundary attention map Mb to obtain feature map E′. Then, E′ and E perform concatenation. Finally, the concatenated feature map sequentially passes through 1 × 1 Conv, BN, and ReLU to get Eout. The overall details of the BE module are described as follows:

BE module. This module mainly utilizes boundary attention to enhance the model's vessel boundary perception capability.
Deep feature aggregation (DFA) module
Feature maps from different layers carry different feature information. By fusing these information, the model can understand the image content more comprehensively, thereby enhancing model performance. To fully leverage feature information from different layers, we introduce the DFA module in the decoder. As shown in Figure 5, the DFA module deeply fuses features from multiple decoder layers and combines ACS that can better attend to the significant channel features.

DFA module. This module deeply fuses features from multiple decoder layers and combines ACS that can focus on important channel features.
Firstly, the decoder feature maps D1, D2, and D3 by using bilinear interpolation to upsample them by 8, 4, and 2 times, respectively, to obtain feature maps
Next, some convolution and concatenation operations are performed in
Next,
Finally,
Experimental setup
Datasets
BPENet was evaluated on three retinal vessel datasets, DRIVE, 24 CHASE-DB1, 25 and STARE, 26 as well as two coronary angiogram datasets, DCA1 27 and XCAD. 28 If the dataset image has two expert's annotation maps, we employ the first expert's annotation map as ground truth (GT). The images in the DRIVE, CHASE-DB1, and STARE datasets are all color images, with 40, 28, and 20 annotated images, respectively. The images in the DCA1 dataset are all grayscale images, with 134 annotated images. The images in the XCAD dataset are all 1 channel, with 126 annotated images. The DRIVE dataset has been officially divided into training and test sets. We use the first 20, 16, 100, and 100 images of CHASE-DB1, STARE, DCA1, and XCAD datasets as the training set and the last 8, 4, 34, and 26 as the test set, respectively. Detailed information about the five datasets is illustrated in Table 1.
Dataset information.
Data preprocessing
The image preprocessing process consists of converting images to grayscale images (if dataset images are color images), normalization, contrast limited adaptive histogram equalization (CLAHE) to strengthen contrast, and gamma correction to deal with the unevenness of image brightness. In addition, for boundary (or contour) labeled maps acquisition, we draw on the approach adopted by Xu et al. 18 and Liu et al.. 19 Specifically, we go through the findContours function of OpenCV to get corresponding vessel boundary labeled maps from vessel segmentation labeled maps of training samples.
For training samples of DRIVE, CHASE-DB1, STARE, and DCA1, we crop the overlapping blocks from the preprocessed training set images to expand the training sample numbers using the sliding cropping window with a stride of 24 and a size of 128 × 128. For training samples of XCAD, we crop the overlapping blocks from the preprocessed training set images to expand the training sample numbers using the sliding cropping window with a stride of 36 and a size of 128 × 128. For test samples of DRIVE, CHASE-DB1, STARE, and DCA1, we perform the 0-value padding on the right and bottom sides of the images to fill the resolution sizes to 608 × 608, 1024 × 1024, 704 × 704, and 320 × 320, respectively. For test samples of XCAD, we do not perform the 0-value padding process.
We process the overlapping blocks by random vertical flipping, horizontal flipping, and rotations of
Evaluation metrics
In order to comprehensively evaluate BPENet, we employ five evaluation metrics: accuracy (ACC), F1-score (F1), area under curve (AUC), sensitivity (SE), and specificity (SP). Among them, AUC denotes the area under the receiver operating characteristic (ROC) curve, while ACC, F1, SE, and SP can be expressed as formulas:
Implementation details
BPENet was implemented by PyTorch 1.7.1 framework, and the related experiments were performed on a single NVIDIA RTX A6000 with 48GB of GPU memory. During training, we set the batch size to 100 and employed Adam as the optimizer. For experiments on DRIVE, CHASE-DB1, DCA1, and XCAD, the initial learning rate and weight decay were both set to 1e-3. For experiments on STARE, the initial learning rate and weight decay were both set to 1e-4. We employed the cosine annealing algorithm to gradually reduce the learning rate over 30 epochs. For testing, we used obtained model parameters in the 30th epoch.
Both the vessel segmentation main task and the boundary detection auxiliary task exhibit class imbalance, while the Dice loss optimizes based on the overlap between predicted results and ground truth labels, exhibiting strong robustness against class imbalance. Furthermore, both tasks are essentially pixel-level classification problems, while the binary cross-entropy (BCE) loss provides fine-grained pixel-level supervision, which helps improve the model's ability to capture local details, particularly in complex regions such as vessel boundaries. Given this, we jointly use both loss functions, specifically as follows:
Experimental results
Retinal vessel segmentation
Table 2 demonstrates performance comparisons of BPENet with advanced methods on DRIVE and CHASE-DB1. Table 3 demonstrates performance comparisons of BPENet with advanced methods on STARE. As shown in Table 2, the values of ACC, F1, AUC, SE, and SP metrics obtained by BPENet on DRIVE are the best, which are 0.9715, 0.8378, 0.9886, 0.8522, and 0.9878, respectively. The values of ACC, SE, and SP metrics obtained by BPENet on CHASE-DB1 are the best, which are 0.9764, 0.8818, and 0.9883, respectively. Although the value of the F1 metric of BPENet on CHASE-DB1 is 0.97% lower compared to DS2TUNet, the values of ACC, SE, and SP metrics of BPENet are higher than DS2TUNet. Among them, the value of the SE metric of DS2TUNet is 7.17% lower compared to BPENet, so considering the values of several metrics together, BPENet performance on CHASE-DB1 is better than DS2TUNet. Although the value of the AUC metric of BPENet on CHASE-DB1 is 0.21% lower compared to AMF-Net, the values of ACC, SE, and SP metrics of BPENet are higher than AMF-Net. Among them, the value of the SE metric of AMF-Net is 4.74% lower compared to BPENet, so considering the values of several metrics together, BPENet performance on CHASE-DB1 is better than AMF-Net. Additionally, although the value of the AUC metric of BPENet on CHASE-DB1 is 0.03% lower compared to ACC-UNet, the values of ACC, F1, SE, and SP of BPENet are higher than ACC-UNet, and the values of these four metrics are higher by more than 0.03%, so considering the values of several metrics together, BPENet performance on CHASE-DB1 is better than ACC-UNet.
Performance comparisons of BPENet with advanced methods on DRIVE and CHASE-DB1.
Bold font indicates the best result of each evaluation metric, and “-” indicates missing values of the evaluation metric.
Performance comparisons of BPENet with advanced methods on STARE.
Bold font indicates the best result of each evaluation metric, and “-” indicates missing values of the evaluation metric.
As shown in Table 3, the values of ACC, AUC, SE, and SP metrics obtained by BPENet on STARE are the best, which are 0.9742, 0.9876, 0.8161, and 0.9884, respectively. It is worth noting that the value of the AUC metric of FSE-Net on STARE is as good as our BPENet. Although the value of the F1 metric of BPENet on STARE is 1.07% lower compared to TUnet-LBF, the values of ACC, SE, and SP metrics of BPENet are higher than TUnet-LBF. Among them, the value of the SE metric of TUnet-LBF is 1.57% lower compared to BPENet, so considering the values of several metrics together, BPENet performance on STARE is better than TUnet-LBF.
In summary, it can be noted that BPENet shows excellent performance on all three retinal vessel datasets. In addition, to display the segmentation effects of BPENet more intuitively, we visualize the retinal vessel segmentation results of BPENet and nine models, namely UNet++, ResUnet, DoubleU-Net, MSU-Net, SBDF-Net, CANet, ACC-UNet, U-Net v2, and EHMCANet. Among them, except for SBDF-Net and ACC-UNet, in which the batch size is 60 during training, the rest of the implementation details used for these nine models are the same as our BPENet setup, as described in Subsection 3.4 (Implementation details). The performance of these nine models on the DRIVE, CHASE-DB1, and STARE datasets is shown in Tables 2 and 3. The visualization of retinal vessel segmentation results of BPENet and these nine models is shown in Figure 6. By observation, the segmentation effects of BPENet are better than other methods in many details, including the segmentation effects of vessel boundaries and so on. This suggests that BPENet is able to accomplish superior segmentation effects in the complicated retinal vessel trees compared to other methods.

Visualization of segmentation results on retinal vessel datasets. (a) Original Image, (b) Partial Image, (c) GT, (d) UNet++, (e) ResUnet, (f) DoubleU-Net, (g) MSU-Net, (h) SBDF-Net, (i) CANet, (j) ACC-UNet, (k) U-Net v2, (l) EHMCANet, (m) BPENet(ours).
Coronary vessel segmentation
Table 4 demonstrates performance comparisons of BPENet with advanced methods on DCA1 and XCAD. As shown in Table 4, the values of ACC, F1, AUC, SE, and SP metrics obtained by BPENet on DCA1 are the best, which are 0.9788, 0.8119, 0.9930, 0.8564, and 0.9885, respectively. The values of ACC, AUC, and SE metrics obtained by BPENet on XCAD are the best, which are 0.9758, 0.9869, and 0.8518, respectively. Although the value of the F1 metric of BPENet on XCAD is 0.05% lower compared to MSU-Net, the values of ACC, AUC, SE, and SP metrics of BPENet are higher than MSU-Net. Among them, the value of the SE metric of MSU-Net is 2.19% lower compared to BPENet, so considering the values of several metrics together, BPENet performance on XCAD is better than MSU-Net. Additionally, although the value of the SP metric of BPENet on XCAD is 0.05% lower compared to SBDF-Net, the values of ACC, F1, AUC, and SE of BPENet are higher than SBDF-Net, and the values of these four metrics are higher by more than 0.05%, so considering the values of several metrics together, BPENet performance on XCAD is better than SBDF-Net.
Performance comparisons of BPENet with advanced methods on DCA1 and XCAD.
Bold font indicates the best result of each evaluation metric.
In summary, it can be noted that BPENet shows excellent performance on both coronary angiogram datasets. In addition, to display the segmentation effects of BPENet more intuitively, we visualize the coronary vessel segmentation results of BPENet and nine models, namely, UNet++, ResUnet, DoubleU-Net, MSU-Net, SBDF-Net, CANet, ACC-UNet, U-Net v2, and EHMCANet. Among them, the implementation details used for these nine models are the same as those introduced in Subsection 4.1 (Retinal vessel segmentation). The performance of these nine models on the DCA1 and XCAD datasets is shown in Table 4. The visualization of coronary vessel segmentation results of BPENet and these nine models is shown in Figure 7. Through observation, the segmentation effects of BPENet are superior to other methods, indicating that it can better accomplish pixel-level coronary vessel segmentation.

Visualization of segmentation results on coronary angiogram datasets. (a) Original Image, (b) GT, (c) UNet++, (d) ResUnet, (e) DoubleU-Net, (f) MSU-Net, (g) SBDF-Net, (h) CANet, (i) ACC-UNet, (j) U-Net v2, (k) EHMCANet, (l) BPENet(ours).
Complexity analyses
In Figure 8, we demonstrate the floating point operations (FLOPs) and parametric quantities (Params) of BPENet (ours) and nine models, namely, UNet++, ResUnet, DoubleU-Net, MSU-Net, SBDF-Net, CANet, ACC-UNet, U-Net v2, and EHMCANet. The input image size used is 128 × 128. G represents the hundred million, and M represents the million. Among them, BPENet's FLOPs and Params are 28.53G and 38.26 M, respectively. As shown in Figure 8, compared with other models, BPENet's FLOPs and Params are within the acceptable range.

Complexity comparison of different models.
Ablation studies
BPENet mainly introduces the FE2, BDAB, BE, and DFA modules to the U-Net. To verify the effectiveness of these modules, we perform ablation experiments on DRIVE and DCA1, as shown in Table 5.
Ablation experiments.
Bold font indicates the best result of each evaluation metric.
On DRIVE, we first perform ablation experiments of four independent modules, namely FE2, BDAB, BE, and DFA, referred to as Baseline + FE2, Baseline + BDAB, Baseline + BDAB + BE, and Baseline + DFA, respectively. It is worth noting that the BE module inputs must utilize BDAB outputs, so we treat Baseline + BDAB + BE as the independent module ablation experiment of the BE module, rather than Baseline + BE. Through experimental results, it can be observed that on five metrics, including ACC, F1, AUC, SE, and SP, the ablation experiments of four independent modules (Baseline + FE2, Baseline + BDAB, Baseline + BDAB + BE, and Baseline + DFA) all outperform the Baseline, and Baseline + BDAB + BE outperform Baseline + BDAB. This indicates that the FE2, BDAB, BE, and DFA modules we designed can all play their corresponding functions. Among them, the FE2 module can indeed enhance the model's vessel feature extraction capability, the vessel boundary detection auxiliary task introduced by BDAB can indeed enhance the extraction ability of the vessel segmentation main task for vessel boundary information, the BE module can indeed enhance the model's vessel boundary perception capability, and the DFA module can indeed fully leverage feature information from different layers of the decoder. Next, to obtain the final BPENet model (namely, Baseline + FE2 + BDAB + BE + DFA), we employ the module stacking strategy to sequentially complete the remaining experiments on the Baseline + FE2 foundation, including Baseline + FE2 + BDAB, Baseline + FE2 + BDAB + BE, and Baseline + FE2 + BDAB + BE + DFA (ours). Through experimental results, it can be observed that the values of ACC, AUC, SE, and SP metrics obtained by Baseline + FE2 + BDAB + BE + DFA (ours) are the best. Additionally, although the value of the F1 metric of Baseline + FE2 + BDAB + BE + DFA (ours) is 0.03% lower compared to Baseline + FE2 + BDAB + BE, the values of ACC, AUC, SE, and SP of Baseline + FE2 + BDAB + BE + DFA (ours) are higher than Baseline + FE2 + BDAB + BE, and the values of these four metrics are higher by more than 0.03%, so considering the values of several metrics together, Baseline + FE2 + BDAB + BE + DFA (ours) performance is better than Baseline + FE2 + BDAB + BE. In summary, Baseline + FE2 + BDAB + BE + DFA (ours) demonstrates the best overall performance in the ablation experiments on DRIVE. On DCA1, we apply a similar analysis approach, which yields conclusions consistent with those from the ablation experiments on DRIVE.
Through ablation experiments, it is evident that incorporating FE2, BDAB, BE, and DFA modules into U-Net is effective for retinal and coronary artery vessel segmentation, and the overall architectural design of BPENet is rational. In addition, to show the segmentation effects of BPENet more intuitively, we visualize the vessel segmentation results obtained by ablation experiments, as illustrated in Figure 9. Through observation, it can be seen that the segmentation effects of BPENet (namely, Baseline + FE2 + BDAB + BE + DFA) are the closest to the expert's labeled maps, indicating its superiority.

Visualization of ablation experiments. (a) Original Image, (b) GT, (c) Baseline, (d) Baseline + FE2, (e) Baseline + BDAB, (f) Baseline + BDAB + BE, (g) Baseline + DFA, (h) Baseline + FE2 + BDAB, (i) Baseline + FE2 + BDAB + BE, (j) Baseline + FE2 + BDAB + BE + DFA(ours).
Cross-dataset validations
To further evaluate the BPENet ability, we perform cross-dataset validations. Among them, for cross-dataset validations in the retinal vessel datasets, DRIVE is employed as the training set, while CHASED-B1 and STARE are employed separately as test sets. For cross-dataset validations in the coronary angiogram datasets, DCA1 is employed as the training set, while XCAD is employed as the test set. In this experiment, the implementation details of the nine models (UNet++, ResUnet, DoubleU-Net, MSU-Net, SBDF-Net, CANet, ACC-UNet, U-Net v2, and EHMCANet) are the same as those described in Subsection 4.1 (Retinal vessel segmentation). The results of cross-dataset validations are illustrated in Table 6. We use ACC, AUC, and SP metrics for performance evaluation. Through Table 6, it can be seen that BPENet has the best values of ACC and SP metrics in the “DRIVE (Training), CHASE-DB1 (Test)” experiment, while the value of AUC metric is second-best. Although the value of AUC metric of BPENet in the “DRIVE (Training), CHASE-DB1 (Test)” experiment is 0.13% lower compared to U-Net v2, the values of ACC and SP metrics of U-Net v2 are separately 0.28% and 1.64% lower compared to BPENet. Therefore, the effect of BPENet is still better than the U-Net v2 in the “DRIVE (Training), CHASE-DB1 (Test)” experiment. BPENet has the best values of ACC and AUC metrics in the “DRIVE (Training), STARE (Test)” experiment, while the value of SP metric is second-best. Although the value of SP metric of BPENet in the “DRIVE (Training), STARE (Test)” experiment is 0.02% lower compared to MSU-Net, the values of ACC and AUC metrics of MSU-Net are separately 0.13% and 2.20% lower compared to BPENet. Therefore, the effect of BPENet is still better than the MSU-Net in the “DRIVE (Training), STARE (Test)” experiment. The values of all three metrics of BPENet are the best in the “DCA1 (Training), XCAD (Test)” experiments. To sum up, the results of cross-dataset validations in Table 6 demonstrate that BPENet has better generalization ability compared to other models.
Cross-dataset validations.
Bold font indicates the best result of each evaluation metric.
Conclusion
In response to the existing problem of poor vessel boundary segmentation in current vessel segmentation methods, we propose a BPENet to alleviate the problem. BPENet not only introduces a boundary detection auxiliary task to the vessel segmentation main task, but also skillfully introduces our proposed three modules: FE2, BE, and DFA module. Among them, the FE2 module is introduced to enhance the model's vessel feature extraction capability, the BE module is introduced to enhance the model's vessel boundary perception capability, and the DFA module is introduced to fully leverage feature information from different layers. Experimental results on retinal vessel and coronary angiogram datasets show that BPENet is highly competitive in the vessel segmentation task, indicating its superiority. Of course, the vessel boundary perception enhancement brought by BPENet has practical clinical value. More precise vessel boundaries help improve the measurement accuracy of geometric parameters, such as vessel diameter, curvature, and stenosis degree. These parameters are important bases for diagnosing related diseases, such as diabetic retinopathy and coronary artery disease. In addition, accurate vessel boundary information can provide reliable structural references for clinical operations such as stent implantation and laser surgery navigation.
Although our approach demonstrates good performance on retinal vessel and coronary angiogram datasets, it still has limitations: (1) In the aspect of retinal vessel segmentation, this paper focuses primarily on segmenting the retinal vessel structure without further distinguishing between arteries and veins. However, in retinal analysis, artery/vein classification is equally important for diagnosing related diseases. For example, the arteriovenous ratio (AVR) is a key clinical indicator in diseases such as diabetic retinopathy and hypertensive retinopathy; (2) The network architecture in this paper is primarily based on the CNN implementation. Although some innovative modules have been introduced to improve vessel segmentation performance, convolution operations are inherently limited by their local receptive fields, resulting in a limited ability to extract global contextual information. However, this ability is crucial for global topological relationship modeling of complex vessel structures. In future work, we will focus on addressing the limitations mentioned above. Specifically, through multi-task learning, we will combine the retinal artery/vein classification task with the approach presented in this paper, while introducing the Transformer, which has global modeling capabilities. The new model constructed in this way will not only retain the advantages of vessel boundary perception but also have the ability to capture global contextual information, and will be able to simultaneously obtain both vessel segmentation and artery/vein classification outputs.
Footnotes
Funding
The authors disclosed receipt of the following financial support for the research, authorship, and/or publication of this article: This work was supported by the National Natural Science Foundation of China (grant number 62266011).
Declaration of conflicting interests
The authors declared no potential conflicts of interest with respect to the research, authorship, and/or publication of this article.
Data availability statement
The data that support the findings of this study are available from the corresponding author upon reasonable request.
