Transferable and Differentiable Discrete Network Embedding for multi-domains with Hierarchical Knowledge Distillation

Tao He,Lianli Gao,Jingkuan Song,Yuan-Fang Li
DOI: https://doi.org/10.1016/j.ins.2023.01.146
IF: 8.1
2023-02-11
Information Sciences
Abstract:Discrete network embedding aims to learn compact and low-dimensional discrete representations for network (graph) nodes by preserving semantical and structural information of networks. In practice, there are numerous domains for network data, but expensive semantical labels may only be available for some of them. This demands those embedding algorithms to be equipped with transferability, i.e., unsupervisedly adapting to other domains. However, the prevailing conventional discrete network embedding methods force on a single domain. This work bridges this gap by developing a transferable discrete network embedding method. We focus on solving two problems: knowledge transfer from a labeled domain to a new unlabeled domain and non-differentiable discretization in the discrete embedding. Specifically, for the former, we propose a hierarchical knowledge distillation strategy to mitigate the knowledge gap between the domains. At the same time, for the latter, we novelly treat the discretization as a classification problem instead of the conventional regression problem. Finally, we conduct a wide range of experiments to evaluate our method on a suite of tasks, including link prediction, node classification, and neighbor recommendation. Our evaluation results demonstrate that our model performs better than the conventional discrete embedding methods over all the tasks.
computer science, information systems
What problem does this paper attempt to address?