arXiv reaDer
ViT/MLP-Mixer のグラフへの一般化
A Generalization of ViT/MLP-Mixer to Graphs
グラフ ニューラル ネットワーク (GNN) は、グラフ表現学習の分野で大きな可能性を示しています。標準 GNN は、複数のレイヤーを積み重ねることによってグラフ ドメイン全体に情報を伝達するローカル メッセージ パッシング メカニズムを定義します。このパラダイムには、グローバルな注意を使用して解決できるが、計算コストが大幅に増加して 2 次の複雑さになるという 2 つの主要な制限があります。この作業では、コンピューター ビジョンに導入された ViT/MLP-Mixer アーキテクチャを活用して、これらの構造上の制限を克服するための代替アプローチを提案します。 Graph MLP-Mixer と呼ばれる新しいクラスの GNN を導入します。これは 3 つの主要なプロパティを保持します。まず、Long Range Graph Benchmark (LRGB) と TreeNeighbourMatch データセットで実証されているように、長期的な依存関係をキャプチャし、過圧縮の問題を軽減します。第 2 に、ノードとエッジの数に比例した複雑さで速度とメモリ効率が向上し、関連するグラフ トランスフォーマーや表現力豊かな GNN モデルを凌駕します。第 3 に、少なくとも 3-WL 非同型グラフを区別できるため、グラフ同型に関して高い表現力を示します。 4 つのシミュレートされたデータセットと 7 つの実世界のベンチマークでアーキテクチャをテストし、それらすべてで非常に競争力のある結果を示しています。
Graph Neural Networks (GNNs) have shown great potential in the field of graph representation learning. Standard GNNs define a local message-passing mechanism which propagates information over the whole graph domain by stacking multiple layers. This paradigm suffers from two major limitations, over-squashing and poor long-range dependencies, that can be solved using global attention but significantly increases the computational cost to quadratic complexity. In this work, we propose an alternative approach to overcome these structural limitations by leveraging the ViT/MLP-Mixer architectures introduced in computer vision. We introduce a new class of GNNs, called Graph MLP-Mixer, that holds three key properties. First, they capture long-range dependency and mitigate the issue of over-squashing as demonstrated on the Long Range Graph Benchmark (LRGB) and the TreeNeighbourMatch datasets. Second, they offer better speed and memory efficiency with a complexity linear to the number of nodes and edges, surpassing the related Graph Transformer and expressive GNN models. Third, they show high expressivity in terms of graph isomorphism as they can distinguish at least 3-WL non-isomorphic graphs. We test our architecture on 4 simulated datasets and 7 real-world benchmarks, and show highly competitive results on all of them.
updated: Tue Dec 27 2022 03:27:46 GMT+0000 (UTC)
published: Tue Dec 27 2022 03:27:46 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アソシエイト