3D vessel-like structure segmentation in medical images by an edge-reinforced network.

ABSTRACT

The vessel-like structure in biomedical images, such as within cerebrovascular and nervous pathologies, is an essential biomarker in understanding diseases' mechanisms and in diagnosing and treating diseases. However, existing vessel-like structure segmentation methods often produce unsatisfactory results due to challenging segmentations for crisp edges. The edge and nonedge voxels of the vessel-like structure in three-dimensional (3D) medical images usually have a highly imbalanced distribution as most voxels are non-edge, making it challenging to find crisp edges. In this work, we propose a generic neural network for the segmentation of the vessel-like structures in different 3D medical imaging modalities. The new edge-reinforced neural network (ER-Net) is based on an encoder-decoder architecture. Moreover, a reverse edge attention module and an edge-reinforced optimization loss are proposed to increase the weight of the voxels on the edge of the given 3D volume to discover and better preserve the spatial edge information. A feature selection module is further introduced to select discriminative features adaptively from an encoder and decoder simultaneously, which aims to increase the weight of edge voxels, thus significantly improving the segmentation performance. The proposed method is thoroughly validated using four publicly accessible datasets, and the experimental results demonstrate that the proposed method generally outperforms other state-of-the-art algorithms for various metrics.