Variational Neuron Shifting for Few-Shot Image Classification Across Domains
Liyun Zuo,Baoyan Wang,Lei Zhang,Jun Xu,Xiantong Zhen
DOI: https://doi.org/10.1109/tmm.2023.3282492
IF: 7.3
2023-01-01
IEEE Transactions on Multimedia
Abstract:Few-shot image classification aims to recognize unseen classes with few labeled samples. Existing meta-learning models learn good representations or model parameters to adapt to new tasks with a few training samples. However, when there a domain gap exists between training and test tasks, the learned ability often does not generalize well across domains, resulting in degraded performance on new tasks. In this paper, we propose variational neuron shifting to generate adapted feature representations for few-shot learning. We introduce a working memory module to store the shifted neurons from the support set, which will be accessed to generate adapted feature representations of query samples. Under the meta-learning paradigm, the model is learned to acquire the ability of adaptation with a single sample at meta-training time to further adapt itself to each single test sample at meta-test time. We formulate the adaptation process as a variational Bayesian inference problem, which incorporates the test sample as the condition into the generation of the model neuron shifting. We conduct extensive experiments on both within- and across-domain few-shot classification tasks. Good performance substantiates the effectiveness of the proposed variational neuron shifting, and thorough ablation studies demonstrate the benefit of each component in the proposed model.
computer science, information systems,telecommunications, software engineering