Lumbar and pelvic CT image segmentation based on cross-scale feature fusion and linear self-attention mechanism.
Authors
Affiliations (3)
Affiliations (3)
- Yancheng Third People's Hospital, Affiliated Hospital 6 of Nantong University, The Affiliated Hospital of Jiangsu Medical College, Yancheng, 224001, Jiangsu, China.
- School of Medical Imaging, Jiangsu Medical College, Yancheng, 224005, Jiangsu, China. [email protected].
- Yancheng Third People's Hospital, Affiliated Hospital 6 of Nantong University, The Affiliated Hospital of Jiangsu Medical College, Yancheng, 224001, Jiangsu, China. [email protected].
Abstract
The lumbar spine and pelvis are critical stress-bearing structures of the human body, and their rapid and accurate segmentation plays a vital role in clinical diagnosis and intervention. However, conventional CT imaging poses significant challenges due to the low contrast of sacral and bilateral hip tissues and the complex and highly similar intervertebral space structures within the lumbar spine. To address these challenges, we propose a general-purpose segmentation network that integrates a cross-scale feature fusion strategy with a linear self-attention mechanism. The proposed network effectively extracts multi-scale features and fuses them along the channel dimension, enabling both structural and boundary information of lumbar and pelvic regions to be captured within the encoder-decoder architecture.Furthermore, we introduce a linear mapping strategy to approximate the traditional attention matrix with a low-rank representation, allowing the linear attention mechanism to significantly reduce computational complexity while maintaining segmentation accuracy for vertebrae and pelvic bones. Comparative and ablation experiments conducted on the CTSpine1K and CTPelvic1K datasets demonstrate that our method achieves improvements of 1.5% in Dice Similarity Coefficient (DSC) and 2.6% in Hausdorff Distance (HD) over state-of-the-art models, validating the effectiveness of our approach in enhancing boundary segmentation quality and segmentation accuracy in homogeneous anatomical regions.