arXiv reaDer
TLRM:GNNベースの少数ショット学習のためのタスクレベルの関係モジュール
TLRM: Task-level Relation Module for GNN-based Few-Shot Learning
最近、グラフニューラルネットワーク(GNN)は、クラスごとに限られたラベル付きサンプルでトレーニングされたときに見えないサンプルを分類することを目的とした、数ショットの分類問題を処理する強力な機能を示しました。 GNNベースの数ショット学習アーキテクチャは、ほとんどの場合、従来のメトリックを学習可能なGNNに置き換えます。 GNNでは、ノードが埋め込みサンプルとして設定され、接続された2つのノード間の関係は、ネットワークによって取得できます。ネットワークの入力は、埋め込み機能の違いです。サンプルの関係を測定するこの方法は、サンプル間の関係をモデル化するだけであり、さまざまなタスクの特異性を無視していると考えています。つまり、関係を測定するこの方法では、タスクレベルの情報は考慮されません。この目的のために、1つのサンプルと他のすべてのサンプルのタスクレベルの関係を明示的にモデル化するために、新しい関係測定方法、つまりタスクレベルの関係モジュール(TLRM)を提案します。提案されたモジュールは、サンプル間の埋め込み機能ではなく、サンプル間の埋め込み機能を検討することにより、ノード間の関係表現をキャプチャします。 mini-ImageNet、tiered-ImageNet、CUB-200-2011、CIFAR-FSの4つのベンチマークデータセットで広範な実験を実施しました。実験結果は、提案されたモジュールがGNNベースの数ショット学習に効果的であることを示しています。
Recently, graph neural networks (GNNs) have shown powerful ability to handle few-shot classification problem, which aims at classifying unseen samples when trained with limited labeled samples per class. GNN-based few-shot learning architectures mostly replace traditional metric with a learnable GNN. In the GNN, the nodes are set as the samples embedding, and the relationship between two connected nodes can be obtained by a network, the input of which is the difference of their embedding features. We consider this method of measuring relation of samples only models the sample-to-sample relation, while neglects the specificity of different tasks. That is, this method of measuring relation does not take the task-level information into account. To this end, we propose a new relation measure method, namely the task-level relation module (TLRM), to explicitly model the task-level relation of one sample to all the others. The proposed module captures the relation representations between nodes by considering the sample-to-task instead of sample-to-sample embedding features. We conducted extensive experiments on four benchmark datasets: mini-ImageNet, tiered-ImageNet, CUB-200-2011, and CIFAR-FS. Experimental results demonstrate that the proposed module is effective for GNN-based few-shot learning.
updated: Thu Sep 02 2021 12:32:34 GMT+0000 (UTC)
published: Mon Jan 25 2021 00:53:04 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アソシエイト