In Journal of stroke and cerebrovascular diseases : the official journal of National Stroke Association
OBJECTIVES : Classifying the possibility of home discharge is important during stroke rehabilitation to support decision-making. There have been several studies on supervised machine learning algorithms, but only a few have compared the performance of different algorithms based on the same dataset for the classification of home discharge possibility. Therefore, we aimed to evaluate five supervised machine learning algorithms for the classification of home discharge possibility in stroke patients.
MATERIALS AND METHODS : This was a secondary analysis based on the data of 481 stroke patients from the database of our institution. Five models developed by supervised machine learning algorithms, including decision tree (DT), linear discriminant analysis (LDA), k-nearest neighbors (k-NN), support vector machine (SVM), and random forest (RF) were compared by constructing a classification system based on the same dataset. Several parameters including classification accuracy, area under the curve (AUC), and F1 score (a weighted average of precision and recall) were used for model evaluation.
RESULTS : The k-NN model had the best classification accuracy (84.0%) with a moderate AUC (0.88) and F1 score (87.8). The SVM model also showed high classification accuracy (82.6%) along with the highest AUC (0.91), sensitivity (94.4), negative predictive value (87.5), and negative likelihood ratio (0.088). The DT, LDA, and RF models had high classification accuracies (≥ 79.9%) with moderate AUCs (≥ 0.84) and F1 scores (≥ 83.8).
CONCLUSIONS : Regarding model performance, the k-NN and SVM seemed the best candidate algorithms for classifying the possibility of home discharge in stroke patients.
Imura Takeshi, Toda Haruki, Iwamoto Yuji, Inagawa Tetsuji, Imada Naoki, Tanaka Ryo, Inoue Yu, Araki Hayato, Araki Osamu
Algorithm, Artificial intelligence, Home discharge, Machine learning, Stroke