HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Bayesian Meta-Learning for the Few-Shot Setting via Deep Kernels

Massimiliano Patacchiola Jack Turner Elliot J. Crowley Michael O&#39 Boyle Amos Storkey

Bayesian Meta-Learning for the Few-Shot Setting via Deep Kernels

Abstract

Recently, different machine learning methods have been introduced to tackle the challenging few-shot learning scenario that is, learning from a small labeled dataset related to a specific task. Common approaches have taken the form of meta-learning: learning to learn on the new problem given the old. Following the recognition that meta-learning is implementing learning in a multi-level model, we present a Bayesian treatment for the meta-learning inner loop through the use of deep kernels. As a result we can learn a kernel that transfers to new tasks; we call this Deep Kernel Transfer (DKT). This approach has many advantages: is straightforward to implement as a single optimizer, provides uncertainty quantification, and does not require estimation of task-specific parameters. We empirically demonstrate that DKT outperforms several state-of-the-art algorithms in few-shot classification, and is the state of the art for cross-domain adaptation and regression. We conclude that complex meta-learning routines can be replaced by a simpler Bayesian model without loss of accuracy.

Code Repositories

keanson/revisit-logistic-softmax
pytorch
Mentioned in GitHub
BayesWatch/deep-kernel-transfer
Official
pytorch
Mentioned in GitHub
hhl60492/deep-kernel-transfer
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
few-shot-image-classification-on-cub-200-5DKT + BNCosSim
Accuracy: 85.64
few-shot-image-classification-on-cub-200-5-1DKT + BNCosSim
Accuracy: 72.27
few-shot-image-classification-on-mini-2DKT + BNCosSim
Accuracy: 62.96
few-shot-image-classification-on-mini-3DKT + BNCosSim
Accuracy: 64.0
few-shot-image-classification-on-mini-5DKT + CosSim
Accuracy: 40.22
few-shot-image-classification-on-mini-6DKT + BNCosSim
Accuracy: 56.40
few-shot-image-classification-on-omniglotDKT + BNCosSim
Accuracy: 75.40
few-shot-image-classification-on-omniglot-2DKT + BNCosSim
Accuracy: 90.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
Bayesian Meta-Learning for the Few-Shot Setting via Deep Kernels | Papers | HyperAI