Meta

PyTorch-BigGraph: Faster embeddings of extremely large graphs

Apr 3, 2019
PyTorch-BigGraph: Faster embeddings of extremely large graphs
SummaryAI, from the source post

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 on single or multiple machines, and outputs node embeddings. It is built on PyTorch, allowing for easy customization of loss functions and models. PBG is designed to be robust, scalable, and orders of magnitude faster than commonly used embedding software, producing comparable quality embeddings.

More from the AI Research and Development trail