Node and Edge Joint Embedding for Heterogeneous Information Network
Due to the heterogeneity of nodes and edges, heterogeneous network embedding is a very challenging task to embed highly coupled networks into a set of low-dimensional vectors. Existing models either only learn embedding vectors for nodes or only for edges. These two methods of embedding learning are...
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Tsinghua University Press
2024-09-01
|
| Series: | Big Data Mining and Analytics |
| Subjects: | |
| Online Access: | https://www.sciopen.com/article/10.26599/BDMA.2023.9020037 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1832568877923434496 |
|---|---|
| author | Lei Chen Yuan Li Hualiang Liu Haomiao Guo |
| author_facet | Lei Chen Yuan Li Hualiang Liu Haomiao Guo |
| author_sort | Lei Chen |
| collection | DOAJ |
| description | Due to the heterogeneity of nodes and edges, heterogeneous network embedding is a very challenging task to embed highly coupled networks into a set of low-dimensional vectors. Existing models either only learn embedding vectors for nodes or only for edges. These two methods of embedding learning are rarely performed in the same model, and they both overlook the internal correlation between nodes and edges. To solve these problems, a node and edge joint embedding model is proposed for Heterogeneous Information Networks (HINs), called NEJE. The NEJE model can better capture the latent structural and semantic information from an HIN through two joint learning strategies: type-level joint learning and element-level joint learning. Firstly, node-type-aware structure learning and edge-type-aware semantic learning are sequentially performed on the original network and its line graph to get the initial embedding of nodes and the embedding of edges. Then, to optimize performance, type-level joint learning is performed through the alternating training of node embedding on the original network and edge embedding on the line graph. Finally, a new homogeneous network is constructed from the original heterogeneous network, and the graph attention model is further used on the new network to perform element-level joint learning. Experiments on three tasks and five public datasets show that our NEJE model performance improves by about 2.83% over other models, and even improves by 6.42% on average for the node clustering task on Digital Bibliography & Library Project (DBLP) dataset. |
| format | Article |
| id | doaj-art-dd5e750538b646d9b3fd035fccb498b7 |
| institution | Kabale University |
| issn | 2096-0654 |
| language | English |
| publishDate | 2024-09-01 |
| publisher | Tsinghua University Press |
| record_format | Article |
| series | Big Data Mining and Analytics |
| spelling | doaj-art-dd5e750538b646d9b3fd035fccb498b72025-02-03T00:12:56ZengTsinghua University PressBig Data Mining and Analytics2096-06542024-09-017373075210.26599/BDMA.2023.9020037Node and Edge Joint Embedding for Heterogeneous Information NetworkLei Chen0Yuan Li1Hualiang Liu2Haomiao Guo3School of Information and Electrical Engineering, Hunan University of Science and Technology, Xiangtan 411201, ChinaSchool of Information and Electrical Engineering, Hunan University of Science and Technology, Xiangtan 411201, ChinaChangde Water Meter Manufacture Co. Ltd., Changde 415000, ChinaSchool of Information and Electrical Engineering, Hunan University of Science and Technology, Xiangtan 411201, ChinaDue to the heterogeneity of nodes and edges, heterogeneous network embedding is a very challenging task to embed highly coupled networks into a set of low-dimensional vectors. Existing models either only learn embedding vectors for nodes or only for edges. These two methods of embedding learning are rarely performed in the same model, and they both overlook the internal correlation between nodes and edges. To solve these problems, a node and edge joint embedding model is proposed for Heterogeneous Information Networks (HINs), called NEJE. The NEJE model can better capture the latent structural and semantic information from an HIN through two joint learning strategies: type-level joint learning and element-level joint learning. Firstly, node-type-aware structure learning and edge-type-aware semantic learning are sequentially performed on the original network and its line graph to get the initial embedding of nodes and the embedding of edges. Then, to optimize performance, type-level joint learning is performed through the alternating training of node embedding on the original network and edge embedding on the line graph. Finally, a new homogeneous network is constructed from the original heterogeneous network, and the graph attention model is further used on the new network to perform element-level joint learning. Experiments on three tasks and five public datasets show that our NEJE model performance improves by about 2.83% over other models, and even improves by 6.42% on average for the node clustering task on Digital Bibliography & Library Project (DBLP) dataset.https://www.sciopen.com/article/10.26599/BDMA.2023.9020037node embeddingedge embeddingjoint embeddingheterogeneous information network (hin) |
| spellingShingle | Lei Chen Yuan Li Hualiang Liu Haomiao Guo Node and Edge Joint Embedding for Heterogeneous Information Network Big Data Mining and Analytics node embedding edge embedding joint embedding heterogeneous information network (hin) |
| title | Node and Edge Joint Embedding for Heterogeneous Information Network |
| title_full | Node and Edge Joint Embedding for Heterogeneous Information Network |
| title_fullStr | Node and Edge Joint Embedding for Heterogeneous Information Network |
| title_full_unstemmed | Node and Edge Joint Embedding for Heterogeneous Information Network |
| title_short | Node and Edge Joint Embedding for Heterogeneous Information Network |
| title_sort | node and edge joint embedding for heterogeneous information network |
| topic | node embedding edge embedding joint embedding heterogeneous information network (hin) |
| url | https://www.sciopen.com/article/10.26599/BDMA.2023.9020037 |
| work_keys_str_mv | AT leichen nodeandedgejointembeddingforheterogeneousinformationnetwork AT yuanli nodeandedgejointembeddingforheterogeneousinformationnetwork AT hualiangliu nodeandedgejointembeddingforheterogeneousinformationnetwork AT haomiaoguo nodeandedgejointembeddingforheterogeneousinformationnetwork |