Versatile feature learning with graph convolutions and graph structures

Guojing Cong, Seung Hwan Lim

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

3 Scopus citations

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 languageEnglish
Title of host publicationProceedings - 21st IEEE International Conference on Data Mining Workshops, ICDMW 2021
EditorsBing Xue, Mykola Pechenizkiy, Yun Sing Koh
PublisherIEEE Computer Society
Pages669-677
Number of pages9
ISBN (Electronic)9781665424271
DOIs
StatePublished - 2021
Event21st IEEE International Conference on Data Mining Workshops, ICDMW 2021 - Virtual, Online, New Zealand
Duration: Dec 7 2021Dec 10 2021

Publication series

NameIEEE International Conference on Data Mining Workshops, ICDMW
Volume2021-December
ISSN (Print)2375-9232
ISSN (Electronic)2375-9259

Conference

Conference21st IEEE International Conference on Data Mining Workshops, ICDMW 2021
Country/TerritoryNew Zealand
CityVirtual, Online
Period12/7/2112/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

Fingerprint

Dive into the research topics of 'Versatile feature learning with graph convolutions and graph structures'. Together they form a unique fingerprint.

Cite this