Lumbar intervertebral disc detection and classification with novel deep learning models

Abstract

Low back pain (LBP) is a prevalent spinal issue, affecting eight out of ten individuals. Notably, lumbar intervertebral disc (IVD) abnormalities frequently contribute to LBP. To diagnose LBP, Magnetic Resonance Imaging (MRI) is crucial for obtaining detailed spinal images. This paper employs deep learning (DL) to detect and locate lumbar IVD in sagittal MR images. It further classifies lumbar IVDs as healthy or herniated, utilizing both novel convolutional neural network (CNN) and conventional CNN models. The dataset utilized comprises MR images from 32 patients, with 10 exhibiting healthy discs and the remaining 22 posing a mix of healthy and herniated discs, totaling 160 lumbar discs, incorporating 112 healthy and 48 herniated discs. In this study, ResNet-50 architecture in the Novel Lumbar IVD detection (NLID) model served as the feature extractor to segment the five lumbar IVDs from MR images. The features extracted from ResNet-50 were input into YOLOv2 for the identification of the region of interest (ROI). The findings indicate that optimal performance was achieved at the 22nd Rectified Linear Unit (ReLU) activation layer, boasting a remarkable 99.59% average precision, 97.22% F1-score, 94.59% precision, and a perfect 100% recall. This commendable performance consistently held above the 85% threshold until the 22nd ReLU activation layer. Regarding imbalanced dataset classification, AlexNet emerged as the frontrunner among other pre-trained networks, boasting the highest test accuracy of 90.63%, and an impressive F1 score of 88.77%. Meanwhile, the Novel Lumbar IVD Classification (NLIC) model achieved superior results with 93.75% test accuracy, and 92.27% F1-score. In the setting of the balanced dataset, NLIC achieved 96.88% test accuracy, and 96.46% F1-score with fewer epochs compared to AlexNet, affirming the robustness of the novel trained-from-scratch network. These findings distinctly underscore the effectiveness of CNNs in both medical image segmentation and classification.