MDCure用自动构建数据提升大模型多文档处理能力,无需额外训练。
MDCure: A Scalable Pipeline for Multi-Document Instruction-Following
- 通过定向提示生成高质量多文档指令数据
- 在多个基准上使模型性能最高提升75.1%
- 适合想提升多文档理解能力的研究者和开发者
多文档处理对大模型完成摘要、问答等真实任务至关重要。尽管大模型已能处理长文本,但多文档场景仍面临跨文档依赖管理、冗余和结构不连贯等挑战。为此,我们提出MDCure——一种可扩展的指令数据生成框架,无需预训练计算成本或人工标注数据即可增强大模型的多文档能力。MDCure通过针对性提示在文章集合上生成高质量合成指令数据,并引入低成本的多文档专用奖励模型MDCureRM,用于评估和筛选数据的训练价值。该框架兼容开源与闭源模型及PPO等策略优化方法,使小型开源模型也能生成优于专有模型的高质量多文档指令数据。我们在FlanT5、Qwen2和LLAMA3.1系列中对多种高达700亿参数的大模型进行了微调。在涵盖多种任务与领域的多文档与长上下文基准上的广泛评估显示,MDCure相较预训练基线和基础模型性能提升最高达75.1%。代码、数据集与模型已在https://github.com/yale-nlp/MDCure公开。
原文摘要 · Abstract (English)
Multi-document (MD) processing is crucial for LLMs to handle real-world tasks such as summarization and question-answering across large sets of documents. While LLMs have improved at processing long inputs, MD contexts still present unique difficulties, including management of inter-document dependencies, redundancy, and incoherent structures. To address this challenge, we introduce MDCure, a scalable and effective instruction data generation framework to enhance the MD capabilities of LLMs without the computational cost of pre-training or reliance on human-annotated data. MDCure generates high-quality synthetic MD instruction data over sets of articles via targeted prompts. We also introduce MDCureRM, a cost-effective, MD-specific reward model to score and filter generated data based on their training utility for MD settings. MDCure is compatible with open- and closed-source models in addition to policy optimization methods such as PPO, enabling even small open-source models to surpass proprietary LLMs as strong generators of high-quality MD instruction data without further data filtering. With MDCure, we fine-tune a wide variety of LLMs up to 70B parameters in size from the FlanT5, Qwen2, and LLAMA3.1 model families. Extensive evaluations on a wide range of MD and long-context benchmarks spanning various tasks and domains show MDCure consistently improves performance over pre-trained baselines and base models by up to 75.1%. Our code, datasets, and models are available at https://github.com/yale-nlp/MDCure.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。