Saved in:
| Main Authors: | , , , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2021
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2101.01425 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866909369546833920 |
|---|---|
| author | Soto-Gomez, Mauricio Robinson, Peter Cano, Carlos Pashaeibarough, Ali Cavalleri, Emanuele Reese, Justin Mesiti, Marco Valentini, Giorgio Casiraghi, Elena |
| author_facet | Soto-Gomez, Mauricio Robinson, Peter Cano, Carlos Pashaeibarough, Ali Cavalleri, Emanuele Reese, Justin Mesiti, Marco Valentini, Giorgio Casiraghi, Elena |
| contents | Many real-world problems are naturally modeled as heterogeneous graphs, where nodes and edges represent multiple types of entities and relations. Existing learning models for heterogeneous graph representation usually depend on the computation of specific and user-defined heterogeneous paths, or in the application of large and often not scalable deep neural network architectures. We propose Het-node2vec, an extension of the node2vec algorithm, designed for embedding heterogeneous graphs. Het-node2vec addresses the challenge of capturing the topological and structural characteristics of graphs and the semantic information underlying the different types of nodes and edges of heterogeneous graphs, by introducing a simple stochastic node and edge type switching strategy in second order random walk processes. The proposed approach also introduces an ''attention mechanism'' to focus the random walks on specific node and edge types, thus allowing more accurate embeddings and more focused predictions on specific node and edge types of interest. Empirical results on benchmark datasets show that Hetnode2vec achieves comparable or superior performance with respect to state-of-the-art methods for heterogeneous graphs in node label and edge prediction tasks. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2101_01425 |
| institution | arXiv |
| publishDate | 2021 |
| record_format | arxiv |
| spellingShingle | Het-node2vec: second order random walk sampling for heterogeneous multigraphs embedding Soto-Gomez, Mauricio Robinson, Peter Cano, Carlos Pashaeibarough, Ali Cavalleri, Emanuele Reese, Justin Mesiti, Marco Valentini, Giorgio Casiraghi, Elena Machine Learning Social and Information Networks Physics and Society I.2; I.5 Many real-world problems are naturally modeled as heterogeneous graphs, where nodes and edges represent multiple types of entities and relations. Existing learning models for heterogeneous graph representation usually depend on the computation of specific and user-defined heterogeneous paths, or in the application of large and often not scalable deep neural network architectures. We propose Het-node2vec, an extension of the node2vec algorithm, designed for embedding heterogeneous graphs. Het-node2vec addresses the challenge of capturing the topological and structural characteristics of graphs and the semantic information underlying the different types of nodes and edges of heterogeneous graphs, by introducing a simple stochastic node and edge type switching strategy in second order random walk processes. The proposed approach also introduces an ''attention mechanism'' to focus the random walks on specific node and edge types, thus allowing more accurate embeddings and more focused predictions on specific node and edge types of interest. Empirical results on benchmark datasets show that Hetnode2vec achieves comparable or superior performance with respect to state-of-the-art methods for heterogeneous graphs in node label and edge prediction tasks. |
| title | Het-node2vec: second order random walk sampling for heterogeneous multigraphs embedding |
| topic | Machine Learning Social and Information Networks Physics and Society I.2; I.5 |
| url | https://arxiv.org/abs/2101.01425 |