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 条
[41]  
Riesen K, 2010, ADV DATABASE SYST, V40, P217, DOI 10.1007/978-1-4419-6045-0_7
[42]  
Rossi RA., 2018, IEEE T KNOWL DATA EN
[43]   The Graph Neural Network Model [J].
Scarselli, Franco ;
Gori, Marco ;
Tsoi, Ah Chung ;
Hagenbuchner, Markus ;
Monfardini, Gabriele .
IEEE TRANSACTIONS ON NEURAL NETWORKS, 2009, 20 (01) :61-80
[44]  
Shervashidze N., 2009, P 12 INT C ART INT S, P1
[45]  
Shervashidze N, 2011, J MACH LEARN RES, V12, P2539
[46]   LINE: Large-scale Information Network Embedding [J].
Tang, Jian ;
Qu, Meng ;
Wang, Mingzhe ;
Zhang, Ming ;
Yan, Jun ;
Mei, Qiaozhu .
PROCEEDINGS OF THE 24TH INTERNATIONAL CONFERENCE ON WORLD WIDE WEB (WWW 2015), 2015, :1067-1077
[47]   Statistical evaluation of the Predictive Toxicology Challenge 2000-2001 [J].
Toivonen, H ;
Srinivasan, A ;
King, RD ;
Kramer, S ;
Helma, C .
BIOINFORMATICS, 2003, 19 (10) :1183-1193
[48]  
Trivedi R, 2017, PR MACH LEARN RES, V70
[49]  
Tsitsulin A, 2018, ARXIV PREPRINT ARXIV
[50]  
van der Maaten L, 2008, J MACH LEARN RES, V9, P2579