HyperAIHyperAI

Command Palette

Search for a command to run...

PyTorch-BigGraph : Un Système d'Embedding de Graphe à Grande Échelle

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

Résumé

Les méthodes d'embedding de graphes produisent des caractéristiques de nœuds non supervisées à partir de graphes, qui peuvent ensuite être utilisées pour diverses tâches d'apprentissage automatique. Les graphes modernes, en particulier dans les applications industrielles, contiennent des milliards de nœuds et des billions d'arêtes, ce qui dépasse les capacités des systèmes d'embedding existants. Nous présentons PyTorch-BigGraph (PBG), un système d'embedding qui intègre plusieurs modifications aux systèmes traditionnels d'embedding multi-relationnels, permettant ainsi de traiter des graphes comportant des milliards de nœuds et des billions d'arêtes. PBG utilise la partition de graphe pour former des embeddings arbitrairement grands sur une seule machine ou dans un environnement distribué. Nous démontrons une performance comparable à celle des systèmes d'embedding existants sur des benchmarks courants, tout en permettant l'évolutivité vers des graphes arbitrairement grands et la parallélisation sur plusieurs machines. Nous formons et évaluons des embeddings sur plusieurs grands graphes de réseaux sociaux ainsi que sur l'ensemble du jeu de données Freebase, qui contient plus de 100 millions de nœuds et 2 milliards d'arêtes.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp
PyTorch-BigGraph : Un Système d'Embedding de Graphe à Grande Échelle | Articles | HyperAI