arXiv reaDer
説明-クロスドメインの少数ショット分類のためのガイド付きトレーニング
Explanation-Guided Training for Cross-Domain Few-Shot Classification
クロスドメイン少数ショット分類タスク(CD-FSC)は、少数ショット分類と、データセットで表されるドメイン全体を一般化する要件を組み合わせたものです。この設定は、各クラスの限られたラベル付きデータに起因する課題に直面し、さらに、トレーニングセットとテストセット間のドメインシフトに起因します。この論文では、既存のFSCモデルの新しいトレーニングアプローチを紹介します。これは、FSCモデルの予測に適用されたときに既存の説明方法から取得され、モデルの中間特徴マップに対して計算された説明スコアを活用します。まず、FSCモデルの予測を説明するために、レイヤーワイズ関連性伝播(LRP)メソッドを調整します。次に、予測に重要な機能を動的に見つけて強調する、モデルにとらわれない説明に基づくトレーニング戦略を開発します。私たちの貢献は、新しい説明方法を対象とはしていませんが、トレーニングフェーズの説明の新しいアプリケーションにあります。説明に基づくトレーニングがモデルの一般化を効果的に改善することを示します。 miniImagenet、CUB、Cars、Places、Plantaeの5つの数ショット学習データセットで、RelationNet、クロスアテンションネットワーク、グラフニューラルネットワークベースの定式化の3つの異なるFSCモデルの精度が向上していることを確認します。ソースコードはhttps://github.com/SunJiamei/few-shot-lrp-guidedで入手できます。
Cross-domain few-shot classification task (CD-FSC) combines few-shot classification with the requirement to generalize across domains represented by datasets. This setup faces challenges originating from the limited labeled data in each class and, additionally, from the domain shift between training and test sets. In this paper, we introduce a novel training approach for existing FSC models. It leverages on the explanation scores, obtained from existing explanation methods when applied to the predictions of FSC models, computed for intermediate feature maps of the models. Firstly, we tailor the layer-wise relevance propagation (LRP) method to explain the predictions of FSC models. Secondly, we develop a model-agnostic explanation-guided training strategy that dynamically finds and emphasizes the features which are important for the predictions. Our contribution does not target a novel explanation method but lies in a novel application of explanations for the training phase. We show that explanation-guided training effectively improves the model generalization. We observe improved accuracy for three different FSC models: RelationNet, cross attention network, and a graph neural network-based formulation, on five few-shot learning datasets: miniImagenet, CUB, Cars, Places, and Plantae. The source code is available at https://github.com/SunJiamei/few-shot-lrp-guided
updated: Wed Dec 09 2020 09:53:24 GMT+0000 (UTC)
published: Fri Jul 17 2020 07:28:08 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト