Three-dimensional lumbar spine generation using variational autoencoder

Abstract

The disease analysis of the lumbar spine often requires a large number of three-dimensional (3D) models. Currently, there is a lack of 3D model of the lumbar spine for research, especially for the diseases such as scoliosis where it is difficult to collect sufficient data in a short period of time. To solve this problem, we develop an end-to-end network based on 3D variational autoencoder for randomly generating 3D lumbar spine model. In this network, the dual path encoder structure is used to fit two individual variables, i.e., mean and variance. Spatial coordinate attention modules are added to the encoder to improve the learning ability of the network to the 3D spatial structure of the lumbar spine. To enhance the power of the network to reconstruct the lumbar spine, a regularization loss is added to constrain the distribution loss. Additionally, Gaussian noise layers are added to the decoder to improve the authenticity and diversity of generated model. The experiments were conducted on the data of the entire lumbar spine and the individual lumbar vertebra, respectively. The results showed that the voxel intersection over union was 0.588 and 0.684, the voxel Dice coefficient was 0.739 and 0.811, the average surface distance was 0.807 and 1.189, and the Hausdorff distance was 2.615 and 3.710, for the entire lumbar spine and individual lumbar vertebra, respectively. The developed approach is comparable to the most commonly used model generation method of statistical shape model (SSM) in both visual and objective indicators, while the developed approach does not require the landmarks that is needed in the SSM method. Therefore, this fully automatic method can be easily used for population-based modeling of the lumbar spine which has the potential to be a powerful clinical tool.

Keywords: Dual path encoder; Gaussian noise layer; Lumbar spine generation; Regularization loss; Spatial coordinate attention module.