Saved in:
Bibliographic Details
Main Authors: Xu, Aobo, Chang, Bingyu, Liu, Qingpeng, Jian, Ling
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2407.17722
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910570877288448
author Xu, Aobo
Chang, Bingyu
Liu, Qingpeng
Jian, Ling
author_facet Xu, Aobo
Chang, Bingyu
Liu, Qingpeng
Jian, Ling
contents Identifying significant references within the complex interrelations of a citation knowledge graph is challenging, which encompasses connections through citations, authorship, keywords, and other relational attributes. The Paper Source Tracing (PST) task seeks to automate the identification of pivotal references for given scholarly articles utilizing advanced data mining techniques. In the KDD CUP OAG-Challenge PST track, we design a recommendation-based framework tailored for the PST task. This framework employs the Neural Collaborative Filtering (NCF) model to generate final predictions. To process the textual attributes of the papers and extract input features for the model, we utilize SciBERT, a pre-trained language model. According to the experimental results, our method achieved a score of 0.37814 on the Mean Average Precision (MAP) metric, outperforming baseline models and ranking 11th among all participating teams. The source code is publicly available at https://github.com/MyLove-XAB/KDDCupFinal.
format Preprint
id arxiv_https___arxiv_org_abs_2407_17722
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Text-Driven Neural Collaborative Filtering Model for Paper Source Tracing
Xu, Aobo
Chang, Bingyu
Liu, Qingpeng
Jian, Ling
Information Retrieval
Machine Learning
Identifying significant references within the complex interrelations of a citation knowledge graph is challenging, which encompasses connections through citations, authorship, keywords, and other relational attributes. The Paper Source Tracing (PST) task seeks to automate the identification of pivotal references for given scholarly articles utilizing advanced data mining techniques. In the KDD CUP OAG-Challenge PST track, we design a recommendation-based framework tailored for the PST task. This framework employs the Neural Collaborative Filtering (NCF) model to generate final predictions. To process the textual attributes of the papers and extract input features for the model, we utilize SciBERT, a pre-trained language model. According to the experimental results, our method achieved a score of 0.37814 on the Mean Average Precision (MAP) metric, outperforming baseline models and ranking 11th among all participating teams. The source code is publicly available at https://github.com/MyLove-XAB/KDDCupFinal.
title Text-Driven Neural Collaborative Filtering Model for Paper Source Tracing
topic Information Retrieval
Machine Learning
url https://arxiv.org/abs/2407.17722