arXiv:2505.20380cs.LG2025-05NeurIPS被引 4

GRAPE动态调整数据与任务权重,让模型在多任务上更稳健。

GRAPE: Optimize Data Mixture for Group Robust Multi-target Adaptive Pretraining

  • 通过任务难度自适应调节权重,优先提升难任务表现。
  • 在6个基准上优于基线,多语言场景下8个低资源语言效果更好。
  • 适合需要跨任务、跨语言鲁棒性的大模型预训练场景。

大型语言模型在不同下游应用中的表现,根本上取决于其预训练语料的质量与构成。现有领域重加权算法主要针对单一目标任务优化数据混合,导致模型过度拟合特定目标,而在其他基准上性能显著下降。本文提出组鲁棒多目标自适应预训练(GRAPE),一种新型多源多目标领域重加权框架,旨在同时优化多个目标任务的鲁棒表现。GRAPE动态调整源领域权重(域权重),并同步调节量化各目标任务相对重要性的任务权重。该自适应过程依据学习难度优先处理困难任务。我们将其建模为一个极小极大优化问题:内层最大化通过群体分布式鲁棒优化(DRO)调整任务权重,当前数据混合下提升最少的任务获得更高权重;外层最小化则优化域权重,以最大化在优先任务上的损失减少。在ClimbLab和SlimPajama数据集上的实验表明,GRAPE在6个基准上的推理性能持续优于基线方法。此外,在多语言目标场景下,GRAPE能有效从主流语言中识别出最优训练混合,实现对8个低资源目标语言的优越语言建模能力。

原文摘要 · Abstract (English)

The performance of large language models (LLMs) across diverse downstream applications is fundamentally governed by the quality and composition of their pretraining corpora. Existing domain reweighting algorithms primarily optimize data mixtures for a single target task, thereby resulting in models that overfit to specialized objectives while exhibiting substantial performance degradation on other benchmarks. This paper introduces Group Robust Multi-target Adaptive PrEtraining (GRAPE), a novel multi-source-multi-target domain reweighting framework designed to calibrate pretraining data mixtures for robust performance across multiple target tasks simultaneously. GRAPE dynamically adjusts sampling weights across source domains (domain weights) while concurrently modulating task weights that quantify the relative importance of each individual target task. This adaptive process prioritizes tasks based on their learning difficulty throughout training. We formulate this interleaved reweighting mechanism as a minimax optimization problem: The inner maximization adjusts task weights leveraging group distributed-robust-optimization (DRO), where those tasks demonstrating the least improvement under the current data mixture are prioritized with higher weights; The outer minimization then optimizes domain weights to maximize loss reduction on the prioritized tasks. Experiments on ClimbLab and SlimPajama datasets demonstrate that GRAPE consistently outperforms baseline methods in terms of reasoning performance across 6 benchmarks. Furthermore, when applied to multilingual targets, GRAPE effectively identifies optimal training mixtures from mainstream languages, achieving superior language modeling capabilities across 8 low-resource target languages.

预训练多任务鲁棒性数据混合

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