Clinical Psychopharmacology and Neuroscience Papers in Press available online.

 
Machine learning algorithms for the prediction of locomotor activity by infrared motion detector on sleep-wake states in mice
Yoo Rha Hong 1, Kyungwon Kim2, Eunsoo Moon 2,3,*, Jeonghyun Park 2, Chi Eun Oh 1, Jung Hyun Lee 1, Min Yoon 3
1Department of Pediatrics, College of Medicine, Kosin University, Busan, Republic of Korea, 2Department of Psychiatry and Biomedical Research Institute, Pusan National University Hospital, Busan, Republic of Korea, 3Department of Psychiatry, Pusan National University School of Medicine, Yangsan, Republic of Korea, 4Department of Applied Mathematics, Pukyung National University, Busan, Republic of Korea
Abstract
Objective: Even though studies using machine learning on sleep-wake states have been performed, studies in various conditions are still necessary. This study aimed to examine the performance of the prediction model of locomotor activities on sleep-wake states using machine learning algorithms.
Methods: The processed data using moving average of locomotor activities were used as predicting features. The sleep-wake states were used as true labels. The prediction models were established by machine learning classifiers such as support vector machine with radial basis function (SVM-RBF), linear discriminant analysis (LDA), naïve Bayes (NB), and random forest (RF). The prediction model was evaluated by a six-fold cross validation.
Results: The SVM-RBF and RF showed acceptable performance within a window of moving average from 480 to 1,200 seconds. The highest accuracy (0.869) was shown by the RF at the interval of 480 seconds. Meanwhile, the highest area under the curve (0.939) was shown by LDA at the interval of 870 seconds.
Discussion: This study suggested that the prediction model on sleep-wake state using machine learning could show an improvement of the model performance when using moving average with raw data. The prediction model using locomotor activity can be useful in research on sleep-wake state.
Accepted Manuscript [Submitted on 2021-11-12, Accepted on 2022-03-08]