Home

Awesome

Unsupervised Attributed Multiplex Network Embedding (DMGI)

<p align="center"> <a href="https://aaai.org/Conferences/AAAI-20/" alt="Conference"> <img src="https://img.shields.io/badge/AAAI'20-brightgreen" /></a> <a href="https://pytorch.org/" alt="PyTorch"> <img src="https://img.shields.io/badge/PyTorch-%23EE4C2C.svg?e&logo=PyTorch&logoColor=white" /></a> </p>

Overview

Nodes in a multiplex network are connected by multiple types of relations. However, most existing network embedding methods assume that only a single type of relation exists between nodes. Even for those that consider the multiplexity of a network, they overlook node attributes, resort to node labels for training, and fail to model the global properties of a graph. We present a simple yet effective unsupervised network embedding method for attributed multiplex network called DMGI, inspired by Deep Graph Infomax (DGI) that maximizes the mutual information between local patches of a graph, and the global representation of the entire graph. We devise a systematic way to jointly integrate the node embeddings from multiple graphs by introducing 1) the consensus regularization framework that minimizes the disagreements among the relation-type specific node embeddings, and 2) the universal discriminator that discriminates true samples regardless of the relation types. We also show that the attention mechanism infers the importance of each relation type, and thus can be useful for filtering unnecessary relation types as a preprocessing step. Extensive experiments on various downstream tasks demonstrate that DMGI outperforms the state-of-the-art methods, even though DMGI is fully unsupervised.

DMGI is introduced in the official repository of PyTorch Geometric (Link)

Paper

Requirements

How to Run

git clone https://github.com/pcy1302/DMGI.git
cd DMGI
mkdir saved_model
python main.py --embedder DMGI --dataset imdb --metapaths MAM,MDM --isAttn

Data format [(ex) IMDB]

Cite (Bibtex)

@article{park2019unsupervised,
  title={Unsupervised Attributed Multiplex Network Embedding},
  author={Park, Chanyoung and Kim, Donghyun and Han, Jiawei and Yu, Hwanjo},
  booktitle={AAAI},
  year={2020}
}