Spine-GFlow: A hybrid learning framework for robust multi-tissue segmentation in lumbar MRI without manual annotation

  • Xihe Kuang
  • , Jason Pui Yin Cheung
  • , Kwan Yee K. Wong
  • , Wai Yi Lam
  • , Chak Hei Lam
  • , Richard W. Choy
  • , Christopher P. Cheng
  • , Honghan Wu
  • , Cao Yang
  • , Kun Wang
  • , Yang Li*
  • , Teng Zhang
  • *Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

Most learning-based magnetic resonance image (MRI) segmentation methods rely on the manual annotation to provide supervision, which is extremely tedious, especially when multiple anatomical structures are required. In this work, we aim to develop a hybrid framework named Spine-GFlow that combines the image features learned by a CNN model and anatomical priors for multi-tissue segmentation in a sagittal lumbar MRI. Our framework does not require any manual annotation and is robust against image feature variation caused by different image settings and/or underlying pathology. Our contributions include: 1) a rule-based method that automatically generates the weak annotation (initial seed area), 2) a novel proposal generation method that integrates the multi-scale image features and anatomical prior, 3) a comprehensive loss for CNN training that optimizes the pixel classification and feature distribution simultaneously. Our Spine-GFlow has been validated on 2 independent datasets: HKDDC (containing images obtained from 3 different machines) and IVDM3Seg. The segmentation results of vertebral bodies (VB), intervertebral discs (IVD), and spinal canal (SC) are evaluated quantitatively using intersection over union (IoU) and the Dice coefficient. Results show that our method, without requiring manual annotation, has achieved a segmentation performance comparable to a model trained with full supervision (mean Dice 0.914 vs 0.916).

Original languageEnglish
Article number102091
JournalComputerized Medical Imaging and Graphics
Volume99
DOIs
StatePublished - Jul 2022

Keywords

  • Auto-annotation
  • Multi-tissue segmentation
  • Spine
  • Weakly-supervised segmentation

Fingerprint

Dive into the research topics of 'Spine-GFlow: A hybrid learning framework for robust multi-tissue segmentation in lumbar MRI without manual annotation'. Together they form a unique fingerprint.

Cite this