SDT-GNN: Streaming-based Distributed Training Framework for Graph Neural Networks
arxiv.org·2h
🌊CBOR Streaming
Preview
Report Post

arXiv:2404.02300v2 Announce Type: replace Abstract: Recently, distributed GNN training frameworks, such as DistDGL and PyG, have been developed to enable training GNN models on large graphs by leveraging multiple GPUs in a distributed manner. Despite these advances, their memory requirements are still excessively high, thereby hindering GNN training on large graphs using commodity workstations. In this paper, we propose SDT-GNN, a streaming-based distributed GNN training framework. Unlike the existing frameworks that load the entire graph in memory, it takes a stream of edges as input for graph partitioning to reduce the memory requirement for partitioning. It also enables distributed GNN training even when the aggregated memory size of GPUs is smaller than the size of the graph and featur…

Similar Posts

Loading similar posts...