Data-driven diagnosis of spinal abnormalities using feature selection and machine learning algorithms

PLoS One. 2020 Feb 6;15(2):e0228422. doi: 10.1371/journal.pone.0228422. eCollection 2020.

Abstract

This paper focuses on the application of machine learning algorithms for predicting spinal abnormalities. As a data preprocessing step, univariate feature selection as a filter based feature selection, and principal component analysis (PCA) as a feature extraction algorithm are considered. A number of machine learning approaches namely support vector machine (SVM), logistic regression (LR), bagging ensemble methods are considered for the diagnosis of spinal abnormality. The SVM, LR, bagging SVM and bagging LR models are applied on a dataset of 310 samples publicly available in Kaggle repository. The performance of classification of abnormal and normal spinal patients is evaluated in terms of a number of factors including training and testing accuracy, recall, and miss rate. The classifier models are also evaluated by optimizing the kernel parameters, and by using the results of receiver operating characteristic (ROC) and precision-recall curves. Results indicate that when 78% data are used for training, the observed training accuracies for SVM, LR, bagging SVM and bagging LR are 86.30%, 85.47%, 86.72% and 85.06%, respectively. On the other hand, the accuracies for the test dataset for SVM, LR, bagging SVM and bagging LR are the same being 86.96%. However, bagging SVM is the most attractive as it has a higher recall value and a lower miss rate compared to others. Hence, bagging SVM is suitable for the classification of spinal patients when applied on the most five important features of spinal samples.

Publication types

  • Evaluation Study

MeSH terms

  • Algorithms*
  • Datasets as Topic / statistics & numerical data*
  • Diagnosis, Computer-Assisted / methods*
  • Diagnosis, Differential
  • Humans
  • Image Interpretation, Computer-Assisted / methods
  • Logistic Models
  • Machine Learning
  • Posture / physiology
  • Predictive Value of Tests
  • Reproducibility of Results
  • Spinal Diseases / diagnosis*
  • Spinal Diseases / epidemiology
  • Spine / abnormalities*
  • Spine / diagnostic imaging*
  • Support Vector Machine

Grants and funding

The authors received no specific funding for this work.