Object-Aware Attention in Few-Shot Learning

Yeqing Shen,Lisha Mo,Huimin Ma,Tianyu Hu,Yuhan Dong
DOI: https://doi.org/10.1007/978-981-16-7189-0_8
2021-01-01
Abstract:Embedding networks trained with a limited number of samples have a poor capability in localizing objects. Therefore, models of few-shot learning (FSL) are easily affected by object-irrelevant information in the background, which will lead to low accuracy. An Object-Aware Attention (OAA) mechanism is proposed to improve the generalization ability of models. In OAA module, object-relevant area is obtained by a fully convolutional network to guide the network in extracting object-relevant features. Besides, a general few-shot learning framework with OAA as a plug-and-play module is proposed, in which original images and object-aware images are fused to get the rectified prototypes. Under the general framework, the performance of most existing few-shot learning methods can be improved effectively. Comprehensive experiments show that the OAA can improve the accuracy of four mainstream baselines significantly. On benchmark mini-ImageNet, the method achieves a state-of-the-art performance on the 5-way-1-shot task and 5-way-5-shot task.
What problem does this paper attempt to address?