HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Image Deformation Meta-Networks for One-Shot Learning

Zitian Chen; Yanwei Fu; Yu-Xiong Wang; Lin Ma; Wei Liu; Martial Hebert

Image Deformation Meta-Networks for One-Shot Learning

Abstract

Humans can robustly learn novel visual concepts even when images undergo various deformations and lose certain information. Mimicking the same behavior and synthesizing deformed instances of new concepts may help visual recognition systems perform better one-shot learning, i.e., learning concepts from one or few examples. Our key insight is that, while the deformed images may not be visually realistic, they still maintain critical semantic information and contribute significantly to formulating classifier decision boundaries. Inspired by the recent progress of meta-learning, we combine a meta-learner with an image deformation sub-network that produces additional training examples, and optimize both models in an end-to-end manner. The deformation sub-network learns to deform images by fusing a pair of images --- a probe image that keeps the visual content and a gallery image that diversifies the deformations. We demonstrate results on the widely used one-shot learning benchmarks (miniImageNet and ImageNet 1K Challenge datasets), which significantly outperform state-of-the-art approaches. Code is available at https://github.com/tankche1/IDeMe-Net.

Code Repositories

tankche1/IDeMe-Net
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
few-shot-image-classification-on-imagenet-fsIDeMe-Net (ResNet-50)
Top-5 Accuracy (%): 60.1
few-shot-image-classification-on-imagenet-fs-1IDeMe-Net (ResNet-50)
Top-5 Accuracy (%): 69.6
few-shot-image-classification-on-imagenet-fs-6IdeMe-Net (ResNet-50)
Top-5 Accuracy (%): 77.4
few-shot-image-classification-on-mini-2IDeMe-Net
Accuracy: 59.14
few-shot-image-classification-on-mini-3IDeMe-Net
Accuracy: 74.63

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
Image Deformation Meta-Networks for One-Shot Learning | Papers | HyperAI