41
v1v2 (latest)

Animal behavioral analysis and neural encoding with transformer-based self-supervised pretraining

Yanchen Wang
Han Yu
Ari Blau
Yizi Zhang
International Brain Laboratory
Liam Paninski
Cole Hurwitz
Matt Whiteway
Main:11 Pages
13 Figures
Bibliography:7 Pages
15 Tables
Appendix:17 Pages
Abstract

The brain can only be fully understood through the lens of the behavior it generates -- a guiding principle in modern neuroscience research that nevertheless presents significant technical challenges. Many studies capture behavior with cameras, but video analysis approaches typically rely on specialized models requiring extensive labeled data. We address this limitation with BEAST(BEhavioral Analysis via Self-supervised pretraining of Transformers), a novel and scalable framework that pretrains experiment-specific vision transformers for diverse neuro-behavior analyses. BEAST combines masked autoencoding with temporal contrastive learning to effectively leverage unlabeled video data. Through comprehensive evaluation across multiple species, we demonstrate improved performance in three critical neuro-behavioral tasks: extracting behavioral features that correlate with neural activity, and pose estimation and action segmentation in both the single- and multi-animal settings. Our method establishes a powerful and versatile backbone model that accelerates behavioral analysis in scenarios where labeled data remains scarce.

View on arXiv
Comments on this paper