Application of knowledge discovery process on the prediction of stroke

• We predicted the outcome of stroke using knowledge discovery process (KDP) methods, Artificial Neural Networks (ANN) and Support Vector Machine (SVM) models.
• The performance of the models, assessed as accuracy and area under curve (AUC), for predicting stroke was 85.9% and 0.928 for ANN, and 84.62% and 0.91 for SVM, respectively.
• KDP was used for providing better understanding of the stroke data, and reducing unnecessary processes to be performed.
• The proposed ANN model predicted successfully the stroke disease based on the selected predictors.

ObjectiveStroke is a prominent life-threatening disease in the world. The current study was performed to predict the outcome of stroke using knowledge discovery process (KDP) methods, artificial neural networks (ANN) and support vector machine (SVM) models.Materials and methodsThe records of 297 (130 sick and 167 healthy) individuals were acquired from the databases of the department of emergency medicine. Nine predictors (coronary artery disease, diabetes mellitus, hypertension, history of cerebrovascular disease, atrial fibrillation, smoking, the findings of carotid Doppler ultrasonography [normal, plaque, plaque + stenosis ≥ 50%], the levels of cholesterol and C-reactive protein) were used for predicting the stroke. Feature selection based on the Cramer's V test was carried out for reducing the predictors. Multilayer perceptron (MLP) ANN and SVM with radial basis function (RBF) kernel were used for the prediction based on the selected predictors.ResultsThe accuracy values were 81.82% for ANN and 80.38% for SVM in the training dataset (n = 209), and 85.9% for ANN and 84.62% for SVM in the testing dataset (n = 78), respectively. ANN and SVM models yielded area under curve (AUC) values of 0.905 and 0.899 in the training dataset, and 0.928 and 0.91 in the testing dataset, consecutively.ConclusionThe findings of the current study pointed out that ANN had more predictive performance when compared with SVM in predicting stroke. The proposed ANN model would be useful when making clinical decisions regarding stroke.