arXiv reaDer
署名グラフネットワーク
Signature-Graph Networks
署名グラフニューラルネットワーク(SGN)と呼ばれる視覚表現学習のための新しいアプローチを提案します。 SGNは、畳み込みニューラルネットワーク(CNN)の機能表現を補強する潜在的なグローバル構造を学習します。 SGNは、CNN特徴マップに基づいて、画像ごとに一意の無向グラフを作成します。フィーチャマップは、等しく重複しないパッチのセットに分割されます。グラフノードは、これらのパッチの極大値または極小値を持つ高コントラストの鋭い畳み込み特徴に配置されます。ノードの埋め込みは、水平および垂直のエッジ接続に基づく新しい署名グラフによって集約されます。次に、グラフのスペクトルラプラシアン固有値に基づいて表現ベクトルが計算されます。 SGNは、最近のグラフ畳み込みネットワーク、生成的敵対的ネットワーク、および自動エンコーダーの既存の方法よりも優れており、画像分類の精度はASIRRAで99.65%、MNISTで99.91%、Fashion-MNISTで98.55%、CIFAR-10で96.18%、CIFAR-10で84.71%です。 CIFAR-100、STL10で94.36%、SVHNデータセットで95.86%。また、提案されたSGNに加えて、最先端のマルチヘッドアテンション(MHA)の新しい実装を紹介します。 SGNをMHAに追加すると、STL10データセットの画像分類精度が86.92%から94.36%に向上しました。
We propose a novel approach for visual representation learning called Signature-Graph Neural Networks (SGN). SGN learns latent global structures that augment the feature representation of Convolutional Neural Networks (CNN). SGN constructs unique undirected graphs for each image based on the CNN feature maps. The feature maps are partitioned into a set of equal and non-overlapping patches. The graph nodes are located on high-contrast sharp convolution features with the local maxima or minima in these patches. The node embeddings are aggregated through novel Signature-Graphs based on horizontal and vertical edge connections. The representation vectors are then computed based on the spectral Laplacian eigenvalues of the graphs. SGN outperforms existing methods of recent graph convolutional networks, generative adversarial networks, and auto-encoders with image classification accuracy of 99.65% on ASIRRA, 99.91% on MNIST, 98.55% on Fashion-MNIST, 96.18% on CIFAR-10, 84.71% on CIFAR-100, 94.36% on STL10, and 95.86% on SVHN datasets. We also introduce a novel implementation of the state-of-the-art multi-head attention (MHA) on top of the proposed SGN. Adding SGN to MHA improved the image classification accuracy from 86.92% to 94.36% on the STL10 dataset
updated: Fri Oct 22 2021 01:44:14 GMT+0000 (UTC)
published: Fri Oct 22 2021 01:44:14 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アソシエイト