Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Task Augmentation by Rotating for Meta-Learning

About

Data augmentation is one of the most effective approaches for improving the accuracy of modern machine learning models, and it is also indispensable to train a deep model for meta-learning. In this paper, we introduce a task augmentation method by rotating, which increases the number of classes by rotating the original images 90, 180 and 270 degrees, different from traditional augmentation methods which increase the number of images. With a larger amount of classes, we can sample more diverse task instances during training. Therefore, task augmentation by rotating allows us to train a deep network by meta-learning methods with little over-fitting. Experimental results show that our approach is better than the rotation for increasing the number of images and achieves state-of-the-art performance on miniImageNet, CIFAR-FS, and FC100 few-shot learning benchmarks. The code is available on \url{www.github.com/AceChuse/TaskLevelAug}.

Jialin Liu, Fei Chao, Chih-Min Lin• 2020

Related benchmarks

TaskDatasetResultRank
Few-shot Image ClassificationMini-Imagenet (test)
Accuracy82.13
235
Few-shot Image ClassificationFC100 5-way 1-shot (test)
Average Accuracy51.35
28
Few-shot Image ClassificationFC100 5-way 5-shot (test)
Accuracy67.66
28
Few-shot Image ClassificationMiniImageNet ILSVRC-2012 (test)
Accuracy81.08
22
Few-shot Image ClassificationCIFAR-FS 100 variant (test)
Accuracy87.63
12
Image ClassificationCIFAR-FS 1-shot (test)
Accuracy77.66
6
Image ClassificationCIFAR-FS 5-shot (test)
Accuracy88.38
6
Showing 7 of 7 rows

Other info

Code

Follow for update