A Tensor-based Markov Chain Model for Heterogeneous Information Network Collective Classification : Extended abstract

Chao Han, Jian Chen, Mingkui Tan, Michael K. Ng, Qingyao Wu

Abstract: Heterogeneous Information Network(HIN) collective classification aims to classify one type of node, which is associated with multiple types of nodes through multiple types of relations. Previous studies have revealed that exploiting the relative importance of relation types is quite useful for improving node classification performance. We propose a Tensor-based Markov chain (T-Mark) model to improve the nodes classification accuracy by predicting the labels for unlabeled nodes and the importance ranking of relationship types automatically and simultaneously. Specifically, we build two tensor equations according to the HIN structure and content similarities among nodes of both labeled and unlabeled data. Consequently, We solve the semi-supervised T-Mark model by using an iterative process until obtaining two stationary distributions for labels and relation types. Experimental results on several real-world datasets demonstrate the effectiveness of T-Mark.