少数ショット画像分類は、限られたラベル付きデータから新しいカテゴリを認識することを学習します。メトリック学習ベースのアプローチは広く調査されており、クエリ サンプルは、特徴の類似性に基づいてサポート セットから最も近いプロトタイプを見つけることによって分類されます。ニューラル ネットワークには、異なるペアの計算された類似性に異なる不確実性があります。類似性の不確実性を理解してモデル化することで、少数ショットの最適化における限られたサンプルの活用を促進することができます。この作業では、クエリとサポートのペアの類似性の不確実性をモデル化し、不確実性を意識した最適化を実行することにより、画像分類のための不確実性を意識した少数ショット フレームワークを提案します。特に、観測された類似性を確率的表現に変換することによってそのような不確実性を利用し、より効果的な最適化のためにそれらを損失に組み込みます。サポート セット内のクエリとプロトタイプの間の類似性を一緒に検討するために、グラフ ベースのモデルを使用してペアの不確実性を推定します。広範な実験により、提案された方法が強力なベースラインに加えて大幅な改善をもたらし、最先端のパフォーマンスを達成することが示されています。
Few-shot image classification learns to recognize new categories from limited labelled data. Metric learning based approaches have been widely investigated, where a query sample is classified by finding the nearest prototype from the support set based on their feature similarities. A neural network has different uncertainties on its calculated similarities of different pairs. Understanding and modeling the uncertainty on the similarity could promote the exploitation of limited samples in few-shot optimization. In this work, we propose Uncertainty-Aware Few-Shot framework for image classification by modeling uncertainty of the similarities of query-support pairs and performing uncertainty-aware optimization. Particularly, we exploit such uncertainty by converting observed similarities to probabilistic representations and incorporate them to the loss for more effective optimization. In order to jointly consider the similarities between a query and the prototypes in a support set, a graph-based model is utilized to estimate the uncertainty of the pairs. Extensive experiments show our proposed method brings significant improvements on top of a strong baseline and achieves the state-of-the-art performance.