Spinal MRI image segmentation based on improved U-Net

Xiajin Mei,Wenxin Zhang,Mingxu Sun,Yuliang Ma
DOI: https://doi.org/10.1007/s11760-024-03548-9
IF: 1.583
2024-09-12
Signal Image and Video Processing
Abstract:Segmenting vertebrae and intervertebral discs from spinal MRI images is essential for the diagnosis and treatment of spinal diseases. However, the similarity among different vertebrae and discs, coupled with pathological variations, complicates the segmentation process of MRI images. To tackle this challenge, this study introduces an improved U-Net architecture for spinal MRI image segmentation. The encoder employs a Multi-scale Feature Fusion Module that integrates shallow and deep features through feature downsampling. Additionally, a Dual Attention Mechanism Module is embedded within both the encoder and decoder to enhance feature extraction capabilities. Furthermore, a Graph Convolutional Module is integrated to capture contextual relationships between vertebrae and intervertebral discs. To mitigate background interference, a Spinal Region Guidance Module is also proposed. Experiments conducted on the public dataset from the MRSpineSeg Challenge yielded an average Dice Similarity Coefficient of 85.11%. The results indicate that our proposed method not only improves the segmentation performance of spinal MRI images but also surpasses other models, which holds significant implications for the clinical diagnosis and treatment of spinal diseases.
engineering, electrical & electronic,imaging science & photographic technology
What problem does this paper attempt to address?