Abstract
Graphs represent real world relationships, and graph embedding projects nodes in a graph to a latent space that can help simplify downstream tasks. Recent development of graph convolutions in deep learning significantly improves the performance of many learning tasks on graphs. Unfortunately, prior embedding methods either do not embed graphs with node features, or fail to produce high-quality embeddings for downstream learning tasks that result in large performance gap in comparison to direct learning on graphs.We present a versatile and effective embedding method, Conv2Vec, for embedding graphs with or without node features. It is based on graph convolutions with objective functions motivated by concepts and structures from classical graph algorithms. Conv2Vec produce high-quality embedding for both plain graphs and graphs with node features for downstream tasks.We evaluate the embeddings generated by Conv2Vec with a transductive node classification task. With the generated embeddings and very simple machine learning approaches, we are able to achieve accuracies similar to those achieved by direct learning with graph convolutions. Interestingly, if we strip the node features from the graph and thus learning an embedding has to rely entirely on the graph topology, node classification with our embedding significantly outperforms direct learning with various graph convolutions. This suggests that structures from classical graph algorithms may play an important role in learning on graphs.
Original language | English |
---|---|
Title of host publication | Proceedings - 21st IEEE International Conference on Data Mining Workshops, ICDMW 2021 |
Editors | Bing Xue, Mykola Pechenizkiy, Yun Sing Koh |
Publisher | IEEE Computer Society |
Pages | 669-677 |
Number of pages | 9 |
ISBN (Electronic) | 9781665424271 |
DOIs | |
State | Published - 2021 |
Event | 21st IEEE International Conference on Data Mining Workshops, ICDMW 2021 - Virtual, Online, New Zealand Duration: Dec 7 2021 → Dec 10 2021 |
Publication series
Name | IEEE International Conference on Data Mining Workshops, ICDMW |
---|---|
Volume | 2021-December |
ISSN (Print) | 2375-9232 |
ISSN (Electronic) | 2375-9259 |
Conference
Conference | 21st IEEE International Conference on Data Mining Workshops, ICDMW 2021 |
---|---|
Country/Territory | New Zealand |
City | Virtual, Online |
Period | 12/7/21 → 12/10/21 |
Funding
This research used resources of the Compute and Data Environment for Science (CADES) at the Oak Ridge National Laboratory, which is supported by the Office of Science of the U.S. Department of Energy under Contract No. DE-AC05-00OR22725. This material is based upon work supported in part by the U.S. Department of Energy, Office of Science, Office of Advanced Scientific Computing Research, under contract number DE-AC05-00OR22725, and in part by the Laboratory Directed Research and Development Program of Oak Ridge National Laboratory, managed by UT-Battelle, LLC.
Keywords
- Deep learning
- GNN
- Graph Embedding
- Unsupervised learning