Pixel Distribution Learning for Vessel Segmentation under Multiple Scales

Chenqiu Zhao, Anup Basu

Abstract: In this work we try to address if there is a better way to classify two distributions, rather than using histograms; and answer if we can make a deep learning network learn and classify distributions automatically. These improvements can have wide ranging applications in computer vision and medical image processing. More specifically, we propose a new vessel segmentation method based on pixel distribution learning under multiple scales. In particular, a spatial distribution descriptor named Random Permutation of Spatial Pixels (RPoSP) is derived from vessel images and used as the input to a convolutional neural network for distribution learning. Based on our preliminary experiments we currently believe that a wide network, rather than a deep one, is better for distribution learning. There is only one convolutional layer, one rectified linear layer and one fully connected layer followed by a softmax loss in our network. Furthermore, in order to improve the accuracy of the proposed approach, the RPoSP features are captured at multiple scales and combined together to form the input of the network. Evaluations using standard benchmark datasets demonstrate that the proposed approach achieves promising results compared to the state-of-the-art.