Minglei Yuan

Papers from this author

IFSM: An Iterative Feature Selection Mechanism for Few-Shot Image Classification

Chunhao Cai, Minglei Yuan, Tong Lu

Responsive image

Auto-TLDR; Iterative Feature Selection Mechanism for Few-Shot Learning

Slides Poster Similar

Nowadays many deep learning algorithms have been employed to solve different types of multimedia problems; however, most of them require a great amount of training data and tend to struggle in few-shot learning tasks. On the other hand, those methods designed for few-shot learning usually face the difficulty that once one or more samples show a relatively large bias, the predicted result may be much less reliable due to the fact that the sample will cause a large shift of class-level features during few-shot learning. To solve this problem, this paper presents a novel and Iterative Feature Selection Mechanism (IFSM) for few-shot image classification, which can be applied to lots of metric-based few-shot learners. IFSM learns to construct a more feasible class-level feature which is less affected by samples with relatively large biases, using an iterative approach. The proposed mechanism is tested on three previous state-of-the-art few-shot learning methods, and the experimental results show that the proposed mechanism considerably improves (by 1% to 2%) the image classification accuracies of both methods on the miniImageNet, tieredImageNet or CUB benchmarks in 5-way 5-shot tasks. This approves the effectiveness and generality of the proposed mechanism.