Implementation of cube2net, ICDM 2019 PhD Forum.
Please cite the following work if you find the code useful.
@inproceedings{yang2019cube2net,
Author = {Yang, Carl and Liu, Mengxiong and He, Frank and Peng, Jian and Han, Jiawei},
Booktitle = {ICDM PhD Forum},
Title = {cube2net: efficient quality network construction with data cube organization},
Year = {2019}
}
Contact: Carl Yang (yangji9181@gmail.com)
step 1: cell network construction (mengxiong) based on current cell construction code, construct three networks, assume single hiararchy for each of them for now.
- venue network based on venue names (in case venues are too many, do clustering first, it will also be useful when we later consider hiararchical cells)
- year network
- content network based on topic models
step 2: reinforcement learning state and value function design (mengxiong) follow nips 2017 graph dqn work ([1])
- implement the embedding of each dimension based on Eq 3 in [1], combine three dimensions and implement the value function based on Eq 4 in [1].
step 3: reinforcement learning algorithm (shibi)
- design actions: add new cells: choose a dimension and a hiararchy to extend
- trade-off between exploit and exploration
step 4; reward and evaluation (mengxiong) use author clustering as the task for now, based on dblp labeled authors on the server
- construct author network N based on chosen cells
- evaluate chosen cells by evaluating a basic graph clustering algorithm like mincut on N