pp. 2341-2356
S&M2976 Research Paper of Special Issue https://doi.org/10.18494/SAM3725 Published in advance: March 9, 2022 Published: June 30, 2022 Applying Machine Learning to Determine the Behavioral Characteristics of Rodents with Traumatic Brain Injury in an Eight-arm Maze [PDF] Shu-Cing Wu, Chi-Yuan Lin, Liang-Jyun Hong, and Chi-Chun Chen (Received November 4, 2021; Accepted January 20, 2022) Keywords: cognitive parameters, eight-arm maze, machine learning, traumatic brain injury, support vector machine, decision tree, random forest, k-nearest neighbor
In this study, we identified the cognitive parameters of rats with traumatic brain injury (TBI) in an eight-arm radial maze and used them for TBI classification through machine learning models. A total of 16 cognitive parameters were derived using a sensing trajectory bitmap in the eight-arm maze. Of these 16 parameters, five (i.e., short-term memory error, latency, total distance, frequency of movement from an arm without food to an arm with food, and frequency of entry into the arm on the right after exiting an arm) were selected as representative parameters and were input into four machine learning models, namely, support vector machine (SVM), decision tree, random forest, and k-nearest neighbor (KNN) models, to classify and compare sham rats and rats with TBI. The performance evaluation results for the machine learning models revealed that the SVM model had the best performance among the models. Its overall accuracy, sensitivity, and area under the receiver operating characteristic curve (AUC) were >85, 98, and >94%, respectively. At some postsurgical time points, the sensitivity and AUC of the SVM model even approached 100%. The random forest and KNN models had satisfactory performance on Day 28 postsurgery. Overall, the SVM model had satisfactory performance in classifying both mild and severe TBI. Our findings can serve as a reference for future research on TBI feature classification.
Corresponding author: Chi-Chun ChenThis work is licensed under a Creative Commons Attribution 4.0 International License. Cite this article Shu-Cing Wu, Chi-Yuan Lin, Liang-Jyun Hong, and Chi-Chun Chen, Applying Machine Learning to Determine the Behavioral Characteristics of Rodents with Traumatic Brain Injury in an Eight-arm Maze, Sens. Mater., Vol. 34, No. 6, 2022, p. 2341-2356. |