HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Pretraining boosts out-of-domain robustness for pose estimation

Mathis Alexander ; Biasi Thomas ; Schneider Steffen ; Yüksekgönül Mert ; Rogers Byron ; Bethge Matthias ; Mathis Mackenzie W.

Pretraining boosts out-of-domain robustness for pose estimation

Abstract

Neural networks are highly effective tools for pose estimation. However, asin other computer vision tasks, robustness to out-of-domain data remains achallenge, especially for small training sets that are common for real-worldapplications. Here, we probe the generalization ability with three architectureclasses (MobileNetV2s, ResNets, and EfficientNets) for pose estimation. Wedeveloped a dataset of 30 horses that allowed for both "within-domain" and"out-of-domain" (unseen horse) benchmarking - this is a crucial test forrobustness that current human pose estimation benchmarks do not directlyaddress. We show that better ImageNet-performing architectures perform betteron both within- and out-of-domain data if they are first pretrained onImageNet. We additionally show that better ImageNet models generalize betteracross animal species. Furthermore, we introduce Horse-C, a new benchmark forcommon corruptions for pose estimation, and confirm that pretraining increasesperformance in this domain shift context as well. Overall, our resultsdemonstrate that transfer learning is beneficial for out-of-domain robustness.

Code Repositories

DeepLabCut/DeepLabCut
Official
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
animal-pose-estimation-on-horse-10DeepLabCut-EfficientNet-B6
PCK@0.3 (OOD): 88.4
animal-pose-estimation-on-horse-10DeepLabCut-MOBILENETV2-1
PCK@0.3 (OOD): 77.6
animal-pose-estimation-on-horse-10DeepLabCut-EfficientNet-B4
PCK@0.3 (OOD): 86.9
animal-pose-estimation-on-horse-10DeepLabCut-RESNET-101
PCK@0.3 (OOD): 84.3
animal-pose-estimation-on-horse-10DeepLabCut-MOBILENETV2 0.35
PCK@0.3 (OOD): 63.5
animal-pose-estimation-on-horse-10DeepLabCut-RESNET 50
PCK@0.3 (OOD): 81.3

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Pretraining boosts out-of-domain robustness for pose estimation | Papers | HyperAI