错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Explore Across-Dimensional Feature Correlations for Few-Shot Learning

  • Tianyuan Li,
  • Wenming Cao

摘要

Few-shot learning (FSL) aims to learn new concepts with only few examples, which is a challenging problem in deep learning. Attention-oriented methods have shown great potential in addressing the FSL problem. However, many of these methods tend to separately focus on different dimensions of the samples, like channel dimension or spatial dimension only, which may lead to limitations in extracting discriminative features. To address this problem, we propose an across-dimensional attention network (ADANet) to explore the feature correlations across channel and spatial dimensions of input samples. The ADANet can capture across-dimensional feature dependencies and produce reliable representations for the similarity metric. In addition, we also design a three-dimensional offset position encoding (TOPE) method that embeds the 3D position information into the across-dimensional attention, enhancing the robustness and generalization ability of the model. Extensive experiments on standard FSL benchmarks have demonstrated that our method can achieve better performances compared to the state-of-the-art approaches.