Learning Calibrated Class Centers for Few-Shot Classification by Pair-Wise Similarity | IEEE Journals & Magazine | IEEE Xplore

Learning Calibrated Class Centers for Few-Shot Classification by Pair-Wise Similarity


Abstract:

Metric-based methods achieve promising performance on few-shot classification by learning clusters on support samples and generating shared decision boundaries for query ...Show More

Abstract:

Metric-based methods achieve promising performance on few-shot classification by learning clusters on support samples and generating shared decision boundaries for query samples. However, existing methods ignore the inaccurate class center approximation introduced by the limited number of support samples, which consequently leads to biased inference. Therefore, in this paper, we propose to reduce the approximation error by class center calibration. Specifically, we introduce the so-called Pair-wise Similarity Module (PSM) to generate calibrated class centers adapted to the query sample by capturing the semantic correlations between the support and the query samples, as well as enhancing the discriminative regions on support representation. It is worth noting that the proposed PSM is a simple plug-and-play module and can be inserted into most metric-based few-shot learning models. Through extensive experiments in metric-based models, we demonstrate that the module significantly improves the performance of conventional few-shot classification methods on four few-shot image classification benchmark datasets. Codes are available at: https://github.com/PRIS-CV/Pair-wise-Similarity-module.
Published in: IEEE Transactions on Image Processing ( Volume: 31)
Page(s): 4543 - 4555
Date of Publication: 29 June 2022

ISSN Information:

PubMed ID: 35767479

Funding Agency:


Contact IEEE to Subscribe

References

References is not available for this document.