Sequence-to-sequence modeling for graph representation learning

被引:5
作者
Taheri, Aynaz [1 ]
Gimpel, Kevin [2 ]
Berger-Wolf, Tanya [1 ]
机构
[1] Univ Illinois, Chicago, IL 60607 USA
[2] Toyota Technol Inst Chicago, Chicago, IL USA
关键词
Graph representation learning; Deep learning; Graph classification; Recurrent models;
D O I
10.1007/s41109-019-0174-8
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
We propose sequence-to-sequence architectures for graph representation learning in both supervised and unsupervised regimes. Our methods use recurrent neural networks to encode and decode information from graph-structured data. Recurrent neural networks require sequences, so we choose several methods of traversing graphs using different types of substructures with various levels of granularity to generate sequences of nodes for encoding. Our unsupervised approaches leverage long short-term memory (LSTM) encoder-decoder models to embed the graph sequences into a continuous vector space. We then represent a graph by aggregating its graph sequence representations. Our supervised architecture uses an attention mechanism to collect information from the neighborhood of a sequence. The attention module enriches our model in order to focus on the subgraphs that are crucial for the purpose of a graph classification task. We demonstrate the effectiveness of our approaches by showing improvements over the existing state-of-the-art approaches on several graph classification tasks.
引用
收藏
页数:26
相关论文
共 59 条
[21]  
Garcia-Duran Alberto, 2017, ARXIV PREPRINT ARXIV
[22]   On graph kernels:: Hardness results and efficient alternatives [J].
Gärtner, T ;
Flach, P ;
Wrobel, S .
LEARNING THEORY AND KERNEL MACHINES, 2003, 2777 :129-143
[23]  
Gilmer J, 2017, PR MACH LEARN RES, V70
[24]   Automatic Chemical Design Using a Data-Driven Continuous Representation of Molecules [J].
Gomez-Bombarelli, Rafael ;
Wei, Jennifer N. ;
Duvenaud, David ;
Hernandez-Lobato, Jose Miguel ;
Sanchez-Lengeling, Benjamin ;
Sheberla, Dennis ;
Aguilera-Iparraguirre, Jorge ;
Hirzel, Timothy D. ;
Adams, Ryan P. ;
Aspuru-Guzik, Alan .
ACS CENTRAL SCIENCE, 2018, 4 (02) :268-276
[25]  
Graves A, 2012, STUD COMPUT INTELL, V385, P1, DOI [10.1162/neco.1997.9.1.1, 10.1007/978-3-642-24797-2]
[26]   node2vec: Scalable Feature Learning for Networks [J].
Grover, Aditya ;
Leskovec, Jure .
KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, :855-864
[27]  
Haussler D., 1999, Univ.California, Tech. Rep. UCSC-CRL-99-10
[28]  
Kriege NM, 2016, ADV NEUR IN, V29
[29]   Towards Unbiased BFS Sampling [J].
Kurant, Maciej ;
Markopoulou, Athina ;
Thiran, Patrick .
IEEE JOURNAL ON SELECTED AREAS IN COMMUNICATIONS, 2011, 29 (09) :1799-1809
[30]   Graph Classification using Structural Attention [J].
Lee, John Boaz ;
Rossi, Ryan ;
Kong, Xiangnan .
KDD'18: PROCEEDINGS OF THE 24TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2018, :1666-1674