Accurate Segmentation of Lumbar Spine Magnetic Resonance Imaging by Deep Neural Network with Size-Adaptive Loss.

Yingdi Zhang,Zelin Shi,Huan Wang,Chongnan Yan,Lanbo Wang,Yueming Mu,Yunpeng Liu,Shuhang Wu,Tianci Liu,Mingqi Pang
DOI: https://doi.org/10.1145/3587716.3587777
2023-01-01
Abstract:Accurate segmentation of lumbar spine is crucial for the assessment of spine disease diagnosis. The accuracy and robustness of segmentation is challenged by morphological complexity of lumbar spine anatomy structure, low contrast and background noise. Moreover, unbalanced anatomical structure impede the segmentation of small-size anatomy. To overcome these problems, an automatic segmentation method of lumber spine MRI is needed. In this paper, we propose to segment lumbar spine using a convolutional neural network (CNN) model. Based on the canonical u-shaped model, a multi-scale feature fusion module (MSFF) is proposed as bottleneck to better capture lumbar spine anatomy information of different sizes. To address the problem of anatomy size imbalance, we reshape the standard cross entropy loss as Size-Adaptive loss such that it adaptively weights the loss assigned to anatomy with different sizes. In order to fully verify the accuracy and effectiveness of the proposed method, we construct the first lumbar spine anatomy segmentation dataset which consists of 747 lumbar axial images from 83 patients. The dice similarity coefficient (DSC) is used to evaluate the segmentation methods. Compared with other methods, the proposed method achieves the best results with the highest average DSC of 0.8911. Ablation study shows that the proposed MSFF module and SizeAdaptive loss can promote segmentation accuracy a lot compared with the baseline model. The results indicate that our proposed method can overcome the influence of morphological variety in some degree and is competitive with other current medical image segmentation algorithms. This novel methodology will be expected to use in clinic for image diagnosis in the future.
What problem does this paper attempt to address?