TY - JOUR
T1 - Multi-task network embedding
AU - Xu, Linchuan
AU - Wei, Xiaokai
AU - Cao, Jiannong
AU - Yu, Philip S.
N1 - Funding Information:
The work described in this paper was partially supported by National Key R & D Program of China—2018 YFB1004801, RGC General Research Fund under Grant PolyU 152199/17E, the funding for Project of Strategic Importance provided by The Hong Kong Polytechnic University (Project Code: 1-ZE26), NSF through Grants IIS-1526499, IIS-1763325, CNS-1626432, and NSFC 61672313.
Publisher Copyright:
© 2018, Springer Nature Switzerland AG.
Copyright:
Copyright 2020 Elsevier B.V., All rights reserved.
PY - 2019/9/1
Y1 - 2019/9/1
N2 - As there are various data mining applications involving network analysis, network embedding is frequently employed to learn latent representations or embeddings that encode the network structure. However, existing network embedding models are only designed for a single network scenario. It is common that nodes can have multiple types of relationships in big data era, which results in multiple networks, e.g., multiple social networks and multiple gene regulatory networks. Jointly embedding multiple networks thus may make network-specific embeddings more comprehensive and complete as the same node may expose similar or complementary characteristics in different networks. In this paper, we thus propose an idea of multi-task network embedding to jointly learn multiple network-specific embeddings for each node via enforcing an extra information-sharing embedding. We instantiate the idea in two types of models that are different in the mechanism for enforcing the information-sharing embedding. The first type enforces the information-sharing embedding as a common embedding shared by all tasks, which is similar to the concept of the common metric in multi-task metric learning, while the second type enforces the information-sharing embedding as a consensus embedding on which all network-specific embeddings agree. Moreover, we propose two mechanisms for embedding the network structure, which are first-order proximity preserving and second-order proximity preserving. We demonstrate through comprehensive experiments on three real-world datasets that the proposed models outperform recent network embedding models in applications including visualization, link prediction, and multi-label classification.
AB - As there are various data mining applications involving network analysis, network embedding is frequently employed to learn latent representations or embeddings that encode the network structure. However, existing network embedding models are only designed for a single network scenario. It is common that nodes can have multiple types of relationships in big data era, which results in multiple networks, e.g., multiple social networks and multiple gene regulatory networks. Jointly embedding multiple networks thus may make network-specific embeddings more comprehensive and complete as the same node may expose similar or complementary characteristics in different networks. In this paper, we thus propose an idea of multi-task network embedding to jointly learn multiple network-specific embeddings for each node via enforcing an extra information-sharing embedding. We instantiate the idea in two types of models that are different in the mechanism for enforcing the information-sharing embedding. The first type enforces the information-sharing embedding as a common embedding shared by all tasks, which is similar to the concept of the common metric in multi-task metric learning, while the second type enforces the information-sharing embedding as a consensus embedding on which all network-specific embeddings agree. Moreover, we propose two mechanisms for embedding the network structure, which are first-order proximity preserving and second-order proximity preserving. We demonstrate through comprehensive experiments on three real-world datasets that the proposed models outperform recent network embedding models in applications including visualization, link prediction, and multi-label classification.
KW - Data mining
KW - Multi-task learning
KW - Network embedding
KW - Representation learning
UR - http://www.scopus.com/inward/record.url?scp=85086581395&partnerID=8YFLogxK
U2 - 10.1007/s41060-018-0166-2
DO - 10.1007/s41060-018-0166-2
M3 - Journal article
AN - SCOPUS:85086581395
SN - 2364-415X
VL - 8
SP - 183
EP - 198
JO - International Journal of Data Science and Analytics
JF - International Journal of Data Science and Analytics
IS - 2
ER -