Indian Journal of Industrial and Applied Mathematics
  • Year: 2016
  • Volume: 7
  • Issue: 2

Comparison of Conventional Regression With Machine Learning Methods for Stroke Outcome Prediction

  • Author:
  • Ahmedul Kabir1, Carolina Ruiz2, Sergio A. Alvarez3, Majaz Moonis4,
  • Total Page Count: 13
  • Published Online: Dec 1, 2016
  • Page Number: 93 to 105

1PhD Candidate, Department of Computer Science, Worcester Polytechnic Institute, 100 Institute Road, Worcester, MA, 01609, United States of America

2Associate Professor, Department of Computer Science, Worcester Polytechnic Institute, 100 Institute Road, Worcester, MA, 01609, United States of America

3Associate Professor, Department of Computer Science, Boston College, Chestnut Hill, MA, 02467, United States of America

4Professor, Department of Computer Science, Worcester Polytechnic Institute, 100 Institute Road, Worcester, MA, 01609, United States of America

*Corresponding Author Email Id: Majaz.Moonis@umassmemorial.org

Abstract

In performing statistical analysis in clinical trials, the conventional method is to conduct univariate analysis for variable selection, followed by multivariate logistic regression to find more details about the variables. While this approach is widely used, and does deliveruseful explanations of the data, it has several shortcomings. In this paper, we compare the results obtained with conventional methods with those of alternative machine learning algorithms in terms of their ability to predict stroke outcome of patients.We also examine the different models constructed by the algorithms, and compare them with the logistic regression model. We find that machine learning can be used to predict more precise outcomes and reveal more variables than previously known that may play an important role in determining stroke outcomes.

Keywords

Stroke outcome, Multivariate logistic regression, Supervised machine learning, Variable selection