High throughput machine learning pipeline to characterize larval zebrafish motor behavior

Abstract

Using machine learning, we developed models that rigorously detect and classify larval zebrafish spontaneous and stimulus-evoked behaviors in various well plate formats. Zebrafish are an ideal model system for investigating the neural substrates underlying behavior due to their simple nervous system and well-documented responses to environmental stimuli. To track movement, we utilized an 8 key point pose estimation model, allowing precise capture of zebrafish kinematics. Using this kinematic data, we trained two random forest classifiers in a semi-supervised learning framework to classify various discreet behavioral outputs including stationary, scoot, turn, acoustic-startle like behavior, and visual-startle like behavior. The classifiers were trained on a manually labeled dataset, and their accuracy was validated showing high precision. To validate our machine learning models, we analyzed behavioral outputs during various stimulus evoked responses and during spontaneous behavior. For additional validation, and to show the utility of our recording and analysis pipeline, we investigated the locomotor effects of several established drugs with well-defined impacts on neurophysiology. Here we show that machine learning model development, enabled by semi-supervised learning developed classification models, provide detailed insights into the behavioral phenotypes of zebrafish, offering a powerful, high throughput method for studying neural control of behavior.