提出无需计算海森矩阵的高效数据筛选方法,提升大模型训练效果。
BLADE: Scalable Bi-level Adaptive Data Selection for LLM Training

- 用拉格朗日乘子将双层优化转为单层惩罚目标,避开复杂计算
- 动态参考模型随训练同步更新,比静态模型更适应学习过程
- 在大规模语料上显著优于现有方法,适合训练万亿级参数模型
随着大语言模型数据集规模扩展至数万亿标记,数据筛选成为过滤冗余噪声、构建自适应学习路径的关键。现有方法主要分为两类:基于影响的方法虽具理论基础但需不可行的逆海森计算;基于过剩损失的方法计算高效,却依赖与训练中动态变化的代理模型逐渐失配的静态参考模型。本文提出BLADE(Bi-Level Adaptive Data sElection),一种无海森矩阵的双层自适应数据筛选框架。通过拉格朗日乘子将影响法背后的双层优化问题转化为带惩罚的单层目标,避免逆海森计算,并揭示其与过剩损失法的原理关联。该目标形式上恢复为过剩损失,但以动态参考模型替代静态模型,保持与训练同步。理论上证明该惩罚形式保证一阶收敛性。针对高效在线批次筛选,我们设计了无记忆的随机块坐标Frank-Wolfe算法。大量实验表明,BLADE持续优于当前最优数据筛选基线,为大模型训练提供实用方案。
原文摘要 · Abstract (English)
As Large Language Model (LLM) datasets scale to trillions of tokens, data selection has emerged as a critical frontier to filter out uninformative noise and construct adaptive learning trajectories. Beyond static heuristic filtering, advanced data selection methods for LLM training largely follow two paradigms, each with fundamental limitations. Influence-based methods provide principled bi-level objectives but require intractable inverse-Hessian computations, while excess-loss methods are computationally efficient but rely on a static reference model that becomes misaligned with the evolving proxy model during training. We propose BLADE (Bi-Level Adaptive Data sElection), a Hessian-free framework for data selection. BLADE reformulates the bi-level optimization problem underlying influence-based methods as a penalized single-level objective via Lagrange multipliers, avoiding inverse-Hessian computation while revealing a principled connection to excess-loss based data selection. The resulting objective recovers an excess-loss form but replaces the static reference model with a dynamic one that stays synchronized with training. Theoretically, we prove that this penalized formulation guarantees first-order convergence. For efficient online batch selection, we instantiate BLADE as a memoryless randomized block-coordinate Frank-Wolfe algorithm. Extensive experiments show that BLADE consistently outperforms state-of-the-art data selection baselines, providing a practical recipe for LLM training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。