PyTorch : DGL Tutorials : Basics : ひとめでわかる DGL (翻訳/解説)
【PyTorch: Deep Graph Library Tutorials : ひとめでわかる DGL】
DGL (Deep Graph Library) はグラフ上の深層学習専用の Python パッケージです。PyTorch 上に構築されていて PyTorch のエコシステムの一つにもなっています。
グラフニューラルネットワークの実装が単純化され、数十億のノード/エッジを持つグラフ上の半教師あり学習、グラフ上の生成モデル、TreeLSTM のような樹木モデル etc. が実装可能です。GCN (グラフ畳み込みネットワーク) モデルの実装は最初のチュートリアルで紹介されます。
幾つかチュートリアルが用意されていますので、順次翻訳していきます。最初のチュートリアルの目標は、(1) DGL がどのようにグラフ上の計算を可能にするか高位レベルで理解する、そして (2) グラフのノードを分類するために DGL で単純なグラフニューラルネットワークを作成して訓練することです。良く知られたコミュニティ分割問題「Zachary の空手クラブ」を題材としています。