HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Domain Generalization by Solving Jigsaw Puzzles

Fabio Maria Carlucci; Antonio D'Innocente; Silvia Bucci; Barbara Caputo; Tatiana Tommasi

Domain Generalization by Solving Jigsaw Puzzles

Abstract

Human adaptability relies crucially on the ability to learn and merge knowledge both from supervised and unsupervised learning: the parents point out few important concepts, but then the children fill in the gaps on their own. This is particularly effective, because supervised learning can never be exhaustive and thus learning autonomously allows to discover invariances and regularities that help to generalize. In this paper we propose to apply a similar approach to the task of object recognition across domains: our model learns the semantic labels in a supervised fashion, and broadens its understanding of the data by learning from self-supervised signals how to solve a jigsaw puzzle on the same images. This secondary task helps the network to learn the concepts of spatial correlation while acting as a regularizer for the classification task. Multiple experiments on the PACS, VLCS, Office-Home and digits datasets confirm our intuition and show that this simple method outperforms previous domain generalization and adaptation solutions. An ablation study further illustrates the inner workings of our approach.

Code Repositories

fmcarlucci/JigenDG
Official
pytorch
Mentioned in GitHub
Emma0118/domain-generalization
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
domain-generalization-on-nico-animalJiGen (Resnet-18)
Accuracy: 84.95
domain-generalization-on-nico-vehicleResNet-18
Accuracy: 77.39
domain-generalization-on-pacs-2JiGen (Alexnet)
Average Accuracy: 73.38
domain-generalization-on-pacs-2JiGen (Resnet-18)
Average Accuracy: 80.51
domain-generalization-on-pacs-2Deep All (Resnet-18)
Average Accuracy: 79.05
domain-generalization-on-pacs-2Deep All (Alexnet)
Average Accuracy: 71.52
image-classification-on-colored-mnist-withJiGen
Accuracy : 11.91

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
Domain Generalization by Solving Jigsaw Puzzles | Papers | HyperAI