arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.09112cs.DLcs.AIcs.CL
收藏

HLM-Cite:用于基于文本的科学引文预测的混合语言模型工作流

HLM-Cite: Hybrid Language Model Workflow for Text-based Scientific Citation Prediction

Tsinghua University(清华大学)

Qianyue Hao, Jingyang Fan, Fengli Xu, Jian Yuan, Yong Li

AI总结:

本文提出HLM-Cite混合语言模型工作流,通过课程微调嵌入模型粗略检索核心引文,并利用LLM智能体单次推理排序,将引文预测从二元分类提升为区分核心引文、表面引文与非引文,在19个领域实现17.6%的性能提升。

摘要:

引文网络在现代科学中至关重要,预测一篇新论文(查询)会引用哪些先前的论文(候选)是一个关键问题。然而,论文引文的作用差异很大,从基础知识基础到表面语境都有。区分这些角色需要对论文之间的逻辑关系有更深的理解,而不仅仅是引文网络中的简单边。具有文本推理能力的LLM的出现为识别这些关系提供了新的可能性,但存在两大挑战。首先,在实践中,一篇新论文可能会从海量现有论文中选择其引文,其中文本超出了LLM的上下文长度。其次,论文之间的逻辑关系是隐式的,直接提示LLM预测引文可能导致表面级文本相似性,而非更深层的逻辑推理。在本文中,我们引入了核心引文的新概念,它识别超越表面提及的关键参考文献。因此,我们将引文预测任务从简单的二元分类提升为区分核心引文与表面引文及非引文。为此,我们提出HLM-Cite,一个用于引文预测的混合语言模型工作流,结合了嵌入和生成式LM。我们设计了一个课程微调流程,将预训练文本嵌入模型适配为从大量候选中粗略检索高可能性的核心引文,然后设计一个LLM智能体工作流,通过单次推理对检索到的论文进行排序,揭示论文之间的隐式关系。借助该流程,我们可将候选集扩展至100K篇论文。我们在19个科学领域评估了HLM-Cite,与SOTA方法相比展示了17.6%的性能提升。

英文摘要:

Citation networks are critical in modern science, and predicting which previous papers (candidates) will a new paper (query) cite is a critical problem. However, the roles of a paper's citations vary significantly, ranging from foundational knowledge basis to superficial contexts. Distinguishing these roles requires a deeper understanding of the logical relationships among papers, beyond simple edges in citation networks. The emergence of LLMs with textual reasoning capabilities offers new possibilities for discerning these relationships, but there are two major challenges. First, in practice, a new paper may select its citations from gigantic existing papers, where the texts exceed the context length of LLMs. Second, logical relationships between papers are implicit, and directly prompting an LLM to predict citations may result in surface-level textual similarities rather than the deeper logical reasoning. In this paper, we introduce the novel concept of core citation, which identifies the critical references that go beyond superficial mentions. Thereby, we elevate the citation prediction task from a simple binary classification to distinguishing core citations from both superficial citations and non-citations. To address this, we propose $\textbf{HLM-Cite}$, a $\textbf{H}$ybrid $\textbf{L}$anguage $\textbf{M}$odel workflow for citation prediction, which combines embedding and generative LMs. We design a curriculum finetune procedure to adapt a pretrained text embedding model to coarsely retrieve high-likelihood core citations from vast candidates and then design an LLM agentic workflow to rank the retrieved papers through one-shot reasoning, revealing the implicit relationships among papers. With the pipeline, we can scale the candidate sets to 100K papers. We evaluate HLM-Cite across 19 scientific fields, demonstrating a 17.6% performance improvement comparing SOTA methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑