A Hybrid Citation Recommendation Model With SciBERT and GraphSAGE

被引:1
|
作者
Dinh, Thi N. [1 ]
Pham, Phu [2 ]
Nguyen, Giang L. [3 ]
Nguyen, Ngoc Thanh [4 ]
Vo, Bay [2 ]
机构
[1] Vietnam Acad Sci & Technol, Grad Univ Sci & Technol, Hanoi 100000, Vietnam
[2] HUTECH Univ, Fac Informat Technol, Ho Chi Minh City 700000, Vietnam
[3] Vietnam Acad Sci & Technol, Inst Informat Technol, Hanoi, Vietnam
[4] Wroclaw Univ Sci & Technol, Dept Appl Informat, PL-50370 Wroclaw, Poland
关键词
Context modeling; Semantics; Metadata; Recommender systems; Deep learning; Vectors; Training; Encoding; Bidirectional control; Natural language processing; Citation recommendation; deep learning; graph-based representation; GraphSAGE; natural language processing; SciBERT;
D O I
10.1109/TSMC.2024.3490774
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
As the number of scientific publications continues to increase at a dizzying rate, researchers face challenges related to spending too much time and effort searching for appropriate papers to cite in their work. Citation recommendation models have thus been developed to automatically generate a list of relevant papers for a specific text passage, thus helping to reduce the workload for scientists and contribute to better-quality research. Consequently, this research direction has recently attracted significant interest in the scientific community. However, the current citation recommendation models still focus primarily on the citation context and do not adequately address the metadata of papers, such as the citation links, publication time, and venue. To overcome these problems, in this study, we propose the SciBERT-GraphSAGE which is a hybrid deep learning-based model for recommending a list of academic papers by considering both the citation context and this article's metadata. Our model has two important components: 1) SciBERT for text data representation learning and 2) GraphSAGE for learning the representations of this article's citation links. We validate the effectiveness of our model on three benchmark datasets: 1) FullTextPeerRead; 2) ACL; and 3) RefSeer. The results from experiments demonstrate that our novel SciBERT-GraphSAGE model outperforms previous advanced models in terms of Recall@K, mean reciprocal rank (MRR), and mean average precision (MAP).
引用
收藏
页码:852 / 863
页数:12
相关论文
共 50 条
  • [1] A Hybrid Discriminative Mixture Model for Cumulative Citation Recommendation
    Ma, Lerong
    Song, Dandan
    Liao, Lejian
    Wang, Jingang
    IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2020, 32 (04) : 617 - 630
  • [2] Enhancing local citation recommendation with recurrent highway networks and SciBERT-based embedding
    Dinh, Thi N.
    Pham, Phu
    Nguyen, Giang L.
    Vo, Bay
    EXPERT SYSTEMS WITH APPLICATIONS, 2024, 243
  • [3] Local Citation Recommendation with Hierarchical-Attention Text Encoder and SciBERT-Based Reranking
    Gu, Nianlong
    Gao, Yingqiang
    Hahnloser, Richard H. R.
    ADVANCES IN INFORMATION RETRIEVAL, PT I, 2022, 13185 : 274 - 288
  • [4] A Model of Relevant Common Author and Citation Authority Propagation for Citation Recommendation
    Hsiao, Bo-Yu
    Chung, Chih-Heng
    Dai, Bi-Ru
    2015 16TH IEEE INTERNATIONAL CONFERENCE ON MOBILE DATA MANAGEMENT, VOL 2, 2015, : 117 - 119
  • [5] RAR-SB: research article recommendation using SciBERT with BiGRU
    Nimbeshaho Thierry
    Bing-Kun Bao
    Zafar Ali
    Scientometrics, 2023, 128 : 6427 - 6448
  • [6] Entity Burst Discriminative Model for Cumulative Citation Recommendation
    Lerong Ma
    JournalofBeijingInstituteofTechnology, 2019, 28 (02) : 356 - 364
  • [7] A Neural Probabilistic Model for Context Based Citation Recommendation
    Huang, Wenyi
    Wu, Zhaohui
    Liang, Chen
    Mitra, Prasenjit
    Giles, C. Lee
    PROCEEDINGS OF THE TWENTY-NINTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, 2015, : 2404 - 2410
  • [8] Entity Burst Discriminative Model for Cumulative Citation Recommendation
    Ma L.
    Journal of Beijing Institute of Technology (English Edition), 2019, 28 (02): : 356 - 364
  • [9] Citation recommendation based on citation tendency
    Xi Chen
    Huan-jing Zhao
    Shu Zhao
    Jie Chen
    Yan-ping Zhang
    Scientometrics, 2019, 121 : 937 - 956
  • [10] Graph Neural Collaborative Topic Model for Citation Recommendation
    Xie, Qianqian
    Zhu, Yutao
    Huang, Jimin
    Du, Pan
    Nie, Jian-Yun
    ACM TRANSACTIONS ON INFORMATION SYSTEMS, 2022, 40 (03)