A Deep Learning Model for Automatic Detection and Classification of Disc Herniation in Magnetic Resonance Images

Abstract

Localization of lumbar discs in magnetic resonance imaging (MRI) is a challenging task, due to a vast range of shape, size, number, and appearance of discs and vertebrae. Based on a review of the cutting-edge methods, the majority of applied techniques are either semi-automatic, extremely sensitive to change in parameters, or involve further modification of the results. All of the above represents a motivation for implementing deep learning-based approaches for automatic segmentation and classification of disc herniation in MR images. This paper proposes a complete automated process based on deep learning to diagnose disc herniation. The methodology includes several steps starting from segmentation of region of interest (ROI), in this case disc area, bounding box cropping and enhancement of ROI, after which the image is classified based on convolutional neural network (CNN) into adequate classes (healthy, bulge, central, right or left herniation for axial view and healthy, L4/L5, L5/S1 level of herniation in sagittal view). The results show high accuracy of segmentation for both axial view (dice = 0.961, IOU = 0.925) and sagittal view (dice = 0.897, IOU = 0.813) images. After cropping and enhancing the region of interest, accuracy of classification was 0.87 for axial view images and 0.91 for sagittal view images. Comparison with the literature shows that proposed methodology outperforms state-of-the-art results when it comes to multiclassification problems. A fully automated decision support system for disc hernia diagnosis can assist in generating diagnostic findings in a timely manner, while human mistakes caused by cognitive overload and procedure-related errors can be reduced.