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

只靠国产算力预训练,稳!全流程开源,「开元」盛世真来了

0
分享至

新智元报道

编辑:LRST

【新智元导读】当算力受限、数据参差,如何训练出高效可用的端侧模型?鹏城实验室联合清华不仅给出了答案,更将每一步都开源共享。

如果实验室只有国产算力、有限资源与开源数据,能否玩转端侧模型的预训练?

鹏城实验室与清华大学PACMAN实验室联合发布了鹏城脑海‑2.1‑开元‑2B(PCMind‑2.1‑Kaiyuan‑2B,简称开元‑2B)模型,并以全流程开源的方式回应了这一挑战——从训练数据、数据处理框架、训练框架、完整技术报告到最终模型权重,全部开源。

模型链接:https://huggingface.co/thu-pacman/PCMind-2.1-Kaiyuan-2B

数据集:https://huggingface.co/datasets/thu-pacman/PCMind-2.1-Kaiyuan-2B

技术报告:https://arxiv.org/abs/2512.07612

数据处理框架:https://github.com/thu-pacman/Kaiyuan-Spark

训练框架:https://github.com/thu-pacman/kaiyuan-mindformers

此次开源不仅为国内研究者提供了一条可复现、可迭代的技术路径,也展现了在国产计算平台上实现高效、稳定预训练的完整解决方案。

开元‑2B的训练依托鹏城脑海 2计算平台,数据处理基于华为鲲鹏920,训练系统基于华为昇腾 910A

由于昇腾910A仅支持FP16精度(类似于NVIDIA V100),训练稳定性成为首要挑战。

为此,研究团队在 Qwen3‑1.7B 的架构基础上,引入了三明治范数(Sandwich Norm)软裁剪(Soft Clipping)两项关键技术。

三明治范数在Transformer每层前后保留归一化层,有效控制梯度传播中的数值范围

软裁剪通过对输出logits应用tanh非线性变换,将数值稳定在合理区间内,防止训练发散

加上三明治范数和软裁剪前后的数值稳定性对比

这些技术不仅使训练在FP16环境下保持稳定,也为后续国产平台上的大规模训练提供了重要参考。

开源数据

去重、评价与高效利用

当前开源预训练数据规模已达TB级别,来源多样、质量不均,如何从中筛选出高质量部分并制定混合策略,是训练优质模型的基础。

开元‑2B面对两个核心问题:

  1. 如何从海量数据中筛选高质量子集?

  2. 如何在不同数据集之间建立统一的质量评价标准?


全局去重的高效实现:Kaiyuan‑Spark框架

研究团队开发了Kaiyuan‑Spark数据处理框架,采用树状流水线设计,全程YAML配置管理,易于复现与扩展。

结合诸葛弩计算框架进行本地加速,在 MinHash 去重任务中实现端到端2.5倍的加速比,高效完成TB级数据的全局模糊去重。

分位标定(Quantile Benchmarking)

为解决数据集之间质量标签不可比的问题,团队提出分位标定:通过设计小规模探针实验,构建数据质量分数与下游任务表现之间的映射关系。

分位标定流程示意图

具体做法为:从不同数据集中按质量分位数抽取多个子集,然后使用小规模模型进行训练测试,最后根据下游任务表现反向标定各数据集的质量区间。

该方法成本低、可迁移,为后续大规模训练的数据配比提供了科学依据。

关键发现

基于分位标定,团队得出若干有趣发现:

数据集的优势与任务类型相关

Fineweb‑Edu在知识问答与阅读理解类任务上表现更好,DCLM‑Baseline 在常识推理类任务中略有优势。

DCLM Baseline和Fineweb-Edu在MMLU和WinoGrande上的分位标定结果

同一数据集内部质量差异巨大

最高与最低质量数据在ARC‑Easy上的表现相差可达8%‑15%

DCLM Baseline和Fineweb-Edu在ARC-Easy上的分位标定结果

可以看到,高质量数据占比虽小,却是提升模型能力的关键。

数据利用策略

动态调整与课程学习

针对数据分布不均的问题,开元‑2B提出三个策略。

多阶段动态比例调整

随着训练推进,逐步提升数学、代码、中文等领域数据的比例,缓解模型遗忘,使关键领域在训练后期得到充分学习。

多阶段调整数据比例

策略性数据重复

小规模实验表明,对高质量数据进行适度重复训练,效果优于单轮训练。开元‑2B在训练后期逐步提高高质量数据的重复比例。

以Fineweb-Edu为例:Top 10%的高质量数据在整个训练过程中出现多次,中低质量数据仅出现一次,从而优化数据效用分布。

Fineweb-Edu在多阶段训练中采样比例变化

多领域混合课程学习

研究团队提出Curriculum Decay Model Average,在课程学习的基础上引入权重平均,缓解训练噪声,提升收敛稳定性。

论文链接:https://arxiv.org/abs/2511.18903

该方法通过领域内质量排序领域间比例均衡的结合,实现多领域数据的渐进式学习。

学习率衰减会影响课程学习效果

不同学习率衰减条件下,最优的课程学习+权重平均配置(Ascend+EMA)要强于最优的常见做法(Uniform+WSD)

多领域课程的实现

配合精心设计的学习率调度方案,模型在高质量数据上得以充分学习,最终在多类评测中接近千问系列(仅开源权重)的模型表现。

全流程开源

推动国产AI生态共建

开元‑2B不仅是一次端侧模型训练的实践,更是一次全栈开源、全流程透明的技术示范。

从数据处理框架Kaiyuan‑Spark、训练数据集,到数据评价体系、课程学习策略,所有环节均公开可复现。训练中涉及的所有原始数据集均具有宽松的开源协议(如CC、Apache、MIT、ODC等,详细列表技术报告附录B),个人、院校、企业均可自由使用,进一步提升了训练的可复现性。

注:目前部分声称「开放」的公开数据集,事实上本身并非使用自由许可证授权,或者间接混入了带有非自由许可证的原始数据;另有大量的公开数据集,对来源数据的许可证未加任何筛选或说明。在模型训练中使用此类数据集并发布或使用,都可能带来潜在的法律合规风险。

这为国内研究者在国产算力平台上开展模型预训练提供了完整工具箱,也为构建开放、协作的 AI 研发生态迈出坚实一步。

模型权重、技术报告与相关代码已在官方平台发布,欢迎开发者、研究者共同参与测试、改进与拓展。

结语

真正的技术进步源自开放的协作与共享,开元‑2B是一个起点,未来与社区一起,在国产算力的土壤上,生长出更多创新的AI成果,从开元系列,开启国产算力训练的「开元盛世」

附录

左右滑动查看

参考资料:

秒追ASI

⭐点赞、转发、在看一键三连⭐

点亮星标,锁定新智元极速推送!

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

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.

相关推荐
热点推荐
“套路加盟”骗取300余人5400余万元,四人获刑

“套路加盟”骗取300余人5400余万元,四人获刑

新京报
2026-08-12 14:32:26
瑞典大满贯,国乒日韩全胜,中国小将轰出11-0,陈熠对手出炉

瑞典大满贯,国乒日韩全胜,中国小将轰出11-0,陈熠对手出炉

南海浪花
2026-08-12 01:08:50
A股CPO大爆发!300亿大牛股,20CM直线封板涨停

A股CPO大爆发!300亿大牛股,20CM直线封板涨停

21世纪经济报道
2026-08-12 17:28:20
泽连斯基谈重点远程打击俄内地石化企业

泽连斯基谈重点远程打击俄内地石化企业

大国老记老顾
2026-08-12 19:34:39
60岁生下双胞胎女儿的盛海琳已76岁,女儿亭亭玉立,让人赞叹不已

60岁生下双胞胎女儿的盛海琳已76岁,女儿亭亭玉立,让人赞叹不已

文刀贰
2026-08-10 22:44:04
“想一巴掌呼死他!”小男孩在海底捞因为蛋糕被分大哭,妈妈也是够爱了!

“想一巴掌呼死他!”小男孩在海底捞因为蛋糕被分大哭,妈妈也是够爱了!

林林先生
2026-08-11 12:00:03
女孩卧铺车求救武警,战士转头装睡,4小时后所有人都愣住了

女孩卧铺车求救武警,战士转头装睡,4小时后所有人都愣住了

萧矹影视解说
2026-04-15 13:08:16
比伊朗导弹还致命,5000名美航母兵对外求救,马上就要崩溃了

比伊朗导弹还致命,5000名美航母兵对外求救,马上就要崩溃了

究竟谁主沉浮
2026-08-10 20:24:31
无花果被发现!研究发现:吃得越多,糖尿病患者血管或越干净?

无花果被发现!研究发现:吃得越多,糖尿病患者血管或越干净?

高中医健康说
2026-08-11 15:16:03
油价大跌!180°大变,8月12日全国92,95汽油预跌“3连减、减至290元/吨”,92汽油“直降”近0.23元/升,下次8月14日调价或“再降”!

油价大跌!180°大变,8月12日全国92,95汽油预跌“3连减、减至290元/吨”,92汽油“直降”近0.23元/升,下次8月14日调价或“再降”!

猪友巴巴
2026-08-12 09:14:52
程梦圆遗体刚打捞上来,恶心一幕发生,救援队长:有些人太没底线

程梦圆遗体刚打捞上来,恶心一幕发生,救援队长:有些人太没底线

青梅侃史啊
2026-08-12 19:57:27
榨干中国红利,却忘恩负义投美!商务部雷霆出击,被查大快人心!

榨干中国红利,却忘恩负义投美!商务部雷霆出击,被查大快人心!

莹莹的历史说
2026-08-11 10:17:05
江西奶奶整天打麻将,3岁女童单独在家被电死,事后儿子煮了4碗面

江西奶奶整天打麻将,3岁女童单独在家被电死,事后儿子煮了4碗面

红豆讲堂
2024-09-13 09:46:33
小米SU7 Ultra双风洞被2只猫霸占 网友戏称60万内最好的猫窝

小米SU7 Ultra双风洞被2只猫霸占 网友戏称60万内最好的猫窝

快科技
2026-08-11 11:21:17
天赐良机!巴萨松口放人!阿森纳挖角 7000 万世界级全能王牌

天赐良机!巴萨松口放人!阿森纳挖角 7000 万世界级全能王牌

奶盖熊本熊
2026-08-12 06:32:41
王菲给女儿拎包好温柔!20岁李嫣穿透视衣背1.8W包,又拽又美真飒

王菲给女儿拎包好温柔!20岁李嫣穿透视衣背1.8W包,又拽又美真飒

锅锅爱历史
2026-08-11 15:27:10
我们又赢了!世锦赛,中国女排5战全胜,不失一局晋级淘汰赛,冲击冠军

我们又赢了!世锦赛,中国女排5战全胜,不失一局晋级淘汰赛,冲击冠军

锐评利物浦
2026-08-12 09:55:51
第二批次054B有48部垂发?这不是重点,重要是的054B搭载的直20F

第二批次054B有48部垂发?这不是重点,重要是的054B搭载的直20F

锋芒点兵
2026-08-12 20:46:02
基辅传出大消息!普京想不到:泽连斯基对中国的态度变得这么快!

基辅传出大消息!普京想不到:泽连斯基对中国的态度变得这么快!

杰丝聊古今
2026-08-12 12:59:26
胃部有癌,嘴巴先知!嘴上出现4种异常,胃癌或许已经悄然临近

胃部有癌,嘴巴先知!嘴上出现4种异常,胃癌或许已经悄然临近

熊猫医学社
2026-08-12 11:35:03
2026-08-12 21:43:00
新智元 incentive-icons
新智元
AI产业主平台领航智能+时代
15925文章数 66999关注度
往期回顾 全部

科技要闻

Manus第二季,浪子回头

头条要闻

赖清德"逃跑路线" 网友逐帧拆解用街景地图"拼"出来了

头条要闻

赖清德"逃跑路线" 网友逐帧拆解用街景地图"拼"出来了

体育要闻

杜兰特,所谓的“联盟小王”

娱乐要闻

老戏骨杨昆两夺金鹰奖,因物业费被告

财经要闻

李嘉诚,再一次大撤退!他嗅到了什么?

汽车要闻

这台大众不一般 上汽大众ID.ERA.5S综合续航约2000公里

态度原创

家居
艺术
旅游
游戏
军事航空

家居要闻

2026建博会(广州) 公装联探展交流活动

艺术要闻

荒废的公共广场,现在请来英国鬼才团队翻盘!

旅游要闻

人生第一次去西藏,我发现西方媒体在撒谎

光荣特库摩又一美少女格斗复活!登陆PS5与NS2

军事要闻

通过“毕业大考” 湖北舰“持证上岗”

无障碍浏览 进入关怀版