JournalArticle Open access

Sensitivity Analysis of a BERT-based scholarly recommendation system

  • The Florida AI Research Society
Research footprint

At a glance

Citations
6
References
32
Comments
0
Paper overview

Abstract

With the exponential growth of publicly available datasets, a scholarly recommendation system of datasets would be an essential tool in the field of information filtering. Recommending datasets to users can be formulated as a classification problem where deep learning models can be carefully trained. In such a case, when preparing training data for the learning models, one needs to consider different ratios of false and true pairs. Therefore, a sensitivity analysis is necessary. In this work, we conduct a sensitivity analysis using different class ratios on a deep learning model (BERT) for recommending datasets. We found out that our BERT-based recommender model is relatively robust using recommender metrics such as Mean Reciprocal Rank (MRR)@k, Recall@k, etc., except for the extreme class imbalance case (1:5000). Therefore, we conclude that a moderate ratio of the random negative sampling scheme, (in our case 1:10) is reasonable, sufficient and time-efficient in the recommendation system training

Record transparency

Publication details

DOI
10.32473/flairs.v35i.130595
Semantic Scholar
7078f271b66b84c4304a42fafd5e4fc9270a754e
Document type
JournalArticle
Source
The Florida AI Research Society
Last metadata update
Community

Comments

Log in to join the discussion.

  1. No comments yet. Start the discussion.