李开复旗下AI公司零一万物上海专业的市场调研公司，又一位大模子选手登场：

90亿参数Yi-9B。

它堪称Yi系列中的“理科状元”，“恶补”了代码数学，同期轮廓智商也没落下。

在一系列雷同领域的开源模子（包括Mistral-7B、SOLAR-10.7B、Gemma-7B、DeepSeek-Coder-7B-Base-v1.5等）中，发达最好。

老规则，发布即开源，尤其对确立者友好：

Yi-9B（BF 16） 和其量化版 Yi-9B（Int8）王人能在蹧跶级显卡上部署。

一块RTX 4090、一块RTX 3090就可以。

深度扩增+多阶段增量熟习而成

零一万物的Yi眷属此前依然发布了Yi-6B和Yi-34B系列。

这两者王人是在3.1T token中英文数据上进行的预熟习，Yi-9B则在此基础上，增多了0.8T token持续熟习而成。

数据的限制日历是2023年6月。

来源提到，Yi-9B最大的跨越在于数学和代码，那么这俩智商究竟怎样种植呢？

零一万物先容：

单靠增多数据量并没法达到预期。

靠的是先增多模子大小，在Yi-6B的基础上增至9B，再进行多阶段数据增量熟习。

率先，何如个模子大小增多法？

一个前提是，团队通过分析发现：

Yi-6B熟习得依然很充分，再何如新增更多token练效用可能也不会往上了，是以磋商扩增它的大小。(下图单元不是TB而是B)

何如增？谜底是深度扩增。

零一万物先容：

对原模子进行宽度扩增会带来更多的性能亏空，通过选拔符合的layer对模子进行深度扩增后，新增layer的input/output cosine 越接近1.0，即扩增后的模子性能越能保持原有模子的性能，模子性能亏空幽微。

依照此念念路，零一万物选拔复制Yi-6B相对靠后的16层（12-28 层)，构成了48层的Yi-9B。

本质暴露，这种活动比用Solar-10.7B模子复制中间的16层（8-24层）性能更优。

其次，何如个多阶段熟习法？

谜底是先增多0.4T包含文本和代码的数据，但数据配比与Yi-6B一样。

然后增多另外的0.4T数据，相通包括文本和代码，但重心增多代码和数学数据的比例。

（悟了，就和咱们在大模子发问里的决窍“think step by step”念念路一样）

这两步操作完成后，还没完，上海专业的市场调研公司团队还参考两篇论文（An Empirical Model of Large-Batch Training和Don’t Decay the Learning Rate， Increase the Batch Size）的念念路，优化了调参活动。

即从固定的学习率开动，每当模子loss罢部属降时就增多batch size，使其着落不中断，让模子学习得愈加充分。

最终，Yi-9B本体共包含88亿参数，竣事4k高下文长度。

Yi系列中代码和数学智商最强

实测中，零一万物使用greedy decoding的生成形貌（即每次选拔概率值最大的单词）来进行测试。

参评模子为DeepSeek-Coder、DeepSeek-Math、Mistral-7B、SOLAR-10.7B和Gemma-7B：

6家快递无法做到下单后三小时取件、身份证核对及提供发票

（1）DeepSeek-Coder，来自国内的深度求索公司，其33B的提醒调优版块东谈主类评估杰出GPT-3.5-turbo，7B版人道能则能达到CodeLlama-34B的性能。

DeepSeek-Math则靠7B参数干翻GPT-4，飘荡统统开源社区。

（2）SOLAR-10.7B来自韩国的Upstage AI，2023年12月降生，性能杰出Mixtral-8x7B-Instruct。

（3）Mistral-7B则是首个开源MoE大模子，达到致使杰出了Llama 2 70B和GPT-3.5的水平。

（4）Gemma-7B来自谷歌，零一万物指出：

其有用参数目其实和Yi-9B一个等第。

（两者定名准则不一样，前者只用了Non-Embedding参数，后者用的是一齐参数目并进取取整）

收尾如下。

率先在代码任务上，Yi-9B性能仅次于DeepSeek-Coder-7B，其余四位一齐被KO。

在数学智商上，Yi-9B性能仅次于DeepSeek-Math-7B，杰出其余四位。

轮廓智商也不赖。

其性能在尺寸邻近的开源模子中最好，杰出了其余一齐五位选手。

终末，还测了学问和推贤惠商：

收尾是Yi-9B与Mistral-7B、SOLAR-10.7B和Gemma-7B不相高下。

以及谈话智商，不仅英文可以，中语亦然广受好评：

最终末，看完这些，有网友示意：依然迫不足待想试试了。

还有东谈主则替DeepSeek持了一把汗：

赶快加强你们的“比赛”吧。全面主导地位依然莫得了==

— 完 —上海专业的市场调研公司