V% N, D' U( N, R+ {( m; ~8 }
2 }6 k4 i9 Z+ ^: J5 M1 e( L+ n/ N+ k! {
在 “大模型预训练” 这件事上,行业里几乎有一条默认信条:想要更强,就得喂更多、更新、更高质量的数据。但最近一篇来自阿里巴巴、上海交大、UW–Madison 等团队的工作,在 Hugging Face Daily Paper 冲上月度 Top1 后,直接把这个共识撕开了一道口子:只从中低质量数据里动态挑选,竟然能打赢 “高质量数据优先” 的经典训练配方。
0 U; V/ o+ C* a1 }1 o( t& N
3 F/ M! F- C" @$ t r; A; O: }2 H* F3 z( g
( G0 { k1 H* `; u这条结论之所以让社区炸锅,不只是因为它 “反直觉”,更因为它击中了一个长期被忽略的问题:我们今天用 AdamW、Muon 训练大模型,却还在用偏 SGD 时代的思路给数据打分。 说得更直白一点 —— 我们一直在拿 “旧地图” 给 “新导航” 指路。而这篇工作真正厉害的地方是:它不是在 “玄学筛数据” 上做小修小补,而是把数据选择这件事,第一次系统性地对齐到了优化器真正决定的更新方向上。& A8 V2 Y+ x* s* q
7 f- e1 m" g% C+ Y* B
) \/ X2 c7 v7 ~" w* y2 h0 n' O
( v& a y" d& w* I9 \6 {
4 c7 |; b. t$ P) d: ?$ ?% @
. a. C8 U, p; o$ `% P- 论文链接:https://arxiv.org/pdf/2602.05409 x7 X' J9 ~: j$ F- H" Q9 r
3 ^0 Y, e; N" D c6 H8 r
从 “数据饥渴” 到 “数据墙”,预训练进入 “每个 token 都要算账” 的时代
' H, ]: i! w0 e- t. D& A. {过去十年,大模型能力提升的主旋律,是一条几乎无人质疑的路径:模型更大、数据更多、算力更猛。但今天,这条路正在撞上数据侧的天花板 —— 高质量公共文本逐渐枯竭,“Data Wall(数据墙)” 正在浮现。预训练也因此被迫从一个 “吞吐问题”,转向一个更关键的 “控制问题”:在第 t 步更新里,到底应该让哪些 token 来决定模型往哪走?
0 V5 {% H8 L8 X5 f" ~2 N; p% S+ Z围绕这个问题,行业里常见的两条路线都各有硬伤:) Y+ ?" g- P: Z( \' s9 R% B
0 h( {# o( X4 Y6 M2 C) t2 T6 j P
5 {' m9 U- l# U" v5 ]- 静态过滤:一次打分,长期食用。优点是简单、稳定、工程上好落地;但问题也很明显:它默认数据价值是 “固定的”。可现实是,模型会变强、训练阶段会切换、目标能力会迁移 —— 配方却不变,容易越训越 “钝”。5 _0 A9 E; r; a8 }& v
- 动态选择:能够随训练过程实时调整,听起来更合理;但大多数方法仍然默认 SGD 视角,用 “原始梯度” 来衡量样本价值。问题在于:现实中的 LLM 训练,早已全面转向 AdamW、Muon 等现代优化器。$ u d/ m3 g6 k; j% r
- & V. W5 N, i, C$ k. C6 Y( g
7 V' Z m& D' ?2 S; U8 c. ^2 `; F4 ~, [: s作者指出:当优化器改变了 “实际更新方向”,你用原始梯度选数据,就会出现一个 “方向偏差(misalignment gap)”—— 你以为喂的是有效训练信号,优化器却把更新投到另一条轨迹上。这就是 OPUS 的起点:数据选择不能再 “优化器无关”。! }) B. K1 H( i0 u% ^5 P
别替优化器点菜 —— 让优化器自己决定 “这口饭有没有用” * ~# N7 \2 V) v# D3 A9 w. U8 \" H+ s
4 V5 [* J( u; T7 U' c. e+ u
, K4 @6 C4 l: S1 j
" p- z2 E5 Z) u1 T1 O
OPUS 的全称很直白:Optimizer-induced Projected Utility Selection。
0 ]% C! g' e. ]5 r它做了一件看起来 “理所当然”,但过去很少被系统落实的事:: |3 U7 E$ b" t' M5 c
不再在 “原始梯度空间” 里评估样本价值,而是把效用(utility)定义在 “优化器诱导的有效更新空间” 里。
) @. y/ u2 G4 F& S9 }换句话说,在 AdamW / Muon 训练中,真正推动参数变化的,并不是原始梯度本身,而是经过优化器预处理之后的有效更新方向。OPUS 做的,就是直接计算(或近似计算)每个候选样本在当前 step 下对参数的 “有效推动”,并进一步追问一个更本质的问题:如果我按 AdamW / Muon 的真实更新方式走这一步,这个样本会不会让模型在目标分布上变得更好?7 `" e7 z7 |8 B2 X
于是,“数据选择” 这件事就不再只是文本质量打分,也不只是梯度相似度技巧,而是被升级成一个更原则化的目标:最大化每一步更新的收益(utility)。0 {6 H& M, S+ U0 W1 ], t7 t
OPUS 的 “三件套”—— 目标对齐、可扩展估计、稳定选择
! H0 R* K& N8 B! g/ ~+ K. U
( X+ h N. {0 d) i2 Z5 t8 ]6 f: ^
& ^+ Y3 z7 e- O/ M6 C
1 @* }. N. p8 U: @/ M9 @: B从论文 Figure 3 可以看到,OPUS 在每个训练 step 里,不再用原始梯度去 “猜” 样本价值,而是把样本效用定义在 AdamW / Muon 等优化器诱导的有效更新空间中。它的核心闭环可以概括为三步:# q# C% }% F# Q) x
" p, ~9 N" v, `; b1 P. a1 e
2 r* m& L6 u9 J( N7 E. j* u- 先对齐目标:构建与目标 benchmark 语义对齐的 Bench-Proxy 池,提供稳定的 “目标方向”;' I% J3 E" n' m2 R3 D# d
- 再高效估计:用 Ghost + CountSketch 近似估计候选样本对 proxy 方向的对齐收益;; t$ m0 Y+ ~" K
- 最后稳定选择:加入冗余惩罚,并通过 Boltzmann 软采样 选出当步最合适的训练样本。! G' \. m; @$ J& A
9 o( z& e S% l" W) I% n
这套设计的关键意义在于:它让 “数据选择” 第一次真正和 “优化器实际执行的更新轨迹” 处在同一几何、同一方向上,从而显著提升预训练效率与下游泛化表现。
0 b n6 b) C% ]. o! y4 r1)效用怎么定义?—— 在 “有效更新空间” 里做对齐,而不是在原始梯度里 “看着像”. i! t s1 V, Z6 a7 N1 @" s2 X
/ w9 y/ j8 q- G Z
" j& P, U; r% }' H6 L, ^% |+ v4 `6 t l2 K+ i8 @5 J1 ?9 u) r# ]
OPUS 把每个候选样本的价值拆成两部分:! B" r% H S7 H4 P1 [3 ?
* y* e. Q: E- m- c
# g! [8 ?- X! Z* | b- [' K D2 P- Alignment(对齐收益):样本带来的有效更新方向,是否与 “目标方向” 一致;& z: h- o+ H. o# m; F' ]3 u
- Redundancy Penalty(冗余惩罚):避免连续选到一堆方向高度相似的样本,导致更新过于集中、训练不稳、收益递减。 E$ m) M x# U5 |
: W5 h# F0 L& E# E! G0 ?4 C这套设计把 “选最有用” 与 “选得多样” 统一进同一个原则框架里:每一步不仅要更快下降,还要避免把更新压成一条细线。& E) }! `' {6 A
2)目标方向从哪来?——Bench-Proxy:既贴近 benchmark,又不脱离预训练流形& b; R' E, _ }$ a g8 \- z- n
Bench-Proxy 并不是 “随便找一批相似文本” 作为代理目标,而是通过一个检索式构建流程得到的。具体来说,作者使用冻结的句向量模型,将:* B) c8 B8 k8 x$ }0 H, G
8 x- `8 b9 u) N2 k7 A& M' K# O2 ?& n, ^$ g, ?% Z
- 目标评测基准的验证集样本(如 MMLU、HellaSwag 等),以及$ a( q6 o3 n/ z9 K- J! Y V- T
- 预训练语料中的文档# F8 v& Z8 W9 x0 \: T
% j: a% _1 Z- w1 ^" L
映射到同一语义空间,并计算余弦相似度。* ^+ A& ]" F1 S, v) s; `
随后,对每篇预训练文档分配一个 “相关性分数”(例如基于其与 benchmark 样本的最大相似度),再按分数排序并在给定 token 预算内选出一批文档,构成 Bench-Proxy 池。这样得到的代理池具有两个优点:& ~6 H' l& E0 E2 e5 i' U
4 M; E2 H* ?% Z( @5 i6 b6 Q+ ]
0 m$ |3 h7 b$ {5 s9 O1 D
- 语义上贴近目标 benchmark(有明确任务指向性);5 K0 g+ O: G9 k5 \/ U
- 分布上仍属于预训练语料流形(不会过度偏离预训练过程)。
5 u; o) |7 t8 W' H$ ~ 2 c/ L6 W2 `) h( M" x6 u
训练过程中,模型反复从这个 proxy 池抽样,用于提供更稳定、低噪声的目标方向,从而让每一步的数据选择更可靠。这一点很关键:OPUS 不是直接拿 benchmark 当训练数据,而是用 benchmark 去 “定义方向”,再在预训练分布里找可执行的推进路径。9 a3 |3 y3 Q5 @5 s3 G7 j
3)怎么把它做得足够快?——Ghost + CountSketch,把在线打分开销压到 “几乎可忽略”: ?* |4 @3 S2 P8 e0 X, P2 V6 N
在线数据选择最大的现实门槛,不是 “想法对不对”,而是 “算不算得动”。' U6 t3 N! O& q0 M) c8 A% |$ [* t
你不可能在每个 step 都为大量候选样本显式计算全维梯度并逐一打分。
$ B9 ^! q% N" Y5 ]; yOPUS 的工程解法是一套组合拳:
' C0 P5 h& x9 v
' m% ?" O, D& ]4 {3 h( Q2 A1 s/ P% M8 Z$ ]
- Ghost technique:利用线性层梯度的结构(如外积形式),避免显式构造完整高维梯度,降低显存与计算开销;
* ]8 ]# g4 e. Y$ L; F5 p - CountSketch:将高维有效更新投影到低维 sketch 空间,在近似保持内积结构的前提下完成对齐、相似度与冗余相关计算;
8 Y7 n; j$ R3 E - Boltzmann sampling:不直接贪心 top-k,而是通过温度控制的概率采样进行软选择,在利用高分样本的同时保留一定探索性,提升稳定性与多样性。
4 y3 R- P: n* J$ S 2 r6 I- ^. r( J2 y0 ]
结果是,OPUS 把 “每步在线选择数据” 的额外成本压到了一个可接受区间,使这件事在大规模预训练中也具备实际可行性。' z6 ]9 ^; ~$ I0 M" }& o
" ~& Q% h j c6 E
* V, l8 T4 n2 u2 Q' b9 z# _! {1 l, i4 O o6 k* D
在论文的实现与测量中,OPUS 的额外计算开销约为 4.7%,使得 “每一步都做数据选择” 在大规模训练中也依然可承受。
* b m4 k- \2 `$ U6 y6 g实验结果:不是小修小补,而是 “效率 + 性能” 同时抬头
: Z9 z. d3 O. H3 L: L1)FineWeb 预训练:平均 +2.2% 准确率,GPT-XL 上 8× 计算量节省
2 R( M' e c# [4 h( ^% b在 GPT-2 Large / XL 的 30B token 预训练设置中,OPUS 在 10 个基准上对比随机选择取得平均 1.5% 的准确率提升;在 GPT-XL 上还展示了 8× 计算效率提升的结果(相同效果所需计算显著降低)。更 “刺激” 的一点是:论文还报告 OPUS 能在某些设置中优于更大 token 预算(例如 60B token)训练的对照配置 —— 强调 “每步选对比多吃更关键”。
5 u) F" i+ X8 Q- [# D
- A: W+ ~/ C6 v5 D1 y6 N
/ @% Q% @4 `9 l- P+ Q3 g: \! r
$ P8 B$ W. f) \8 F W6 P2)FineWeb-Edu:反直觉名场面 —— 只用中等质量数据,也能打赢 “吃高质量数据” 的基线, |5 A G0 p0 Y; h" t1 y4 G
作者专门做了 “难度更高” 的对照:把数据按质量分层后,OPUS 只从中等质量(如 score 3)里动态挑选,却能超过一些使用更高质量分区(score 4–5)训练的强基线。在 GPT2 Large/XL 30B 使用 FineWeb-Edu 的预训练设置中,OPUS 在 10 个基准上对比从高质量数据随机选择取得平均 3.18% 的准确率提升。它传递的信号很清晰:& x1 ~- V0 T' t' U
数据质量很重要,但 “在正确的几何里、在正确的时机喂对样本”,可能更重要。+ A. y$ O! S; w5 t' K
2 j' l. Z9 u8 [- z& u
+ V: C9 u0 l' k% q. _: f7 T: j
. Q, \4 s* I+ M& `
2 j' l7 }- B; z. Q4 z
7 _$ h1 x2 I" u; m: _# l0 b
" K D: ]! U% t/ j; ]! E& L
除了主套件上的提升,论文还专门做了一组更 “刁钻” 的检验:把同一批 GPT-2 XL checkpoint 拿去测 不包含在 Bench-Proxy 构建目标里的 out-of-distribution 基准,包括 BBH、RACE、SuperGLUE 等。结果显示,OPUS 仍然取得最佳平均表现,达到了 40.07,明显高于随机选择以及多种静态和动态筛选基线。这一点非常关键:它说明 OPUS 的收益并不是 “对齐 proxy 就刷 proxy”,也不是把模型过拟合到那一小撮基准上;相反,即使评测换成 proxy 未覆盖的 OOD 任务,OPUS 依然能稳定带来泛化收益,侧面印证了其 “在优化器诱导更新空间里选有效训练信号” 的机制更接近提升真实能力,而非 benchmark 取巧。% B: O2 ]8 h: B" N+ j, Q
3)Domain PPL: 验证 “泛化而非刷分”
* T2 z: G' |! S; L9 p8 [除了任务准确率,论文还用一个更 “底层” 的指标检验模型的广谱语言建模能力:在 Health、Business、Politics、Education、History、Lifestyle、SCIence、Arts & Lit.、Entertainment、Computing 等 10 个不同领域的保留验证集上统计 PPL,越低越好。结果非常稳定:在 FineWeb 上训练 30B tokens 时,OPUS 在 GPT-2 Large 与 GPT-2 XL 两个规模下都拿到最低的平均 PPL—— 分别是 3.35 与 3.26,优于 Random、DSIR、QuRating、GREATS 等基线。更有意思的是,在 FineWeb-Edu 这类 “更高质量” 的子集上,OPUS 仍然保持领先:GPT-2 Large 的平均困惑度降到 3.49,GPT-2 XL 进一步到 3.45。这说明 OPUS 的提升不只体现在某几个 benchmark 上 “刷分”,而是在跨领域的语言建模质量上同样带来一致收益 —— 更接近一种可迁移、可泛化的训练信号增益。
/ y7 r5 m: z' U4)Continued Pre-training:Qwen3-8B 在 SciencePedia 上 0.5B token 追平 / 超过 3B token% K }0 u/ }. E! F3 v6 S& c
更贴近产业的 CPT 场景里,OPUS 在 Qwen3-8B-Base 上继续训练 SciencePedia:仅用0.5B tokens就达到最优表现,并且超过随机选择训练 3B tokens 的对照,等价于约 6× 的数据效率增益。对于 “专业域能力提升” 这种高成本任务,这种量级的效率提升极具吸引力。( f0 c1 U$ ^6 {/ L2 h' y/ X
A& Q& W q+ ]- L/ K/ c8 D, \3 N
3 b% _0 Y v" w1 k, E. G D6 F" c4 {% C% T* v2 `# g- F( G# P+ Y7 v
作者还给出了SciencePedia的分领域拆解结果,把提升拆到 “每个科学子域” 上看清楚:在 0.5B,1B,1.5B 三个 token 预算下,OPUS 在 OlympicArena(图中雷达图,覆盖 Math、Physics、Chemistry、Biology、Geography、Astronomy、CS、Text、以及多语种等维度)与 SciAssess(图中柱状图,Biology/Chemistry/Material/Medicine 等子域)中都表现出更稳定、更加均衡的收益。更关键的是,这种增益并非只靠某一个 “强项领域” 拉动平均分:即使把平均分拆开看,OPUS 在多个子域上都能保持竞争力,尤其在Material 与 Medicine等更偏专业的方向上,优势更明显。总体上,这组分域结果支持了论文的核心论点:OPUS 的改进不是 “挑某个领域刷上去”,而是在有限 token 预算下,把继续预训练的收益更有效地分配到不同科学子域,从而更接近 “用更少 token 覆盖更广能力” 的目标。
, n% U' [3 Q7 A. }
) T/ a; d- X7 [* s; C# ^( E4 w7 @3 j& S6 B
* t4 D3 y! V! D$ e) w从 “挑数据” 到 “挑更新”,OPUS 把预训练的控制权还给了优化器
* l4 E7 |1 h! N$ l: F V) O很多数据选择方法都卡在一个经典矛盾里:要么原则弱,像经验规则;要么原则强,但算不动。6 H& E1 w. @2 P9 g2 `) [0 [& M/ X
OPUS 的可贵之处在于,它不是只在理论上更 “正确”,也不是只在工程上更 “取巧”,而是把两者真正接到了一起:在原则上,它把样本效用定义到与 AdamW / Muon 等现代优化器一致的有效更新空间中;在工程上,它又通过 Ghost + CountSketch + Boltzmann 软采样,把 “每个 step 在线做数据选择” 的额外开销压到了可落地的范围。
2 i& T0 A4 `0 V3 l, ^2 r, X更重要的是,OPUS 并不排斥已有的数据工程手段,反而天然适合与静态过滤协同:静态过滤负责把明显低价值样本挡在门外,OPUS 负责在剩余候选中根据训练动态做细粒度选择。 换句话说,它第一次比较完整地把 “数据治理” 与 “训练动力学” 接成了一个闭环。6 n" H3 X+ H1 L4 H4 y/ e! ?
这也是 OPUS 最值得关注的地方:它真正想回答的,并不是 “如何更聪明地给数据打分”,而是一个更底层的问题 ——在现代优化器主导的训练几何里,什么样的样本,才能带来真实有效的更新?当 “数据墙” 逼近、算力成本高企,预训练已经不再只是 “堆更多数据就能赢” 的游戏,而进入了一个必须精打细算的阶段:每一个 token,都要为更新负责。
?2 }2 k& D+ {/ i/ w2 ?! x) b- s而 OPUS 给出的路线非常清晰,也很有启发性:4 }& C- n5 J( M, Y: R* y
数据选择不该再做优化器无关的旁观者,而应成为与优化器同几何、同方向的在线控制器。
/ G& t- M1 l- ]+ O只有这样,我们才有机会真正榨出 token 的边际收益,把预训练从 “数据吞吐战”,推进到 “更新效率战”。" ]+ N3 Q( K- ]7 w9 H
作者介绍:: {$ V6 C6 }# r& R
本文第一作者为王少博(上海交大 / 阿里 Qwen)、共同第一作者为欧阳轩(UW-Madison)、徐天一(UW–Madison)。通讯作者包括任星彰(阿里 Qwen)、刘大一恒(阿里 Qwen)与张林峰(上海交大)。其余合作者来自阿里、上交、UIUC、Mila 等单位。 |