Contour attention network for cerebrovascular segmentation from TOF-MRA volumetric images

Cerebrovascular segmentation is a crucial step in the computer-assisted diagnosis of cerebrovascular pathologies. However, accurate extraction of cerebral vessels from time-of-flight magnetic resonance angiography (TOF-MRA) data is still challenging due to the complex topology and slender shape.

The existing deep learning-based approaches pay more attention to the skeleton and ignore the contour, which limits the segmentation performance of the cerebrovascular structure. We aim to weight the contour of brain vessels in shallow features when concatenating with deep features. It helps to obtain more accurate cerebrovascular details and narrows the semantic gap between multilevel features.

This work proposes a novel framework for priority extraction of contours in cerebrovascular structures. We first design a neighborhood-based algorithm to generate the ground truth of the cerebrovascular contour from original annotations, which can introduce useful shape information for the segmentation network. Moreover, We propose an encoder-dual decoder-based contour attention network (CA-Net), which consists of the dilated asymmetry convolution block (DACB) and the Contour Attention Module (CAM). The ancillary decoder uses the DACB to obtain cerebrovascular contour features under the supervision of contour annotations. The CAM transforms these features into a spatial attention map to increase the weight of the contour voxels in main decoder to better restored the vessel contour details.

The CA-Net is thoroughly validated using two publicly available datasets, and the experimental results demonstrate that our network outperforms the competitors for cerebrovascular segmentation. We achieved the average dice similarity coefficient ( $$ ) of 68.15 and 99.92% in natural and synthetic datasets. Our method segments cerebrovascular structures with better completeness.

We propose a new framework containing contour annotation generation and cerebrovascular segmentation network that better captures the tiny vessels and improve vessel connectivity.