Trace-back along capsules and its application on semantic segmentation

T Sun, Z Wang, CD Smith, J Liu - 1901 - openreview.net
1901openreview.net
In this paper, we propose a capsule-based neural network model to solve the semantic
segmentation problem. By taking advantage of the extractable part-whole dependencies
available in capsule layers, we derive the probabilities of the class labels for individual
capsules through a recursive, layer-by-layer procedure. We model this procedure as a
traceback pipeline and take it as a central piece to build an end-to-end segmentation
network. Under the proposed framework, image-level class labels and object boundaries …
In this paper, we propose a capsule-based neural network model to solve the semantic segmentation problem. By taking advantage of the extractable part-whole dependencies available in capsule layers, we derive the probabilities of the class labels for individual capsules through a recursive, layer-by-layer procedure. We model this procedure as a traceback pipeline and take it as a central piece to build an end-to-end segmentation network. Under the proposed framework, image-level class labels and object boundaries are jointly sought in an explicit manner, which poses a significant advantage over the state-of-the-art fully convolutional network (FCN) solutions. Experiments conducted on modified MNIST and neuroimages demonstrate that our model considerably enhance the segmentation performance compared to the leading FCN variant.
openreview.net
以上显示的是最相近的搜索结果。 查看全部搜索结果