A Community Detection and Graph-Neural-Network-Based Link Prediction Approach for Scientific Literature

Chunjiang Liu (), Yikun Han, Haiyun Xu, Shihan Yang, Kaidi Wang and Yongye Su
Additional contact information
Chunjiang Liu: Chengdu Library and Information Center, Chinese Academy of Sciences, Chengdu 610299, China
Yikun Han: Department of Statistics, University of Michigan, Ann Arbor, MI 48109, USA
Haiyun Xu: School of Business, Shandong University of Technology, Zibo 255000, China
Shihan Yang: Faculty of Management and Economics, Kunming University of Science and Technology, Kunming 650031, China
Kaidi Wang: School of Business, Macau University of Science and Technology, Macau 999078, China
Yongye Su: Department of Computer Science, Purdue University, West Lafayette, IN 47907, USA

Mathematics, 2024, vol. 12, issue 3, 1-20

Abstract: This study presents a novel approach that synergizes community detection algorithms with various Graph Neural Network (GNN) models to bolster link prediction in scientific literature networks. By integrating the Louvain community detection algorithm into our GNN frameworks, we consistently enhanced the performance across all models tested. For example, integrating the Louvain model with the GAT model resulted in an AUC score increase from 0.777 to 0.823, exemplifying the typical improvements observed. Similar gains were noted when the Louvain model was paired with other GNN architectures, confirming the robustness and effectiveness of incorporating community-level insights. This consistent increase in performance—reflected in our extensive experimentation on bipartite graphs of scientific collaborations and citations—highlights the synergistic potential of combining community detection with GNNs to overcome common link prediction challenges such as scalability and resolution limits. Our findings advocate for the integration of community structures as a significant step forward in the predictive accuracy of network science models, offering a comprehensive understanding of scientific collaboration patterns through the lens of advanced machine learning techniques.

Keywords: link prediction; graph neural network; community detection; network analysis; citation graph; deep learning (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/12/3/369/pdf (application/pdf)
https://www.mdpi.com/2227-7390/12/3/369/ (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:jmathe:v:12:y:2024:i:3:p:369-:d:1325143

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

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