中国首个昇腾原生三值大模型开源:BitCPM-CANN 实现 1.58-bit 端到端训练,8B 模型可跑通旗舰手机

大语言模型5个月前发布 小马良
227 0

在国产算力与大模型高效部署的交汇点上,一项里程碑式的成果正式诞生。面壁智能联合清华大学及 OpenBMB 开源社区,正式发布并开源了 BitCPM-CANN —— 中国首个完全基于**华为昇腾(Ascend)算力平台实现端到端训练并开源的三值(1.58-bit)**大模型系列。

  • HuggingFace:https://huggingface.co/collections/openbmb/bitcpm-cann
  • ModelScope:https://modelscope.cn/collections/OpenBMB/BitCPM-CANN

这一发布不仅标志着低比特大模型训练在国产硬件上的重大突破,更为移动端及边缘侧 AI 应用打开了全新的大门。

中国首个昇腾原生三值大模型开源:BitCPM-CANN 实现 1.58-bit 端到端训练,8B 模型可跑通旗舰手机

核心突破:1.58-bit 的极致效率

BitCPM-CANN 采用了先进的三值量化技术,将模型权重压缩至 1.58-bit。相比传统的 BF16(16-bit)精度,带来了革命性的效率提升:

  • 显存红利释放 6 倍:推理阶段显存占用大幅降低,使得原本需要高性能服务器才能运行的模型,现在可以轻松部署在资源受限的设备上。
  • 手机端侧落地成为现实:官方指出,一个 8B 参数的 BitCPM-CANN 模型,凭借极低的显存需求,已具备在当前主流旗舰手机上流畅运行的能力。
  • 能力保留率高达 90%-97.2%:在与同尺寸 MiniCPM4 全精度家族的逐项对照评测中,BitCPM-CANN 展现了优异的性能,证明极低比特量化并未显著牺牲模型智能。

全栈原生:基于昇腾的完整工程体系

BitCPM-CANN 并非简单的模型移植,而是从底层到上层的全链路原生适配:

  1. 原生量化算子:针对昇腾 NPU 架构定制开发的量化算子,确保计算效率最大化。
  2. 低比特训练算法:优化后的训练算法,稳定支持 1.58-bit 权重的梯度更新与收敛。
  3. 全链路框架适配:基于 MindSpeed × Megatron-LM 主干搭建,包含环境适配、32K 长序列支持、并行策略及融合算子等完整工程体系。

这意味着,未来所有面向昇腾平台的低比特训练工作,都可以建立在 BitCPM-CANN 这套公共基础设施之上,极大降低了开发门槛。

中国首个昇腾原生三值大模型开源:BitCPM-CANN 实现 1.58-bit 端到端训练,8B 模型可跑通旗舰手机

模型家族与开源详情

此次开源涵盖四个尺寸的模型,满足不同场景需求:

模型尺寸参数量适用场景
BitCPM-CANN-0.5B0.5 Billion极端边缘设备、IoT 节点
BitCPM-CANN-1B1 Billion入门级手机端、轻量级助手
BitCPM-CANN-3B3 Billion主流手机端、平板设备
BitCPM-CANN-8B8 Billion旗舰手机、高性能边缘服务器

目前,全系列模型权重已在 Hugging Face 及 ModelScope 等平台开源,开发者可直接下载体验或进行微调。

产业意义:国产算力与端侧 AI 的双赢

BitCPM-CANN 的发布具有双重战略意义:

  • 对于国产算力生态:证明了华为昇腾平台在复杂、前沿的大模型训练任务(如极低比特量化训练)上的成熟度与可用性,增强了开发者信心。
  • 对于端侧 AI 产业:解决了大模型“下手机”的最大瓶颈——显存与算力限制。6 倍的显存红利意味着更多、更强的模型可以离线运行在用户手中,保护隐私的同时提升响应速度。
© 版权声明

相关文章