A machine learning eye movement detection algorithm using electrooculography

Abstract

Abstract Study Objectives Eye movement quantification in polysomnograms (PSG) is difficult and resource intensive. Automated eye movement detection would enable further study of eye movement patterns in normal and abnormal sleep, which could be clinically diagnostic of neurologic disorders, or used to monitor potential treatments. We trained a long short-term memory (LSTM) algorithm that can identify eye movement occurrence with high sensitivity and specificity. Methods We conducted a retrospective, single-center study using one-hour PSG samples from 47 patients 18–90 years of age. Team members manually identified and trained an LSTM algorithm to detect eye movement presence, direction, and speed. We performed a 5-fold cross validation and implemented a “fuzzy” evaluation method to account for misclassification in the preceding and subsequent 1-second of gold standard manually labeled eye movements. We assessed G-means, discrimination, sensitivity, and specificity. Results Overall, eye movements occurred in 9.4% of the analyzed EOG recording time from 47 patients. Eye movements were present 3.2% of N2 (lighter stages of sleep) time, 2.9% of N3 (deep sleep), and 19.8% of REM sleep. Our LSTM model had average sensitivity of 0.88 and specificity of 0.89 in 5-fold cross validation, which improved to 0.93 and 0.92 respectively using the fuzzy evaluation scheme. Conclusion An automated algorithm can detect eye movements from EOG with excellent sensitivity and specificity. Noninvasive, automated eye movement detection has several potential clinical implications in improving sleep study stage classification and establishing normal eye movement distributions in healthy and unhealthy sleep, and in patients with and without brain injury.