A Two-Stage Deep-Learning Model for Link Prediction Based on Network Structure and Node Attributes

Peng Liu, Liang Gui (), Huirong Wang and Muhammad Riaz
Additional contact information
Peng Liu: School of Economics and Management, Jiangsu University of Science and Technology, Zhenjiang 212003, China
Liang Gui: School of Economics and Management, Jiangsu University of Science and Technology, Zhenjiang 212003, China
Huirong Wang: School of Economics and Management, Jiangsu University of Science and Technology, Zhenjiang 212003, China
Muhammad Riaz: School of Economics and Management, Jiangsu University of Science and Technology, Zhenjiang 212003, China

Sustainability, 2022, vol. 14, issue 23, 1-15

Abstract: Link prediction, which is used to identify the potential relationship between nodes, is an important issue in network science. In existing studies, the traditional methods based on the structural similarity of nodes make it challenging to complete the task of link prediction in large-scale or sparse networks. Although emerging methods based on deep learning can solve this problem, most of the work mainly completes the link prediction through the similarity of the representation vector of network structure information. Many empirical studies show that link formation is affected by node attributes, and similarity is not the only criterion for the formation of links in reality. Accordingly, this paper proposed a two-stage deep-learning model for link prediction (i.e, TDLP), where the node representation vector of the network structure and attributes was obtained in the first stage, while link prediction was realized through supervised learning in the second stage. The empirical results on real networks showed that our model significantly outperforms the traditional methods (e.g., CN and RA), as well as newly proposed deep-learning methods (e.g., GCN and VGAE). This study not only proposed a deep-learning framework for link prediction from the perspective of structure and attribute fusion and link distribution capture, but also lays a methodological foundation for practical applications based on link prediction.

Keywords: link prediction; deep learning; network structure; node attribute (search for similar items in EconPapers)
JEL-codes: O13 Q Q0 Q2 Q3 Q5 Q56 (search for similar items in EconPapers)
Date: 2022
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2071-1050/14/23/16299/pdf (application/pdf)
https://www.mdpi.com/2071-1050/14/23/16299/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jsusta:v:14:y:2022:i:23:p:16299-:d:995313

Access Statistics for this article

Sustainability is currently edited by Ms. Alexandra Wu

More articles in Sustainability from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().