LightTraffic: On Optimizing CPU-GPU Data Traffic for Efficient Large-scale Random Walks
Yipeng Xing, Yongkun Li, Zhiqiang Wang, Yinlong Xu, John C. S. Lui
Abstract
As a fundamental tool for graph analysis, random walk receives extensive attention in both industry and academia. For computing massive random walks, recent works show that GPUs provide a good option to accelerate the performance. However, due to the limited memory space of modern GPUs, it is infeasible to have both the graph data and walk index fully reside in GPU memory when running large-scale random walks. Thus, it necessitates an out-of-GPU-memory design, but this inevitably induces large amounts of CPU-GPU data transmission traffic and thus hinders the overall performance. In this paper, we develop LightTraffic, which optimizes the data transmission between CPU and GPU memory under the constraint of GPU memory capacity with various system designs, including a memory-efficient scheme for partition-based management and multiple scheduling techniques. LightTraffic is a fully out-of-GPU-memory design, so it supports running large-scale random walks on GPUs. Experiments on our prototype show that LightTraffic outperforms various state-of-the-art CPU-based in-memory systems which also support large-scale random walks. For example, compared to the CPU-based systems FlashMob and ThunderRW, which are highly optimized for random walks, LightTraffic achieves 1.7−5.0× and 1.4 − 12.8× performance speedup, respectively. It also achieves up to an order of magnitude speedup when compared to the GPU-based system Subway which also supports large-scale random walks with an out-of-GPU-memory design for graph data.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on20
- GraphSAINT: Graph Sampling Based Inductive Learning MethodHanqing Zeng, Hongkuan Zhou, Ajitesh Srivastava, Rajgopal Kannan et al.ICLR 2020 · 1,155 citations
- ZeRO: memory optimizations toward training trillion parameter modelsSamyam Rajbhandari, Jeff Rasley, Olatunji Ruwase, Yuxiong HeSC 2020 · 852 citations
- GNNLab: a factored system for sample-based GNN training over GPUsJianbang Yang, Dahai Tang, Xiaoniu Song, Lei Wang et al.EuroSys 2022 · 105 citations
- Pump Up the Volume: Processing Large Data on GPUs with Fast InterconnectsClemens Lutz, Sebastian Breß, Steffen Zeuch, Tilmann Rabl et al.SIGMOD 2020 · 99 citations
- Large Graph Convolutional Network Training with GPU-Oriented Data Communication ArchitectureSeungwon Min, Kun Wu, Sitao Huang, Mert Hidayetoglu et al.VLDB 2021 · 85 citations
Related papers
- Random Walks on Huge Graphs at Cache EfficiencyKe Yang, Xiaosong Ma, Saravanan Thirumuruganathan, Kang Chen et al.SOSP 2021 · 26 citations
- ThunderRW: An In-Memory Graph Random Walk EngineShixuan Sun, Yuhang Chen, Shengliang Lu, Bingsheng He et al.VLDB 2021 · 31 citations
- LightRW: FPGA Accelerated Graph Dynamic Random WalksHongshi Tan, Xinyu Chen, Yao Chen, Bingsheng He et al.SIGMOD 2023 · 9 citations
- Subway: minimizing data transfer during out-of-GPU-memory graph processingAmir Hossein Nodehi Sabet, Zhijia Zhao, Rajiv GuptaEuroSys 2020 · 84 citations
- FlowWalker: A Memory-efficient and High-performance GPU-based Dynamic Graph Random Walk FrameworkJunyi Mei, Shixuan Sun, Chao Li, Cheng Xu et al.VLDB 2024 · 10 citations
