Stacked Hourglass Network with a Multi-level Attention Mechanism: Where
  to Look for Intervertebral Disc Labeling
  Labeling vertebral discs from MRI scans is important for the proper diagnosis
of spinal related diseases, including multiple sclerosis, amyotrophic lateral
sclerosis, degenerative cervical myelopathy and cancer. Automatic labeling of
the vertebral discs in MRI data is a difficult task because of the similarity
between discs and bone area, the variability in the geometry of the spine and
surrounding tissues across individuals, and the variability across scans
(manufacturers, pulse sequence, image contrast, resolution and artefacts). In
previous studies, vertebral disc labeling is often done after a disc detection
step and mostly fails when the localization algorithm misses discs or has false
positive detection. In this work, we aim to mitigate this problem by
reformulating the semantic vertebral disc labeling using the pose estimation
technique. To do so, we propose a stacked hourglass network with multi-level
attention mechanism to jointly learn intervertebral disc position and their
skeleton structure. The proposed deep learning model takes into account the
strength of semantic segmentation and pose estimation technique to handle the
missing area and false positive detection. To further improve the performance
of the proposed method, we propose a skeleton-based search space to reduce
false positive detection. The proposed method evaluated on spine generic public
multi-center dataset and demonstrated better performance comparing to previous
work, on both T1w and T2w contrasts. The method is implemented in ivadomed
(https://ivadomed.org).
