网易首页 > 网易号 > 正文 申请入驻

EMNLP 2026 杰出论文 | 大模型训练,数据配比难?日本NICT团队提出最优分布向量合并

0
分享至



不同数据,到底各训多少?每个模型训练师都绕不开这个问题。

训练集中不同数据配比出错,往往只能从头训。每个数据集各放多少入训练集,训练前就得定好;但是配比好不好,却要等训练结束评测后才知道。效果若不理想,可能要调整比例重新训练,试错成本极高。

日本信息通信研究机构(NICT)的宋海越及团队针对这一问题在论文「OptiMer: Optimal Distribution Vector Merging Is Better than Data Mixing for Continual Pre-Training」中提出了合并分布向量的解决方案,让持续预训练中尝试不同配比不必从头训模型。

此论文荣获EMNLP 2026 杰出论文奖(Outstanding Paper Award),是本届 17,661 篇投稿中获此殊荣的 22 篇论文之一。



  • 论文:OptiMer: Optimal Distribution Vector Merging Is Better than Data Mixing for Continual Pre-Training
  • 作者:Haiyue Song, Huy Hien Vu, Masao Utiyama
  • 网站:https://shyyhs.github.io/optimer



方法 | 从合并数据集到合并分布向量

我们提出最优合并,OptiMer。

第一步,分别训练模型,学习各类数据。从基座模型出发,在各个数据集上分别进行持续预训练。例如,在日语,中文和数学数据上,分别各训练一个模型,得到三个模型。

第二步,提取分布向量,记录训练带来的能力变化。将每个训练后模型的参数减去基础模型的参数,可得到对应的分布向量。它代表了此数据上训练后模型发生的变化。

第三步,自动搜索,寻找最优合并比例。基础模型权重,加上不同比例的多个分布向量,外加指令遵循向量,可以得到一个新模型。根据其在验证集上的表现,我们可以不断优化分布向量的比例,达到最优合并。每次尝试只包含模型合并和评估,无需再训练。



效率 | 提升 15~35 倍

完成各个数据集的初始训练后尝试不同配比,传统数据混合方法仅训练一个配比就需要 129.3 小时,而 OptiMer 尝试 100 次配比仅需 8.5 小时。实验中我们发现在配比最优化阶段,OptiMer 实现了15~35 倍加速。



效果 | 优于混合数据训练

OptiMer 在 C-Eval,Japanese Leaderboard,MMLU,GSM8K,HumanEval 等常用基准测试上平均得分达到 70.37,高于传统数据混合训练 63.71,提升约 6.7 分。

我们还发现,OptiMer 找到的最优合并权重,可以用来指导数据混合方案中的数据配比。将合并权重直接用做数据混合的比例,再重新训练模型,平均得分从等比例混合的 63.71 提升至 68.66。



灵活 | 对新任务进行快速再优化

对于新任务,传统数据混合方法需要重新配比,重新训练。此过程成本极高,导致传统方法无法快速适配新任务。而我们的方法可以复用已经训练好的分布向量,仅需更换评估目标,自动调整合并权重即可。比如,我们针对英文任务进行优化,可以得到在对应任务上的最高分;针对数学再优化,可以在数学任务上表现不错;针对日语任务进行再优化,可以快速得到日语特化模型。

这种情况在工业部署上相当常见,如果某公司想做开发不同领域的特化模型,比如教育,金融模型等,传统方法在训练数据配比上需要大量试错,而我们的方法可以自动快速产出特化模型。



分析 1 | 为什么分布向量可以合并

从不同数据学到的能力,在合并时会不会互相干扰?我们计算了每两个分布向量之间的余弦相似度。发现分布向量和指令遵循向量之间则仅约为 0.03,接近垂直;不同分布向量之间的余弦相似度约为 0.3。我们推测,现有的合并算法比如 DARE 可以较无损地将不同向量合并。



分析 2 | 分布向量的主成分分析

我们通过主成分分析,将高维的分布向量投影到二维平面(因为是两个模型参数的差,所以是和模型参数量一个维度)。首先,不同数据上持续预训练出来的的分布向量和指令遵循向量相距较远。然后,通过 OptiMer 的合并权重柱状图我们发现,OptiMer 倾向给指令遵循向量较大的权重,给数据分布向量较小的权重。这导致了 OptiMer 合并出来的最优模型实际在 Base 模型和 IT 模型的连线的附近(我们可以想象从 PT 到 IT 画一条线)。我们推测优秀的模型应该在此线附近。



分析 3 | OptiMer 的搜索过程

我们观察 OptiMer 如何一步步找到最佳合并权重。左图展示了权重空间,右边展示了搜索过程中分数提升情况。

从左图中可以看到,高分组合集中在一个较窄的区域(在这个区域指令遵循向量的权重较大,数据分布向量的权重较小)。这证明了 OptiMer 中贝叶斯优化相对于网格搜索的优势,网格搜索容易漏搜,特别是在更多个数据集的时候,而 OptiMer 大致只需要搜索 10*n 次,其中 n 是数据集的个数。

从右图中可以看到,OptiMer 中的贝叶斯优化会根据已有数据点,将后续搜索逐渐集中到更有希望的区域。我们使用的算法可以并行搜索,每次搜索 8 个点,前期专注探索整个空间,中期迅速收敛,并在后期找到了本轮搜索中的最高分。



分析 4 | 训练中的向量变化

我们记录了持续预训练过程中不同训练步的模型,来追踪模型的变化。主图展示了这些向量在 PCA 平面上的轨迹,小图展示了向量长度和模型得分随训练步数的变化。

从左图可以看到,持续预训练的轨迹近似一条直线。增加或者减少分布向量的权重,类似于增加或者减少在这个数据集上的有效训练量。

从右图可以看到,随着训练进行,分布向量模长越来越长,模型表现在训练早期达到峰值随后逐渐下降。我们推测,这暗示了模型不能偏离基座模型太远。分析 2 中亦有此推测。



分析 5 | 拓展到负权重

分布向量的权重可以设为负数。我们发现有时甚至会有更好的结果。这代表 OptiMer 能进行机器反学习,选择性遗忘数据中的某些知识,从而在目标任务上取得更好的效果吗?本文并未对此进行进一步探究,但我们认为这是一个非常有趣的问题。



应用 | 构建日语特化模型

我们将 OptiMer 应用在来自 AI Singapore 的 Gemma-SEA-LION-v4-27B 模型,这是一个基于 Gemma 3,在东南亚语言上进行过持续预训练的模型。

我们希望在保留其东南亚语言能力的同时,增强其日语能力。

为此,我们分别在日语、中文、英语、数学和代码五类数据上进行持续预训练,提取分布向量,再与 SEA-LION 的指令遵循向量合并,以日语任务表现为目标搜索合并权重。

最终,日语得分从 66.34 提升至 74.40,同时保留了东南亚语言任务上的表现。



结语

OptiMer 适配开源大模型于目标语言或领域,比传统数据混合方式更效率,效果更好,并且更灵活。

同时,此工作仅在持续预训练上进行了实验,未来我们希望能将这一思路拓展到后训练,解决后训练中数量庞大的数据集的最优配比问题。

第一作者简介

宋海越,现任日本 Preferred Networks 大语言模型开发部后训练组工程师,曾任位于日本京都的国立研究开发法人 情报通信研究机构。师从黑桥祯夫教授,2024 年于京都大学获信息科学博士学位。本科就读于上海交通大学计算机科学系及日语系。研究方向包括语言模型及机器翻译。

本工作为第一作者任职于信息通信研究机构期间完成。

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
已经下台近十年的奥巴马,居然是在世美国总统里年纪最小的

已经下台近十年的奥巴马,居然是在世美国总统里年纪最小的

墨兰史书
2026-11-15 00:05:12
闲鱼惊现大量2TB U盘,全新标价35元!到底有啥猫腻?

闲鱼惊现大量2TB U盘,全新标价35元!到底有啥猫腻?

小柱解说游戏
2026-11-15 16:28:19
大瓜!外网新爆料:Jessica Pin自曝与黄仁勋存在婚外亲密关系,为第三起相关传闻

大瓜!外网新爆料:Jessica Pin自曝与黄仁勋存在婚外亲密关系,为第三起相关传闻

火山詩话
2026-11-12 21:09:10
大家今年都“穷”到什么程度了?网友评论区堪比大型“比惨”现场

大家今年都“穷”到什么程度了?网友评论区堪比大型“比惨”现场

夜深爱杂谈
2026-10-20 20:22:43
伊朗:若美国再对伊朗发动攻击,美战机从哪起飞就打哪

伊朗:若美国再对伊朗发动攻击,美战机从哪起飞就打哪

澎湃新闻
2026-11-13 12:15:01
打了五年,俄罗斯突然广而告之要消灭泽连斯基!

打了五年,俄罗斯突然广而告之要消灭泽连斯基!

山河路口
2026-11-15 13:09:35
A股:已经不必等开盘了,种种迹象非常强烈,下周或将有两大变化

A股:已经不必等开盘了,种种迹象非常强烈,下周或将有两大变化

虎哥闲聊
2026-11-15 08:30:54
吴承瑛现状:全家定居香港,三个女儿三条路,老三已经成为冠军

吴承瑛现状:全家定居香港,三个女儿三条路,老三已经成为冠军

东方不败然多多
2026-11-15 13:24:21
CBA快讯,广东引援新消息,利文斯顿空降广东,爱德华兹留队成功!

CBA快讯,广东引援新消息,利文斯顿空降广东,爱德华兹留队成功!

体坛小快灵
2026-11-15 16:09:34
彰化倒数15天,王惠美一纸动员令,蓝营的观望局散了

彰化倒数15天,王惠美一纸动员令,蓝营的观望局散了

爱意随风起呀
2026-11-15 16:29:40
16岁小将3-2逆转松岛辉空、零封张本智和!日乒或将迎来新格局

16岁小将3-2逆转松岛辉空、零封张本智和!日乒或将迎来新格局

不写散文诗
2026-11-14 19:52:09
黄乐这含情脉脉的桃花眼还有人畜无害的笑容,难怪连黄仁勋都沉沦了

黄乐这含情脉脉的桃花眼还有人畜无害的笑容,难怪连黄仁勋都沉沦了

一盅情怀
2026-11-09 13:07:20
一网民在外网诋毁陈祥榕、刘胡兰等英雄烈士,被判公开道歉并赔偿3万元

一网民在外网诋毁陈祥榕、刘胡兰等英雄烈士,被判公开道歉并赔偿3万元

澎湃新闻
2026-11-15 13:16:08
掘金送鹈鹕6连败:约基奇26+8+10穆雷23+7+9 创历史第一纪录

掘金送鹈鹕6连败:约基奇26+8+10穆雷23+7+9 创历史第一纪录

醉卧浮生
2026-11-15 10:35:13
iPhone Duo上市三周,调查显示30%买家已退货

iPhone Duo上市三周,调查显示30%买家已退货

科技兽
2026-11-12 22:10:27
《伟大的长征》中贺老总讲的“有一支部队去了缅甸”,是真的吗?

《伟大的长征》中贺老总讲的“有一支部队去了缅甸”,是真的吗?

三闲看剧
2026-11-15 13:33:11
马天宇称不愿意养狗,因狗太粘人所以养羊,张泉灵直呼“我现在发现你太神奇了”

马天宇称不愿意养狗,因狗太粘人所以养羊,张泉灵直呼“我现在发现你太神奇了”

韩小娱
2026-11-15 11:21:00
李显龙直言不讳地告诫国民:每一个新加坡人都必须明白,本国坚决不承认“华人国家”身份,背后藏着的是全亚洲最为惊心动魄的生存考量

李显龙直言不讳地告诫国民:每一个新加坡人都必须明白,本国坚决不承认“华人国家”身份,背后藏着的是全亚洲最为惊心动魄的生存考量

橙色书卷
2026-07-16 21:35:45
13.79万元,本田刚发布的新车,有点香啊

13.79万元,本田刚发布的新车,有点香啊

音乐时光的娱乐
2026-11-14 14:28:24
杨钰莹美照欣赏

杨钰莹美照欣赏

喜欢历史的阿繁
2026-11-15 18:05:36
2026-11-15 18:40:49
机器之心Pro incentive-icons
机器之心Pro
专业的人工智能媒体
14478文章数 142758关注度
往期回顾 全部

教育要闻

珠海市金湾区第二小学将出席2026第四届课博会,分享“如家七彩·三位一体”跨学科课程实践

头条要闻

客人要求用百元大钞做“现金花束” 花店直接报警

头条要闻

客人要求用百元大钞做“现金花束” 花店直接报警

体育要闻

马刺破勇士:文班与库里各自的价值

娱乐要闻

陈晓楠对话厉娜和许飞引全网热议

财经要闻

英伟达、字节、谷歌,涌入"救命"的生意

科技要闻

历史头一回!马斯克用"筷子"夹住太空飞船

汽车要闻

更豪华更好开 问界新M8细节升级见真章

态度原创

手机
本地
游戏
亲子
公开课

手机要闻

鞠婧祎代言 荣耀700系列四款高颜值配色公布:全系标配原生AI背屏

本地新闻

乌兰察布:以前种土豆,现在种Token

飞燕群岛:新IP城市街区风格 有大量可自由进出房屋

亲子要闻

一图读懂丨科学照护之亲子游戏

公开课

李玫瑾:为什么性格比能力更重要?

无障碍浏览 进入关怀版