arXiv:2412.05137cs.AI2024-12中稿 · COLING 2025被引 17

用大模型+检索解决海量科学文献动态分类难题

Can Large Language Models Serve as Effective Classifiers for Hierarchical Multi-Label Classification of Scientific Documents at Industrial Scale?

  • 结合大模型与密集检索,零样本实现动态标签分类
  • 在SSRN数据集上准确率显著提升,成本大幅降低
  • 适合需要实时更新标签体系的工业级文献管理系统

我们针对工业规模下的层次化多标签分类(HMC)问题,处理数十万篇文档在数千个动态标签间的分类任务。科学出版物快速增长使得分类方法必须具备可扩展性和高效性,而分类体系不断演进(新增、合并、淘汰类别)进一步加剧挑战。传统机器学习方法需每次更新标签体系时重新训练,面临标注数据收集和模型适配的高成本。大语言模型(LLM)虽在复杂任务中表现优异,但在大规模动态标签场景下受限于输入长度。本文提出新方法,融合大模型与密集检索技术,避免重训练,实现零样本实时标签分配。我们在涵盖多学科的预印本库SSRN上评估,验证了方法在分类准确率和成本效率上的显著提升。通过构建面向动态标签体系的定制化评估框架,并开源代码,本研究为大模型在工业级文献分类中的应用提供了关键洞见。

原文摘要 · Abstract (English)

We address the task of hierarchical multi-label classification (HMC) of scientific documents at an industrial scale, where hundreds of thousands of documents must be classified across thousands of dynamic labels. The rapid growth of scientific publications necessitates scalable and efficient methods for classification, further complicated by the evolving nature of taxonomies--where new categories are introduced, existing ones are merged, and outdated ones are deprecated. Traditional machine learning approaches, which require costly retraining with each taxonomy update, become impractical due to the high overhead of labelled data collection and model adaptation. Large Language Models (LLMs) have demonstrated great potential in complex tasks such as multi-label classification. However, applying them to large and dynamic taxonomies presents unique challenges as the vast number of labels can exceed LLMs' input limits. In this paper, we present novel methods that combine the strengths of LLMs with dense retrieval techniques to overcome these challenges. Our approach avoids retraining by leveraging zero-shot HMC for real-time label assignment. We evaluate the effectiveness of our methods on SSRN, a large repository of preprints spanning multiple disciplines, and demonstrate significant improvements in both classification accuracy and cost-efficiency. By developing a tailored evaluation framework for dynamic taxonomies and publicly releasing our code, this research provides critical insights into applying LLMs for document classification, where the number of classes corresponds to the number of nodes in a large taxonomy, at an industrial scale.

大模型文献分类动态标签检索增强

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。