arXiv reaDer
パイプ並列処理を使用したグラフニューラルネットワークのパフォーマンスの分析
Analyzing the Performance of Graph Neural Networks with Pipe Parallelism
機械学習と深層学習に遍在する多くの興味深いデータセットは、グラフを介して説明できます。広大なソーシャルネットワーク、タンパク質の折り畳み、化学相互作用ネットワーク、材料の相転移など、グラフ構造のデータセットの規模と複雑さが増すにつれて、これらに適用される機械学習技術の効率を向上させることが重要です。この研究では、ノードまたはエッジの分類やリンク予測などのタスクで大きな成功を収めているグラフニューラルネットワーク(GNN)に焦点を当てます。ただし、標準のGNNモデルには、メモリと実行時のボトルネックにつながる密グラフの関係を通じて実行される必要な再帰計算のために、スケーリングの制限があります。グラフ手法を進歩させるには、より大規模なネットワークを処理するための新しいアプローチが必要であり、いくつか提案されていますが、ディープラーニングコミュニティで成功することが知られている既存のツールとフレームワークを使用してGNNを並列化する方法を研究します。特に、2018年にGoogleによって導入されたGPipeを使用してパイプライン並列処理をGNNモデルに適用することを調査します。
Many interesting datasets ubiquitous in machine learning and deep learning can be described via graphs. As the scale and complexity of graph-structured datasets increase, such as in expansive social networks, protein folding, chemical interaction networks, and material phase transitions, improving the efficiency of the machine learning techniques applied to these is crucial. In this study, we focus on Graph Neural Networks (GNN) that have found great success in tasks such as node or edge classification and link prediction. However, standard GNN models have scaling limits due to necessary recursive calculations performed through dense graph relationships that lead to memory and runtime bottlenecks. While new approaches for processing larger networks are needed to advance graph techniques, and several have been proposed, we study how GNNs could be parallelized using existing tools and frameworks that are known to be successful in the deep learning community. In particular, we investigate applying pipeline parallelism to GNN models with GPipe, introduced by Google in 2018.
updated: Mon Apr 05 2021 16:59:33 GMT+0000 (UTC)
published: Sun Dec 20 2020 04:20:38 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アソシエイト