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

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Liu, Chunjiang, Han, Yikun, Xu, Haiyun, Yang, Shihan, Wang, Kaidi, Su, Yongye
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913200116596736
author Liu, Chunjiang
Han, Yikun
Xu, Haiyun
Yang, Shihan
Wang, Kaidi
Su, Yongye
author_facet Liu, Chunjiang
Han, Yikun
Xu, Haiyun
Yang, Shihan
Wang, Kaidi
Su, Yongye
contents 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 enhance performance across all models tested. For example, integrating Louvain with the GAT model resulted in an AUC score increase from 0.777 to 0.823, exemplifying the typical improvements observed. Similar gains are noted when Louvain is paired with other GNN architectures, confirming the robustness and effectiveness of incorporating community-level insights. This consistent uplift 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.
format Preprint
id arxiv_https___arxiv_org_abs_2401_02542
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Community Detection and Graph Neural Network Based Link Prediction Approach for Scientific Literature
Liu, Chunjiang
Han, Yikun
Xu, Haiyun
Yang, Shihan
Wang, Kaidi
Su, Yongye
Social and Information Networks
Artificial Intelligence
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 enhance performance across all models tested. For example, integrating Louvain with the GAT model resulted in an AUC score increase from 0.777 to 0.823, exemplifying the typical improvements observed. Similar gains are noted when Louvain is paired with other GNN architectures, confirming the robustness and effectiveness of incorporating community-level insights. This consistent uplift 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.
title A Community Detection and Graph Neural Network Based Link Prediction Approach for Scientific Literature
topic Social and Information Networks
Artificial Intelligence
url https://arxiv.org/abs/2401.02542