Abstract:Graph convolutional network (GCN) has been successfully applied to many graph-based applications; however, training a large-scale GCN remains challenging. Current SGD-based algorithms suffer from either a high computational cost that exponentially grows with number of GCN layers, or a large space requirement for keeping the entire graph and the embedding of each node in memory. In this paper, we propose Cluster-GCN, a novel GCN algorithm that is suitable for SGD-based training by exploiting the graph clustering structure. Cluster-GCN works as the following: at each step, it samples a block of nodes that associate with a dense subgraph identified by a graph clustering algorithm, and restricts the neighborhood search within this subgraph. This simple but effective strategy leads to significantly improved memory and computational efficiency while being able to achieve comparable test accuracy with previous algorithms. To test the scalability of our algorithm, we create a new Amazon2M data with 2 million nodes and 61 million edges which is more than 5 times larger than the previous largest publicly available dataset (Reddit). For training a 3-layer GCN on this data, Cluster-GCN is faster than the previous state-of-the-art VR-GCN (1523 seconds vs 1961 seconds) and using much less memory (2.2GB vs 11.2GB). Furthermore, for training 4 layer GCN on this data, our algorithm can finish in around 36 minutes while all the existing GCN training algorithms fail to train due to the out-of-memory issue. Furthermore, Cluster-GCN allows us to train much deeper GCN without much time and memory overhead, which leads to improved prediction accuracy---using a 5-layer Cluster-GCN, we achieve state-of-the-art test F1 score 99.36 on the PPI dataset, while the previous best result was 98.71 by [16]. Our codes are publicly available at https://github.com/google-research/google-research/tree/master/cluster_gcn.

Dual hybrid frameworks combining graph convolutional network with decoding for covering location problem

Joint CRF and Locality-Consistent Dictionary Learning for Semantic Segmentation.

Mixed Geometry Message and Trainable Convolutional Attention Network for Knowledge Graph Completion

GoodFloorplan: Graph Convolutional Network and Reinforcement Learning-Based Floorplanning

ReCovNet: Reinforcement learning with covering information for solving maximal coverage billboards location problem

Optimizing Spatial Relationships in GCN to Improve the Classification Accuracy of Remote Sensing Images

ConvGCN-RF: A Hybrid Learning Model for Commuting Flow Prediction Considering Geographical Semantics and Neighborhood Effects

Research on Knowledge Graph Completion Model Combining Temporal Convolutional Network and Monte Carlo Tree Search

Coupling Dual Graph Convolution Network and Residual Network for Local Climate Zone Mapping

Joint Approach for Vehicle Routing Problems Based on Genetic Algorithm and Graph Convolutional Network

Learning Spatial Interaction Representation with Heterogeneous Graph Convolutional Networks for Urban Land-Use Inference

Global heterogeneous graph convolutional network: from coarse to refined land cover and land use segmentation

Swap-based Deep Reinforcement Learning for Facility Location Problems in Networks

Overlapping Community Detection using Dynamic Dilated Aggregation in Deep Residual GCN

Dual Graph Convolutional Network for Semantic Segmentation.

HD-GCN:A Hybrid Diffusion Graph Convolutional Network

Self-Paced Learning Based Graph Convolutional Neural Network for Mixed Integer Programming (student Abstract).

Sensing Overlapping Geospatial Communities from Human Movements Using Graph Affiliation Generation Models.

Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks

Learning Cellular Coverage from Real Network Configurations using GNNs

Multi-Context enhanced Lane-Changing prediction using a heterogeneous Graph Neural Network