Use of computer vision analysis for labeling inattention periods in EEG recordings with visual stimuli.

Journal: Scientific reports
Published Date:

Abstract

Electroencephalography (EEG) recordings with visual stimuli require detailed coding to determine the periods of participant's attention. Here we propose to use a supervised machine learning model and off-the-shelf video cameras only. We extract computer vision-based features such as head pose, gaze, and face landmarks from the video of the participant, and train the machine learning model (multi-layer perceptron) on an initial dataset, then adapt it with a small subset of data from a new participant. Using a sample size of 23 autistic children with and without co-occurring ADHD (attention-deficit/hyperactivity disorder) aged 49-95 months, and training on additional 2560 labeled frames (equivalent to 85.3 s of the video) of a new participant, the median area under the receiver operating characteristic curve for inattention detection was 0.989 (IQR 0.984-0.993) and the median inter-rater reliability (Cohen's kappa) with a trained human annotator was 0.888. Agreement with human annotations for nine participants was in the 0.616-0.944 range. Our results demonstrate the feasibility of automatic tools to detect inattention during EEG recordings, and its potential to reduce the subjectivity and time burden of human attention coding. The tool for model adaptation and visualization of the computer vision features is made publicly available to the research community.

Authors

  • Dmitry Yu Isaev
    Department of Electrical and Computer Engineering, Duke University, Durham, NC, USA.
  • Samantha Major
    Duke Center for Autism and Brain Development, Department of Psychiatry and Behavioral Sciences, Duke University School of Medicine, Durham, NC, USA. samantha.major@duke.edu.
  • Kimberly L H Carpenter
    Department of Psychiatry and Behavioral Sciences, Duke University School of Medicine, Durham, North Carolina, United States of America.
  • Jordan Grapel
    Duke Center for Autism and Brain Development, Department of Psychiatry and Behavioral Sciences, Duke University School of Medicine, Durham, NC, USA.
  • Zhuoqing Chang
    Department of Electrical and Computer Engineering, Duke University, Durham, NC, USA.
  • Matias Di Martino
    Universidad Católica del Uruguay, Montevideo, Uruguay.
  • David Carlson
    Department of Biostatistics and Bioinformatics, Duke University School of Medicine, Durham, North Carolina.
  • Geraldine Dawson
    Duke Center for Autism and Brain Development, Department of Psychiatry and Behavioral Science, Duke University, Durham, NC, 27707, United States.
  • Guillermo Sapiro
    Electrical and Computer Engineering, Computer Sciences, Biomedical Engineering, and Math, Duke University, Durham, NC, 27707, United States.