Text-Associated Max-Margin DeepWalk.

Zhonglin Ye,Haixing Zhao,Ke Zhang,Yu Zhu,Yuzhi Xiao
DOI: https://doi.org/10.1007/978-981-13-2922-7_21
IF: 4.426
2018-01-01
Big Data
Abstract:Most existing network representation algorithms learn the network representations based on network structure, however, they neglect the rich external information associated with nodes (i.e. text contents, communities and label information). Meanwhile, the learnt representations usually lack the discriminative ability for the tasks of node classification and linking prediction. We consequently overcame the above challenges by presenting a novel semi-supervised algorithm, text-associated max-margin DeepWalk algorithm (TMDW). TMDW incorporates text contents and network structures into the network representation learning based on the inductive matrix completion algorithm, and then we use node’s category to optimize the learnt network representations based on the mar-margin principle and biased gradient. For integrating the above tasks, we propose a novel and efficient framework of network representation learning, this framework is easy to extend and generate discriminative representations. We then evaluate our model using the multi-class classification tasks. The experimental results demonstrate that TMDW outperforms other baseline methods on three real-world datasets. The visualization task of TMDW shows that our model is more discriminative than the other unsupervised approaches.
What problem does this paper attempt to address?