HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

ChimeraMix: Image Classification on Small Datasets via Masked Feature Mixing

Christoph Reinders; Frederik Schubert; Bodo Rosenhahn

ChimeraMix: Image Classification on Small Datasets via Masked Feature Mixing

Abstract

Deep convolutional neural networks require large amounts of labeled data samples. For many real-world applications, this is a major limitation which is commonly treated by augmentation methods. In this work, we address the problem of learning deep neural networks on small datasets. Our proposed architecture called ChimeraMix learns a data augmentation by generating compositions of instances. The generative model encodes images in pairs, combines the features guided by a mask, and creates new samples. For evaluation, all methods are trained from scratch without any additional data. Several experiments on benchmark datasets, e.g. ciFAIR-10, STL-10, and ciFAIR-100, demonstrate the superior performance of ChimeraMix compared to current state-of-the-art methods for classification on small datasets.

Code Repositories

creinders/chimeramix
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
small-data-image-classification-on-cifair-10-1ChimeraMix+AutoAugment
Accuracy: 70.09
small-data-image-classification-on-cifair-10-1ChimeraMix
Accuracy: 67.30
small-data-image-classification-on-cifar-10ChimeraMix+AutoAugment
Accuracy (%): 70.09
small-data-image-classification-on-cifar-10ChimeraMix
Accuracy (%): 67.3
small-data-image-classification-on-cifar-10-2ChimeraMix+AutoAugment
Accuracy (%): 49.75
small-data-image-classification-on-cifar-10-2ChimeraMix
Accuracy (%): 47.6
small-data-image-classification-on-cifar-10-3ChimeraMix+AutoAugment
Accuracy (%): 76.76
small-data-image-classification-on-cifar-10-3ChimeraMix
Accuracy (%): 74.96
small-data-on-cifar-100-1000-labels-1ChimeraMix
Accuracy: 32.72
small-data-on-cifar-100-1000-labels-1ChimeraMix+AutoAugment
Accuracy: 35.02

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