The aging global population has led to an increased prevalence of spinal pathologies, making the accurate analysis of lumbar spine Magnetic Resonance Imaging (MRI) a critical clinical task. Deep learning-based segmentation of vertebrae and intervertebral discs offers a promising avenue for automating and improving this analysis. This paper presents a comprehensive empirical study comparing the performance of five state-of-the-art deep neural network architectures (U-Net 3D, V-Net, UNETR, Swin UNETR, and U-Net with ResNet-50) in combination with four widely-used loss functions (Dice, Generalized Dice, Dice + CE, and Dice + Focal). We evaluate these 20 configurations on the public SPIDER dataset across two distinct and clinically relevant tasks: semantic segmentation (classifying voxels as vertebra, disc, or background) and instance segmentation (additionally identifying individual anatomical structures). Our findings indicate that while most models perform well on semantic segmentation, with V-Net and Swin UNETR showing superior performance, instance segmentation proves to be a significantly more challenging task that better differentiates model capabilities. In this more complex scenario, V-Net and Swin UNETR again demonstrate the most robust and accurate results. This detailed analysis of results and trade-offs provides a valuable guide for the selection of appropriate models and loss functions for lumbar spine segmentation, clarifying the strengths and weaknesses of different approaches for this important clinical problem.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Comprehensive Evaluation of Deep Learning Architectures and Loss Functions for Lumbar Spine Segmentation in MRI

  • Cláudio Leite,
  • Samuel Felipe dos Santos,
  • Jurandy Almeida

摘要

The aging global population has led to an increased prevalence of spinal pathologies, making the accurate analysis of lumbar spine Magnetic Resonance Imaging (MRI) a critical clinical task. Deep learning-based segmentation of vertebrae and intervertebral discs offers a promising avenue for automating and improving this analysis. This paper presents a comprehensive empirical study comparing the performance of five state-of-the-art deep neural network architectures (U-Net 3D, V-Net, UNETR, Swin UNETR, and U-Net with ResNet-50) in combination with four widely-used loss functions (Dice, Generalized Dice, Dice + CE, and Dice + Focal). We evaluate these 20 configurations on the public SPIDER dataset across two distinct and clinically relevant tasks: semantic segmentation (classifying voxels as vertebra, disc, or background) and instance segmentation (additionally identifying individual anatomical structures). Our findings indicate that while most models perform well on semantic segmentation, with V-Net and Swin UNETR showing superior performance, instance segmentation proves to be a significantly more challenging task that better differentiates model capabilities. In this more complex scenario, V-Net and Swin UNETR again demonstrate the most robust and accurate results. This detailed analysis of results and trade-offs provides a valuable guide for the selection of appropriate models and loss functions for lumbar spine segmentation, clarifying the strengths and weaknesses of different approaches for this important clinical problem.