既存のグラフネットワークベースの数ショット学習方法は、畳み込みニューラルネットワーク(CNN)を介してノード間の類似性を取得します。ただし、CNNは、ベクトル形式のノード機能ではなく、空間情報を含む画像データ用に設計されています。この論文では、ノード間の類似性を暗黙的に更新するためにゲート付き回帰ユニットを利用する、数ショット学習のためのエッジラベリングベースの有向ゲートグラフネットワーク(DGGN)を提案しました。 DGGNは、ゲート付きノード集約モジュールと、改良されたゲート付き回帰ユニット(GRU)ベースのエッジ更新モジュールで構成されています。具体的には、ノード更新モジュールは、エッジ機能のアクティブ化を使用するゲートメカニズムを採用し、学習可能なノード集約プロセスを作成します。さらに、改善されたGRUセルは、ノード間の類似性を計算するためにエッジ更新手順で使用されます。さらに、このメカニズムは、レイヤーをまたがるGRUシーケンスを介した勾配バックプロパゲーションに役立ちます。 2つのベンチマークデータセットで実施された実験結果は、DGGNが最先端の方法と同等のパフォーマンスを達成していることを示しています。
Existing graph-network-based few-shot learning methods obtain similarity between nodes through a convolution neural network (CNN). However, the CNN is designed for image data with spatial information rather than vector form node feature. In this paper, we proposed an edge-labeling-based directed gated graph network (DGGN) for few-shot learning, which utilizes gated recurrent units to implicitly update the similarity between nodes. DGGN is composed of a gated node aggregation module and an improved gated recurrent unit (GRU) based edge update module. Specifically, the node update module adopts a gate mechanism using activation of edge feature, making a learnable node aggregation process. Besides, improved GRU cells are employed in the edge update procedure to compute the similarity between nodes. Further, this mechanism is beneficial to gradient backpropagation through the GRU sequence across layers. Experiment results conducted on two benchmark datasets show that our DGGN achieves a comparable performance to the-state-of-art methods.