SuperAnimal pretrained pose estimation models for behavioral analysis
Quantification of behavior is critical in diverse applications from neuroscience, veterinary medicine to animal conservation. A common key step for behavioral analysis is first extracting relevant keypoints on animals, known as pose estimation. However, reliable inference of poses currently requires domain knowledge and manual labeling effort to build supervised models. We present SuperAnimal, a method to develop unified foundation models that can be used on over 45 species, without additional manual labels. These models show excellent performance across six pose estimation benchmarks. We demonstrate how to fine-tune the models (if needed) on differently labeled data and provide tooling for unsupervised video adaptation to boost performance and decrease jitter across frames. If fine-tuned, SuperAnimal models are 10–100× more data efficient than prior transfer-learning-based approaches. We illustrate the utility of our models in behavioral classification and kinematic analysis. Collectively, we present a data-efficient solution for animal pose estimation.
2-s2.0-85196516003
38906853
EPFL
École Polytechnique Fédérale de Lausanne
École Polytechnique Fédérale de Lausanne
École Polytechnique Fédérale de Lausanne
École Polytechnique Fédérale de Lausanne
École Polytechnique Fédérale de Lausanne
EPFL
EPFL
2024-06-21
15
1
5165
Link to the code to use the DeepLabCut Model Zoo
Link to the code and data to reproduce the figures
Data availability
REVIEWED
EPFL
| Funder | Funding(s) | Grant Number | Grant URL |
Vallee Foundation | |||
Silicon Valley Community Foundation | |||
EPFL | |||
| Show more | |||