ActiveHNE: Active Heterogeneous Network Embedding
ActiveHNE: Active Heterogeneous Network Embedding
Xia Chen, Guoxian Yu, Jun Wang, Carlotta Domeniconi, Zhao Li, Xiangliang Zhang
Proceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence
Main track. Pages 2123-2129.
https://doi.org/10.24963/ijcai.2019/294
Heterogeneous network embedding (HNE) is a challenging task due to the diverse node types and/or diverse relationships between nodes. Existing HNE methods are typically unsupervised. To maximize the profit of utilizing the rare and valuable supervised information in HNEs, we develop a novel Active Heterogeneous Network Embedding (ActiveHNE) framework, which includes two components: Discriminative Heterogeneous Network Embedding (DHNE) and Active Query in Heterogeneous Networks (AQHN).In DHNE, we introduce a novel semi-supervised heterogeneous network embedding method based on graph convolutional neural network. In AQHN, we first introduce three active selection strategies based on uncertainty and representativeness, and then derive a batch selection method that assembles these strategies using a multi-armed bandit mechanism. ActiveHNE aims at improving the performance of HNE by feeding the most valuable supervision obtained by AQHN into DHNE. Experiments on public datasets demonstrate the effectiveness of ActiveHNE and its advantage on reducing the query cost.
Keywords:
Machine Learning: Active Learning
Machine Learning Applications: Networks
Machine Learning: Semi-Supervised Learning
Machine Learning: Deep Learning