TuX2: Distributed Graph Computation for Machine Learning
- Jilong Xue ,
- Youshan Miao ,
- Cheng Chen ,
- Ming Wu ,
- Lidong Zhou
14th USENIX Symposium on Networked Systems Design and Implementation (NSDI 17) |
Published by USENIX Association
TuX2: Distributed Graph Computation for Machine Learning
TuX2 is a new distributed graph engine that bridges graph computation and distributed machine learning. TuX2 inherits the benefits of an elegant graph computation model, efficient graph layout, and balanced parallelism to scale to billion-edge graphs; we extend and optimize it for distributed machine learning to support heterogeneity, a Stale Synchronous Parallel model, and a new MEGA (Mini-batch, Exchange, GlobalSync, and
Apply) model.
We have developed a set of representative distributed machine learning algorithms in TuX2, covering both supervised and unsupervised learning. Compared to implementations on distributed machine learning platforms, writing these algorithms in TuX2 takes only about 25% of the code: Our graph computation model hides the detailed management of data layout, partitioning, and parallelism from developers. Our extensive evaluation of TuX2, using large data sets with up to 64 billion edges, shows that TuX2 outperforms state-of-the-art distributed graph engines PowerGraph and PowerLyra by an order of
magnitude, while beating two state-of-the-art distributed machine learning systems by at least 48%.