ICUnet++: an Inception-CBAM network based on Unet++ for MR spine image segmentation

被引:0
作者
Lei Li
Juan Qin
Lianrong Lv
Mengdan Cheng
Biao Wang
Dan Xia
Shike Wang
机构
[1] Tianjin University of Technology,School of Integrated Circuit Science and Engineering
来源
International Journal of Machine Learning and Cybernetics | 2023年 / 14卷
关键词
Deep learning; Convolutional neural network; MRI; Spine segmentation; Attention mechanism;
D O I
暂无
中图分类号
学科分类号
摘要
In recent years, more attention paid to the spine caused by related diseases, spinal parsing (the multi-class segmentation of vertebrae and intervertebral disc) is an important part of the diagnosis and treatment of various spinal diseases. The more accurate the segmentation of medical images, the more convenient and quick the clinicians can evaluate and diagnose spinal diseases. Traditional medical image segmentation is often time consuming and energy consuming. In this paper, an efficient and novel automatic segmentation network model for MR spine images is designed. The proposed Inception-CBAM Unet++ (ICUnet++) model replaces the initial module with the Inception structure in the encoder-decoder stage base on Unet++ , which uses the parallel connection of multiple convolution kernels to obtain the features of different receptive fields during in the feature extraction. According to the characteristics of the attention mechanism, Attention Gate module and CBAM module are used in the network to make the attention coefficient highlight the characteristics of the local area. To evaluate the segmentation performance of network model, four evaluation metrics, namely intersection over union (IoU), dice similarity coefficient(DSC), true positive rate(TPR), positive predictive value(PPV) are used in the study. The published SpineSagT2Wdataset3 spinal MRI dataset is used during the experiments. In the experiment results, IoU reaches 83.16%, DSC is 90.32%, TPR is 90.40%, and PPV is 90.52%. It can be seen that the segmentation indicators have been significantly improved, which reflects the effectiveness of the model.
引用
收藏
页码:3671 / 3683
页数:12
相关论文
共 88 条
  • [1] Freburger JK(2009)The rising prevalence of chronic low back pain Arch Intern Med 169 251-258
  • [2] Holmes GM(2011)Imaging of lumbar degenerative disk disease: history and current state Skeletal Radiol 40 1175-1189
  • [3] Agans RP(2022)Brain tumor segmentation based on region of interest-aided localization and segmentation U-Net Int J Mach Learn Cyber 13 2435-2445
  • [4] Emch TM(2009)Atlas-based segmentation of degenerated lumbar intervertebral discs from MR images of the spine IEEE Trans Biomed Eng 56 2225-2231
  • [5] Modic MT(2013)Intervertebral disc segmentation in MR images using anisotropic oriented flux Med Image Anal 17 43-61
  • [6] Li S(2022)Evolutionary neural networks for deep learning: a review Int J Mach Learn & Cyber 13 3001-3018
  • [7] Liu J(2022)Feature extraction and classification algorithm, which one is more essential? An experimental study on a specific task of vibration signal diagnosis Int J Mach Learn Cyber 13 1685-1696
  • [8] Song Z(2022)Tuning of data augmentation hyperparameters in deep learning to building construction image classification with small datasets Int J Mach Learn Cyber 81 104445-22
  • [9] Michopoulou SK(2022)Small target deep convolution recognition algorithm based on improved YOLOv4 Int J Mach Learn Cyber 1 S4-1742
  • [10] Costaridou L(2017)Fully convolutional networks for semantic segmentation IEEE Trans Pattern Anal Mach Intell 131 15-1605