arXiv:2501.12766cs.CLcs.AI2025-01ICML被引 25

用负样本干扰增强长文本依赖建模,无需真实长文档也能训练出强长程理解模型。

NExtLong: Toward Effective Long-Context Training without Long Documents

  • 将文档拆成元块,用检索到的负样本交叉插入以强化长距离依赖识别能力。
  • 在HELMET和RULER基准上优于现有合成数据方法及使用真实长文档训练的主流模型。
  • 适合想提升模型长文本理解但缺乏真实长文档数据的研究者与开发者。

大语言模型(LLM)虽已实现更长上下文窗口,但受限于真实长文档稀缺。现有方法多通过合成数据训练,却缺乏明确机制来强化长程依赖建模。为此,我们提出NExtLong框架,通过负样本扩展(Negative document Extension)生成长上下文数据。该方法将文档分解为多个元块,并从预训练语料中检索硬负样本进行交叉插入,迫使模型区分长程相关上下文与干扰内容,从而提升长程依赖建模能力。大量实验表明,NExtLong在HELMET和RULER基准上显著优于现有合成数据方法及基于非合成长文档训练的领先模型。结果证明其可有效降低对真实长文档的依赖,是构建先进长上下文LLM的有效方案。

原文摘要 · Abstract (English)

Large language models (LLMs) with extended context windows have made significant strides yet remain a challenge due to the scarcity of long documents. Existing methods tend to synthesize long-context data but lack a clear mechanism to reinforce the long-range dependency modeling. To address this limitation, we propose NExtLong, a novel framework for synthesizing long-context data through Negative document Extension. NExtLong decomposes a document into multiple meta-chunks and extends the context by interleaving hard negative distractors retrieved from pretraining corpora. This approach compels the model to discriminate long-range dependent context from distracting content, enhancing its ability to model long-range dependencies. Extensive experiments demonstrate that NExtLong achieves significant performance improvements on the HELMET and RULER benchmarks compared to existing long-context synthesis approaches and leading models, which are trained on non-synthetic long documents. These findings highlight NExtLong's ability to reduce reliance on non-synthetic long documents, making it an effective framework for developing advanced long-context LLMs.

长文本建模合成数据负样本LLM训练

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