Learn About Our Meetup

4500+ Members

[Project] GraphVite: A High-Performance CPU-GPU Hybrid System for Node Embedding

We just released a general and high-performance graph embedding system, GraphVite.

Compared to existing machine learning systems that are mainly designed for data with regular structures (e.g., images, speech, and natural language), GraphVite is specifically designed for large-scale graphs. It runs on the CPU-GPU hybrid architectures and scales linearly to the number of GPUs. The system is one or two magnitudes faster than existing implementations. For example, for a graph with one million nodes, it only takes around one minute to learn the node representations with 4 GPUs. Besides the superior efficiency, GraphVite also supports a variety of applications and models, including

  • Node Embedding: DeepWalk, LINE, node2vec
  • Knowledge Graph Embedding: TransE, DistMult, ComplEx, SimplE, RotatE
  • Graph and High-dimensional Data Visualization: LargeVis

There are already more than 30 configurations and benchmarks on standard datasets. We are actively developing new applications and models. The system is expected to support the community of graph embedding or in general, deep learning for graphs.

submitted by /u/kiddozhu
[link] [comments]

Next Meetup




Plug yourself into AI and don't miss a beat