arXiv:2409.11282cs.CLcs.LG2024-09被引 5

用聊天模型知识蒸馏到小模型,让文档理解更高效

Leveraging Distillation Techniques for Document Understanding: A Case Study with FLAN-T5

  • 用知识蒸馏把ChatGPT的文档理解能力迁移到FLAN-T5
  • 在多个文档数据集上达到接近大模型的准确率
  • 适合资源有限但需高精度文档理解的场景

各类数字文档(如商业报告、环境评估)数量激增,推动文档理解技术发展。尽管大语言模型(LLM)在自然语言处理任务中表现卓越,但其直接应用于文档理解仍面临计算资源需求高、部署困难的问题。现有研究虽已验证大模型在该领域的潜力,但专有大模型常优于开源模型,限制了普及。本文提出一种新方法:通过知识蒸馏,将专有模型ChatGPT的文档理解知识迁移至轻量级模型FLAN-T5,结合标注策略与课程学习机制,实现高效知识传递。实验表明,该方法在多个文档理解基准上显著提升小模型性能,为资源受限场景提供了可扩展的解决方案,推动了自然语言处理与文档理解的发展。

原文摘要 · Abstract (English)

The surge of digital documents in various formats, including less standardized documents such as business reports and environmental assessments, underscores the growing importance of Document Understanding. While Large Language Models (LLMs) have showcased prowess across diverse natural language processing tasks, their direct application to Document Understanding remains a challenge. Previous research has demonstrated the utility of LLMs in this domain, yet their significant computational demands make them challenging to deploy effectively. Additionally, proprietary Blackbox LLMs often outperform their open-source counterparts, posing a barrier to widespread accessibility. In this paper, we delve into the realm of document understanding, leveraging distillation methods to harness the power of large LLMs while accommodating computational limitations. Specifically, we present a novel approach wherein we distill document understanding knowledge from the proprietary LLM ChatGPT into FLAN-T5. Our methodology integrates labeling and curriculum-learning mechanisms to facilitate efficient knowledge transfer. This work contributes to the advancement of document understanding methodologies by offering a scalable solution that bridges the gap between resource-intensive LLMs and practical applications. Our findings underscore the potential of distillation techniques in facilitating the deployment of sophisticated language models in real-world scenarios, thereby fostering advancements in natural language processing and document comprehension domains.

知识蒸馏文档理解FLAN-T5大模型

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