Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/maxuewei2/RiWalk
Code for the ICDM 2019 Paper "RiWalk: Fast Structural Node Embedding via Role Identification".
https://github.com/maxuewei2/RiWalk
graph graph-algorithms graph-embedding graph-representation graph-representation-learning icdm network-embedding network-representation-learning representation-learning struc2vec structural-embedding
Last synced: 3 months ago
JSON representation
Code for the ICDM 2019 Paper "RiWalk: Fast Structural Node Embedding via Role Identification".
- Host: GitHub
- URL: https://github.com/maxuewei2/RiWalk
- Owner: maxuewei2
- License: mit
- Created: 2019-08-15T13:53:42.000Z (about 5 years ago)
- Default Branch: master
- Last Pushed: 2020-04-27T10:07:34.000Z (over 4 years ago)
- Last Synced: 2024-06-17T18:01:22.714Z (5 months ago)
- Topics: graph, graph-algorithms, graph-embedding, graph-representation, graph-representation-learning, icdm, network-embedding, network-representation-learning, representation-learning, struc2vec, structural-embedding
- Language: Python
- Homepage:
- Size: 2.63 MB
- Stars: 13
- Watchers: 4
- Forks: 4
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# RiWalk
This repository provides reference implementations of **RiWalk** as described in the paper:
> RiWalk: Fast Structural Node Embedding via Role Identification.
> Xuewei Ma, Geng Qin, Zhiyang Qiu, Mingxin Zheng, Zhe Wang.
> IEEE International Conference on Data Mining, ICDM, 2019.
The RiWalk algorithm learns continuous representations for nodes in graphs. The learned representations capture structural similarities between nodes.
---
RiWalk decouples the structural embedding problem into a role identification procedure and a network embedding procedure.
The key idea of RiWalk can be illustrated as in the following picture.
Two nodes a and u residing far apart in a network have similar local topologies but totally different context nodes. However, after the role identification procedure, they have similar context and are indirectly densely connected, thus typical network embedding methods can be directly applied to learn structural embeddings.
### Usage
This repository provides several different implementations of RiWalk:
- src/RiWalk: a python implementation of RiWalk-SP and RiWalk-WL as described in the paper.
- src/RiWalk-RW: a python implementation of RiWalk-RW-SP and RiWalk-RW-WL, two variations of RiWalk-SP and RiWalk-WL. RiWalk-RW traverses a subgraph induced by random walk sequences instead of breadth-first searching the local subgraph.
- src/RiWalk-C: a C implementation of RiWalk-SP and RiWalk-RW-SP.
#### Full Command List
The full list of command line options is available with
``` bash
# RiWalk
python3 src/RiWalk/RiWalk.py --help
# RiWalk-RW
python3 src/RiWalk-RW/RiWalk-RW.py --help
# RiWalk-C
gcc -lm -pthread -Ofast -march=native -Wall -ffast-math -Wno-unused-result src/RiWalk-C/RiWalk.c -o src/RiWalk-C/RiWalk
src/RiWalk-C/RiWalk
```
#### Example
We provide an example running script for the actor data set in [train_actor.sh](train_actor.sh).
#### Input
The supported input format is an edgelist:
```text
node1_id_int node2_id_int
```
#### Output
The output file has *n+1* lines for a graph with *n* vertices.
The first line has the following format:
```text
num_of_nodes dim_of_representation
```
The next *n* lines are as follows:
```text
node_id dim1 dim2 ... dimd
```
where dim1, ... , dimd is the *d*-dimensional representation learned by RiWalk.
### Running Time Comparison
### Acknowledgements
We would like to thank the authors of [node2vec](https://github.com/aditya-grover/node2vec), [struc2vec](https://github.com/leoribeiro/struc2vec), [GraphWave](https://github.com/snap-stanford/graphwave) and [LINE](https://github.com/tangjianpku/LINE) for the open access of the implementations of their methods.
### Miscellaneous
- Please send any questions you might have about the code and/or the algorithm to .
### Citation
[https://dblp.org/rec/bibtex/conf/icdm/MaQQZW19](https://dblp.org/rec/bibtex/conf/icdm/MaQQZW19)