Discriminant Space Metric Network for Few-Shot Image Classification

Yan Leilei,Li Fanzhang,Zhang Li,Zheng Xiaohan
DOI: https://doi.org/10.1007/s10489-022-04413-3
IF: 5.3
2023-01-01
Applied Intelligence
Abstract:Metric-based few-shot learning has gained considerable attention for simply and effectively addressing the few-shot classification problem. However, a huge number of the existing approaches focus only on the similarity or distance between features of the instances in the embedding space, neglecting the geometric structure of the samples. To remedy this, we propose a novel approach referred to as the discriminant space metric network (DSMNet) for few-shot image classification problem. DSMNet exploits the geometric structure of the samples within each episode to enhance the discriminative ability of the embedding space. Specifically, DSMNet aims to increase the distance between features belonging to different classes while making those from the same class more compact by maximizing the between-class scatter and minimizing the within-class scatter in the embedding space. Moreover, we developed a novel adaptation strategy for improving the model’s generalizing capability. Extensive experiments are conducted on four few-shot classification benchmark datasets to demonstrate the proposed DSMNet. We also performed several ablation studies to analyze its performance. The superiority of DSMNet over existing networks is indicated by the experimental results.
What problem does this paper attempt to address?