TL;DR
GOSH is a novel graph embedding method that efficiently compresses and decomposes large graphs, enabling high-quality embeddings on a single GPU for graphs with millions of vertices and edges, significantly reducing computational resources.
Contribution
It introduces a new graph coarsening and decomposition approach that allows embedding of arbitrarily large graphs on minimal hardware, outperforming existing methods in speed and resource usage.
Findings
Embedded a 65 million vertex graph in under an hour on a single GPU.
Achieved 93% AUCROC in link prediction, improved to 95% with additional runtime.
Outperformed state-of-the-art methods in speed and hardware efficiency.
Abstract
In graph embedding, the connectivity information of a graph is used to represent each vertex as a point in a d-dimensional space. Unlike the original, irregular structural information, such a representation can be used for a multitude of machine learning tasks. Although the process is extremely useful in practice, it is indeed expensive and unfortunately, the graphs are becoming larger and harder to embed. Attempts at scaling up the process to larger graphs have been successful but often at a steep price in hardware requirements. We present GOSH, an approach for embedding graphs of arbitrary sizes on a single GPU with minimum constraints. GOSH utilizes a novel graph coarsening approach to compress the graph and minimize the work required for embedding, delivering high-quality embeddings at a fraction of the time compared to the state-of-the-art. In addition to this, it incorporates a…
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Code & Models
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
