AI Research and Development
PyTorch-BigGraph: Faster embeddings of extremely large graphs

PyTorch-BigGraph: Faster embeddings of extremely large graphs

4/3/2019

What this post added

Introduced PyTorch-BigGraph (PBG), a new tool for training multi-relation graph embeddings on extremely large graphs (billions of nodes, trillions of edges). PBG shards nodes and edges, performs multi-threaded training (single or multiple machines), and outputs node embeddings. It is built on PyTorch, allowing for custom loss functions and models. PBG is designed to be orders of magnitude faster than commonly used embedding software while producing comparable quality embeddings, making graph embedding accessible to researchers and organizations with large graph datasets but limited specialized computing resources.

Read the original post ↗