Conference

A Hybrid Model Based on Pre-trained Language Model and Graph Neural Networks for Scientific Paper Recommendation

  • 2022 4th International Conference on Machine Learning, Big Data and Business Intelligence (MLBDBI)
Research footprint

At a glance

Citations
2
References
32
Comments
0
Paper overview

Abstract

Due to the overload of published scientific articles, paper recommendation has long been a critical research problem for automatically recommending high quality scientific papers for researchers. For scientific paper recommendation, the existing methods are mainly based on the user-paper interaction and the textual content of the paper. Although many scientific paper recommendation systems have been proposed, there are still two shortcomings: 1) the interaction information-based approaches cannot capture the higher-order user paper interaction information. 2) the content-based approaches cannot exploit the deep semantic information of the text, which impairs the interpretability of the models. To improve recommendation quality and model interpretability, we propose Citation Graph Collaborative Filtering (CGCF), which unifies document representation and Graph Neural Network (GNN). Our method consists of three steps:1) Build the user-paper bipartite graph based on citation relationship. 2) Use a pre-trained language model to initialize the paper’s embedding with its title and abstract. 3) Tune node embeddings using high-order connectivity information captured by several stacked message-passing layers on the graph built in the first step and finally make predictions predicted by these learned embeddings. Experiment shows that our model performs better while ensuring training efficiency.

Record transparency

Publication details

DOI
10.1109/mlbdbi58171.2022.00078
Semantic Scholar
ba1c4c27c65a3b744e7812d905cfbbe6554d059a
Document type
Conference
Source
2022 4th International Conference on Machine Learning, Big Data and Business Intelligence (MLBDBI)
Last metadata update
Community

Comments

Log in to join the discussion.

  1. No comments yet. Start the discussion.