arXiv reaDer
ViT/MLP-Mixer のグラフへの一般化
A Generalization of ViT/MLP-Mixer to Graphs
グラフ ニューラル ネットワーク (GNN) は、グラフ表現学習の分野で大きな可能性を示しています。標準 GNN は、複数の層を積み重ねることによってグラフ ドメイン全体に情報を伝播するローカル メッセージ パッシング メカニズムを定義します。このパラダイムには、過剰な潰しと不十分な長距離依存性という 2 つの大きな制限があります。これらは全体的な注意を払えば解決できますが、二次計算の複雑さまで計算コストが大幅に増加します。この研究では、コンピュータ ビジョンで導入された ViT/MLP-Mixer アーキテクチャを活用することで、これらの構造的制限を克服するための代替アプローチを提案します。 Graph ViT/MLP-Mixer と呼ばれる、3 つの重要なプロパティを保持する新しいクラスの GNN を導入します。まず、Long Range Graph Benchmark と TreeNeighbourMatch データセットで実証されているように、長距離の依存関係をキャプチャし、過剰潰しの問題を軽減します。第 2 に、ノードとエッジの数に比例した複雑さで速度とメモリ効率が向上し、関連するグラフ トランスフォーマーや表現力豊かな GNN モデルを上回ります。第三に、少なくとも 3-WL の非同型グラフを区別できるため、グラフの同型性に関して高い表現力を示します。私たちは 4 つのシミュレートされたデータセットと 7 つの現実世界のベンチマークでアーキテクチャをテストし、それらすべてで非常に競争力のある結果を示しました。ソース コードは、https://github.com/XiaoxinHe/Graph-ViT-MLPMixer で再現可能です。
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 ViT/MLP-Mixer, that holds three key properties. First, they capture long-range dependency and mitigate the issue of over-squashing as demonstrated on Long Range Graph Benchmark and 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. The source code is available for reproducibility at: https://github.com/XiaoxinHe/Graph-ViT-MLPMixer.
updated: Wed May 31 2023 03:19:44 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アソシエイト