Affiliations: [a]
Department of Computer Science and Information Engineering, Tamkang University, New Taipei, Taiwan
| [b]
Department of English, Tamkang University, New Taipei, Taiwan
| [c]
Department of Computer Science and Engineering, National Institute of Technology, Shillong, India
Correspondence:
[*]
Corresponding author. Chih-Yung Chang, Department of Computer Science and Information Engineering, Tamkang University, New Taipei 25137, Taiwan. E-mail: [email protected].
Abstract: With the field of technology has witnessed rapid advancements, attracting an ever-growing community of researchers dedicated to developing theories and techniques. This paper proposes an innovative ICRM (Intelligent Citation Recommendation Mechanism), designed to automate the process of suggesting the appropriate number of citations for individual brackets within a document. The proposed ICRM comprises three phases: Coarse-grained Weighted Bag of Word (WCBW), Fine-grained SciBERT (FSB) and Citation Adjustment phases. Firstly, the WCBW phase employs TF-IDF to extract keywords from both target and candidate documents, forming vectors that capture word significance along with metadata like authorship, keywords, and titles. It aims to identify relevant papers from a database, serving as initial candidates for each bracket. Secondly, the FSB phase employs the SciBERT model to assess the similarity between candidate documents and the local context around brackets, enhancing the precision of recommendations. It refines this selection by analyzing candidate-document relationships within the proximity of the brackets. Lastly, the Citation Adjustment phase tackles overlapping citations and ensures that recommended citation numbers align with user-defined criteria, resolving issues of imbalance. The simulation results demonstrate that the proposed ICRM outperforms existing models significantly in terms of precision, recall and F1-score.
Keywords: Citation recommendation, TF-IDF, weighted bag of word, BERT
