Task Augmentation by Rotating for Meta-Learning

arXiv 2020  ·  Jialin Liu, Fei Chao, Chih-Min Lin ·

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}.

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Few-Shot Image Classification CIFAR-FS 5-way (1-shot) R2-D2+Task Aug Accuracy 77.66 # 17
Few-Shot Image Classification CIFAR-FS 5-way (1-shot) MetaOptNet-SVM+Task Aug Accuracy 76.75 # 19
Few-Shot Image Classification CIFAR-FS 5-way (5-shot) MetaOptNet-SVM+Task Aug Accuracy 88.38 # 19
Few-Shot Image Classification CIFAR-FS 5-way (5-shot) R2-D2+Task Aug Accuracy 88.33 # 21
Few-Shot Image Classification FC100 5-way (1-shot) MetaOptNet-SVM+Task Aug Accuracy 49.77 # 6
Few-Shot Image Classification FC100 5-way (1-shot) R2-D2+Task Aug Accuracy 51.35 # 4
Few-Shot Image Classification FC100 5-way (5-shot) R2-D2+Task Aug Accuracy 67.66 # 2
Few-Shot Image Classification FC100 5-way (5-shot) MetaOptNet-SVM+Task Aug Accuracy 67.17 # 3
Few-Shot Image Classification Mini-ImageNet - 1-Shot Learning R2-D2+Task Aug Accuracy 65.95% # 10
Few-Shot Image Classification Mini-ImageNet - 1-Shot Learning MetaOptNet-SVM+Task Aug Accuracy 65.38% # 11
Few-Shot Image Classification Mini-Imagenet 5-way (1-shot) MetaOptNet-SVM+Task Aug Accuracy 65.38 # 50
Few-Shot Image Classification Mini-Imagenet 5-way (1-shot) R2-D2+Task Aug Accuracy 65.95 # 47
Few-Shot Image Classification Mini-Imagenet 5-way (5-shot) R2-D2+Task Aug Accuracy 81.96 # 39
Few-Shot Image Classification Mini-Imagenet 5-way (5-shot) MetaOptNet-SVM+Task Aug Accuracy 82.13 # 38

Methods


No methods listed for this paper. Add relevant methods here