A Lightweight Convolutional Neural Network Based on Dynamic Level‐Set Loss Function for Spine MR Image Segmentation

• Published in: Journal of magnetic resonance imaging Vol. 59; no. 4; pp. 1438 - 1453
• Main Authors: He, Siyuan, Li, Qi, Li, Xianda, Zhang, Mengchao
• Format: Journal Article
• Language: English
• Published: Hoboken, USA John Wiley & Sons, Inc 01.04.2024; Wiley Subscription Services, Inc
• Subjects: Accuracy; Algorithms; Artificial neural networks; Cerebrospinal fluid; Datasets; Degeneration; Degenerative disc disease; dynamic loss function; Field strength; Image processing; Image segmentation; Intervertebral discs; Labels; lightweight convolutional neural network; Mathematical models; medical image segmentation; Medical imaging; Neural networks; Parameters; Pixels; Population studies; Spin dynamics; Spine; Spine (lumbar); spine MR image; Statistical analysis; Statistical tests; Vertebrae; spine MR image; lightweight convolutional neural network; medical image segmentation; dynamic loss function
• ISSN: 1053-1807; 1522-2586; 1522-2586
• DOI: 10.1002/jmri.28877

Background Spine MR image segmentation is important foundation for computer‐aided diagnostic (CAD) algorithms of spine disorders. Convolutional neural networks segment effectively, but require high computational costs. Purpose To design a lightweight model based on dynamic level‐set loss function for high segmentation performance. Study Type Retrospective. Population Four hundred forty‐eight subjects (3163 images) from two separate datasets. Dataset‐1: 276 subjects/994 images (53.26% female, mean age 49.02 ± 14.09), all for disc degeneration screening, 188 had disc degeneration, 67 had herniated disc. Dataset‐2: public dataset with 172 subjects/2169 images, 142 patients with vertebral degeneration, 163 patients with disc degeneration. Field Strength/Sequence T2 weighted turbo spin echo sequences at 3T. Assessment Dynamic Level‐set Net (DLS‐Net) was compared with four mainstream (including U‐net++) and four lightweight models, and manual label made by five radiologists (vertebrae, discs, spinal fluid) used as segmentation evaluation standard. Five‐fold cross‐validation are used for all experiments. Based on segmentation, a CAD algorithm of lumbar disc was designed for assessing DLS‐Net's practicality, and the text annotation (normal, bulging, or herniated) from medical history data were used as evaluation standard. Statistical Tests All segmentation models were evaluated with DSC, accuracy, precision, and AUC. The pixel numbers of segmented results were compared with manual label using paired t‐tests, with P < 0.05 indicating significance. The CAD algorithm was evaluated with accuracy of lumbar disc diagnosis. Results With only 1.48% parameters of U‐net++, DLS‐Net achieved similar accuracy in both datasets (Dataset‐1: DSC 0.88 vs. 0.89, AUC 0.94 vs. 0.94; Dataset‐2: DSC 0.86 vs. 0.86, AUC 0.93 vs. 0.93). The segmentation results of DLS‐Net showed no significant differences with manual labels in pixel numbers for discs (Dataset‐1: 1603.30 vs. 1588.77, P = 0.22; Dataset‐2: 863.61 vs. 886.4, P = 0.14) and vertebrae (Dataset‐1: 3984.28 vs. 3961.94, P = 0.38; Dataset‐2: 4806.91 vs. 4732.85, P = 0.21). Based on DLS‐Net's segmentation results, the CAD algorithm achieved higher accuracy than using non‐cropped MR images (87.47% vs. 61.82%). Data Conclusion The proposed DLS‐Net has fewer parameters but achieves similar accuracy to U‐net++, helps CAD algorithm achieve higher accuracy, which facilitates wider application. Evidence Level 2 Technical Efficacy Stage 1