Troubles neurologiques

Deformed non-local neural network for retinal vessel segmentation

Abstract

Junping Zhao

Vessel segmentation is a challenging problem in medical image segmentation, as it requires knowing the location of each tiny blood vessel and understanding the global semantic information. Previous method has demonstrated that long-range dependencies information plays an important role in understanding global segmentation information. To capture long-range dependencies information, researchers often use non-local structure. However, it requires too much computing power and a large amount of GPU memory. In this paper, we present a deformed non-local (DNL) neural network structure for retinal vessel segmentation. DNL inherits the structure of the Non-local module, but it changes the operation rules of nonlocal weight matrix multiplication, which can greatly reduce the problem of excessive computation and memory usage. Meanwhile, we introduce the atrous spatial pyramid pooling module to increase the receptive field of the networks, which showed that it is effective to resample features at different scales for accurately and efficiently classifying regions of an arbitrary scale. The proposed method was evaluated on retinal vessel datasets and experimental results show that it outperforms state-of-the-art methods. For a 128 × 128 input, DNL is around 2.5 times faster than a non-local block on GPU.