HyperAIHyperAI

Command Palette

Search for a command to run...

PyTorch-BigGraph: 大規模グラフ埋め込みシステム

Adam Lerer; Ledell Wu; Jiajun Shen; Timothee Lacroix; Luca Wehrstedt; Abhijit Bose; Alex Peysakhovich

概要

グラフ埋め込み手法は、グラフから教師なしでノードの特徴を生成し、機械学習タスクに利用することができます。現代のグラフ、特に産業応用におけるものは、数十億のノードと数兆のエッジを含んでおり、既存の埋め込みシステムの能力を超えています。本稿では、PyTorch-BigGraph (PBG) を紹介します。これは伝統的な多関係埋め込みシステムに対するいくつかの改良を加えたもので、数十億のノードと数兆のエッジを持つグラフに対してスケーリングが可能となっています。PBG はグラフ分割を使用して、単一マシンまたは分散環境において任意の大きさの埋め込みを訓練します。一般的なベンチマークでの性能は既存の埋め込みシステムと同等でありながら、任意の大規模グラフへのスケーリングや複数マシン間での並列化が可能です。私たちは PBG を用いて、大規模なソーシャルネットワークグラフや1億以上のノードと20億以上のエッジを含むフル Freebase データセット上で埋め込みを訓練し評価しました。


AIでAIを構築

アイデアからローンチまで — 無料のAIコーディング支援、すぐに使える環境、最高のGPU価格でAI開発を加速。

AI コーディング補助
すぐに使える GPU
最適な料金体系

HyperAI Newsletters

最新情報を購読する
北京時間 毎週月曜日の午前9時 に、その週の最新情報をメールでお届けします
メール配信サービスは MailChimp によって提供されています