References & Citations
Computer Science > Computer Vision and Pattern Recognition
Title: Sparse Spatial Transformers for Few-Shot Learning
(Submitted on 27 Sep 2021 (v1), last revised 10 May 2023 (this version, v3))
Abstract: Learning from limited data is challenging because data scarcity leads to a poor generalization of the trained model. A classical global pooled representation will probably lose useful local information. Many few-shot learning methods have recently addressed this challenge using deep descriptors and learning a pixel-level metric. However, using deep descriptors as feature representations may lose image contextual information. Moreover, most of these methods independently address each class in the support set, which cannot sufficiently use discriminative information and task-specific embeddings. In this paper, we propose a novel transformer-based neural network architecture called sparse spatial transformers (SSFormers), which finds task-relevant features and suppresses task-irrelevant features. Particularly, we first divide each input image into several image patches of different sizes to obtain dense local features. These features retain contextual information while expressing local information. Then, a sparse spatial transformer layer is proposed to find spatial correspondence between the query image and the full support set to select task-relevant image patches and suppress task-irrelevant image patches. Finally, we propose using an image patch-matching module to calculate the distance between dense local representations, thus determining which category the query image belongs to in the support set. Extensive experiments on popular few-shot learning benchmarks demonstrate the superiority of our method over state-of-the-art methods. Our source code is available at \url{this https URL}.
Submission history
From: Huaxiong Li [view email][v1] Mon, 27 Sep 2021 10:36:32 GMT (1594kb,D)
[v2] Mon, 6 Dec 2021 01:52:08 GMT (2894kb,D)
[v3] Wed, 10 May 2023 01:53:11 GMT (3775kb,D)
Link back to: arXiv, form interface, contact.