Spine Segmentation with Multi-view GCN and Boundary Constraint

Multi-class segmentation of vertebrae and inter-vertebral discs (IVDs) is crucial for the diagnosis and treatment of spinal diseases. However, it is still a challenge due to similarities between neighboring vertebrae of a subject and differences among the IVDs from different subjects. In this paper, we propose a novel spine segmentation framework to achieve automatic segmentation of vertebrae and IVDs in MR images. The core component of the new framework is a Multi-View GCN (MVGCN), which utilizes multi-view features and graph convolutional network (GCN) to reason about the relations of vertebrae and IVDs. We additionally use a boundary constraint for better segmentation of the boundary between vertebrae and IVDs. We test our method on a public spine dataset of 172 MR volumetric images for the vertebrae and IVDs segmentation. The experimental results demonstrate the efficacy of our method. Code and models of our method will be available in the future.