IT之家 2024年08月29日
最强端侧开源 AI 模型 Zamba2-mini 登场:12 亿参数,4bit 量化下内存占用小于 700MB
index_new5.html
../../../zaker_core/zaker_tpl_static/wap/tpl_guoji1.html

 

Zyphra 推出 Zamba2-mini 1.2B 模型,参数 12 亿,端侧 SOTA 小语言模型,性能卓越,内存占用低,架构优化,数据集庞大且优质,将开源。

🎯Zamba2-mini 1.2B 是端侧 SOTA 小语言模型,虽尺寸小但性能强,可媲美谷歌、Huggingface、苹果、微软等的更大模型,在推理任务中表现出色,首次令牌时间短,内存占用少。

💻该模型通过高度优化的架构实现,融合不同神经网络设计优势,能保持高质量输出并以小模型的计算和内存效率运行,还集成了两个共享注意力层,增强信息保持能力,提高整体性能。

📚Zamba2-mini 1.2B 在包含三万亿个 token 的海量数据集上预训练,该数据集经过严格过滤和重复处理,在退火阶段进一步完善,包括在 1000 亿个高质量 tokens 上训练。

🌐Zyphra 承诺使 Zamba2-mini 成为 Apache 2.0 许可下的开源模型,促进技术共享和发展。

IT之家 8 月 29 日消息,Zyphra 于 8 月 27 日发布博文,宣布推出 Zamba2-mini 1.2B 模型,共有 12 亿参数,声称是端侧 SOTA 小语言模型,在 4bit 量化下内存占用低于 700MB。

IT之家注:SOTA 全称为 state-of-the-art,并不是特指某个具体的模型,而是指在该项研究任务中,目前最好 / 最先进的模型。

Zamba2-mini 1.2B 虽然尺寸不大,但媲美包括谷歌 Gemma-2B、Huggingface 的 SmolLM-1.7B、苹果的 OpenELM-1.1B 和微软的 Phi-1.5 等更大模型。

在推理任务中,Zamba2-mini 的卓越性能尤为显著,与 Phi3-3.8B 等模型相比,Zamba2-mini 的首次令牌时间(从输入到输出第一个 token 的延迟)是此前的二分之一,内存占用减少了 27%。

Zamba2-mini 1.2B 主要通过高度优化的架构实现,融合了不同神经网络设计的优势,既能保持大型密集变压器的高质量输出,又能以更小模型的计算和内存效率运行。

与前代 Zamba1 相比,Zamba2-mini 的关键进步之一是集成了两个共享注意力层(attention layers)。

这种双层方法增强了模型在不同深度保持信息的能力,从而提高了整体性能。在共享注意力层中加入旋转位置嵌入也略微提高了性能,这表明 Zyphra 致力于在模型设计中进行渐进而有影响力的改进。

Zamba2-mini 是在一个包含三万亿个 token 的海量数据集上进行预训练的,该数据集来自 Zyda 和其他公开来源。

这个庞大的数据集经过了严格的过滤和重复处理,以确保获得最高质量的训练数据,并在退火(annealing)阶段对其进一步完善,其中包括在 1000 亿个质量极高的 tokens 上进行训练。

Zyphra 已承诺使 Zamba2-mini 成为 Apache 2.0 许可下的开源模型。

IT之家附上参考地址

Fish AI Reader

Fish AI Reader

AI辅助创作,多种专业模板,深度分析,高质量内容生成。从观点提取到深度思考,FishAI为您提供全方位的创作支持。新版本引入自定义参数,让您的创作更加个性化和精准。

FishAI

FishAI

鱼阅,AI 时代的下一个智能信息助手,助你摆脱信息焦虑

联系邮箱 441953276@qq.com

相关标签

Zamba2-mini 1.2B 端侧语言模型 性能卓越 开源模型
相关文章