SLoRD: Structural Low-Rank Descriptors for Shape Consistency in Vertebrae Segmentation

TL;DR

This paper introduces SLoRD, a shape-aware contour generation network using structural low-rank descriptors to improve vertebrae segmentation accuracy and consistency in CT images, outperforming existing methods.

Contribution

The paper proposes a novel shape constraint framework using spherical coordinate-based descriptors for vertebrae segmentation, enhancing contour accuracy and intra-vertebrae consistency.

Findings

Outperforms state-of-the-art methods on VerSe datasets

Effectively refines coarse segmentation predictions

Provides a plug-and-play module for segmentation improvement

Abstract

Automatic and precise multi-class vertebrae segmentation from CT images is crucial for various clinical applications. However, due to similar appearances between adjacent vertebrae and the existence of various pathologies, existing single-stage and multi-stage methods suffer from imprecise vertebrae segmentation. Essentially, these methods fail to explicitly impose both contour precision and intra-vertebrae voxel consistency constraints synchronously, resulting in the intra-vertebrae segmentation inconsistency, which refers to multiple label predictions inside a singular vertebra. In this work, we intend to label complete binary masks with sequential indices to address that challenge. Specifically, a contour generation network is proposed based on Structural Low-Rank Descriptors for shape consistency, termed SLoRD. For a structural representation of vertebral contours, we adopt the spherical coordinate system and devise the spherical centroid to calculate contour descriptors. Due to vertebrae's similar appearances, basic contour descriptors can be acquired offline to restore original contours. Therefore, SLoRD leverages these contour priors and explicit shape constraints to facilitate regressed contour points close to vertebral surfaces. Quantitative and qualitative evaluations on VerSe 2019 and 2020 demonstrate the superior performance of our framework over other single-stage and multi-stage state-of-the-art (SOTA) methods. Further, SLoRD is a plug-and-play framework to refine the segmentation inconsistency existing in coarse predictions from other approaches. Source codes are available.