Learning Graph Representations with Embedding Propagation
About
We propose Embedding Propagation (EP), an unsupervised learning framework for graph-structured data. EP learns vector representations of graphs by passing two types of messages between neighboring nodes. Forward messages consist of label representations such as representations of words and other attributes associated with the nodes. Backward messages consist of gradients that result from aggregating the label representations and applying a reconstruction loss. Node representations are finally computed from the representation of their labels. With significantly fewer parameters and hyperparameters an instance of EP is competitive with and often outperforms state of the art unsupervised and semi-supervised learning methods on a range of benchmark data sets.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Transductive Node Classification | Pubmed (transductive) | Accuracy79.56 | 95 | |
| Node Classification | Cora transductive (test) | Accuracy81.72 | 36 | |
| Node Classification | PPI | Micro F125.28 | 29 | |
| Node Classification | Citeseer transductive (test) | Accuracy71.01 | 28 | |
| Node Classification | CORA inductive setting (test) | Accuracy73.09 | 22 | |
| Node Classification | CITESEER inductive setting (test) | Accuracy68.61 | 21 | |
| Multi-Label Classification | POS | Micro-F152.28 | 15 | |
| Multi-Label Classification | BlogCatalog | Micro-F140.41 | 15 | |
| Node Classification | PUBMED inductive setting (test) | Accuracy79.94 | 14 |