DeepEthogram, a machine learning pipeline for supervised behavior classification from raw pixels

James P Bohnslav,Nivanthika K Wimalasena,Kelsey J Clausing,Yu Y Dai,David A Yarmolinsky,Tomás Cruz,Adam D Kashlan,M Eugenia Chiappe,Lauren L Orefice,Clifford J Woolf,Christopher D Harvey
DOI: https://doi.org/10.7554/eLife.63377
IF: 7.7
2021-09-02
eLife
Abstract:Videos of animal behavior are used to quantify researcher-defined behaviors of interest to study neural function, gene mutations, and pharmacological therapies. Behaviors of interest are often scored manually, which is time-consuming, limited to few behaviors, and variable across researchers. We created DeepEthogram: software that uses supervised machine learning to convert raw video pixels into an ethogram, the behaviors of interest present in each video frame. DeepEthogram is designed to be general-purpose and applicable across species, behaviors, and video-recording hardware. It uses convolutional neural networks to compute motion, extract features from motion and images, and classify features into behaviors. Behaviors are classified with above 90% accuracy on single frames in videos of mice and flies, matching expert-level human performance. DeepEthogram accurately predicts rare behaviors, requires little training data, and generalizes across subjects. A graphical interface allows beginning-to-end analysis without end-user programming. DeepEthogram's rapid, automatic, and reproducible labeling of researcher-defined behaviors of interest may accelerate and enhance supervised behavior analysis. Code is available at: https://github.com/jbohnslav/deepethogram.
What problem does this paper attempt to address?