Attention-based Shape-Deformation Networks for Artifact-Free Geometry Reconstruction of Lumbar Spine from MR Images.

Journal: IEEE transactions on medical imaging
Published Date:

Abstract

Lumbar disc degeneration, a progressive structural wear and tear of lumbar intervertebral disc, is regarded as an essential role on low back pain, a significant global health concern. Automated lumbar spine geometry reconstruction from MR images will enable fast measurement of medical parameters to evaluate the lumbar status, in order to determine a suitable treatment. Existing image segmentation-based techniques often generate erroneous segments or unstructured point clouds, unsuitable for medical parameter measurement. In this work, we present UNet-DeformSA and TransDeformer: novel attention-based deep neural networks that reconstruct the geometry of the lumbar spine with high spatial accuracy and mesh correspondence across patients, and we also present a variant of TransDeformer for error estimation. Specially, we devise new attention modules with a new attention formula, which integrate tokenized image features and tokenized shape features to predict the displacements of the points on a shape template. The deformed template reveals the lumbar spine geometry in an image. Experiment results show that our networks generate artifact-free geometry outputs, and the variant of TransDeformer can predict the errors of a reconstructed geometry. Our code is available at https://github.com/linchenq/TransDeformer-Mesh.

Authors

  • Linchen Qian
    Department of Computer Science, University of Miami, Coral Gables, FL, USA.
  • Jiasong Chen
    Department of Computer Science, University of Miami, Coral Gables, FL, USA.
  • Linhai Ma
    Department of Computer Science, University of Miami, Coral Gables, FL, USA.
  • Timur Urakov
    Department of Neurological Surgery, University of Miami Miller School of Medicine, Miami, FL, USA.
  • Weiyong Gu
    Department of Mechanical and Aerospace Engineering, University of Miami, Coral Gables, FL, USA.
  • Liang Liang
    Department of Computer Science, University of Miami, Coral Gables, FL.

Keywords

No keywords available for this article.