第2章:AI芯片基础:AI芯片的分类与关键指标

各位同学,今天我们来聊聊AI芯片。说实话,这个领域变化太快了,我每年都要更新一次自己的知识库。不过别担心,底层逻辑是不变的——搞懂了分类和指标,你就能看懂市面上任何一款AI芯片。

2.1 AI芯片的四大分类

AI芯片说白了,就是专门为神经网络计算设计的硬件。目前主流的有四种:GPU、FPGA、ASIC、NPU。我一个个讲,你一个个记。

2.1.1 GPU(图形处理器)

GPU最早是给游戏用的,后来发现它做矩阵运算特别快。为什么?因为它有上千个计算核心,天生适合并行计算。

我在2016年做第一个AI项目时,用的就是NVIDIA的GTX 1080。那时候训练一个图像分类模型,要跑三天三夜。现在想想,真是原始社会。

GPU的优点:

  • 生态成熟,CUDA、cuDNN这些库太好用了
  • 通用性强,什么模型都能跑
  • 开发门槛低,Python调库就行

GPU的缺点:

  • 功耗高,一块卡动不动300W+
  • 价格贵,A100要十几万
  • 不适合边缘端部署

2.1.2 FPGA(现场可编程门阵列)

FPGA是个有意思的东西。你可以把它理解成一块「乐高积木」——想搭什么电路就搭什么电路。

我有个朋友做工业视觉检测,一开始用GPU,功耗压不住。后来换成Xilinx的FPGA,功耗降了80%,延迟从50ms降到5ms。嗯,这就是FPGA的魅力。

FPGA的特点:

  • 可重构,灵活性高
  • 延迟极低,适合实时处理
  • 功耗比GPU低很多
  • 但开发难度大,要用Verilog/VHDL
我的建议:如果你做原型验证,或者产品量不大(千台级别),FPGA是个好选择。但要是量产上万台,还是考虑ASIC吧。

2.1.3 ASIC(专用集成电路)

ASIC就是「定制芯片」。你想想看,GPU是万能的,ASIC是专能的。专能意味着什么?效率更高、功耗更低、成本更低(量大的话)。

Google的TPU就是典型的ASIC。我2018年去Google总部参观时,看到他们机房里密密麻麻的TPU,那个震撼啊。一块TPU v4的算力,顶得上几十块GPU。

ASIC的优缺点:

  • 性能功耗比最优
  • 量产成本低(百万片级别)
  • 但开发周期长(12-18个月)
  • 一次性工程费用高(几千万起步)
注意:ASIC一旦流片,就不能改了。我曾经见过一个团队,算法还没定型就急着做ASIC,结果流片回来发现算法改了,几千万打了水漂。所以,先拿FPGA验证,再决定要不要做ASIC。

2.1.4 NPU(神经网络处理器)

NPU是ASIC的一个子类,专门为神经网络设计的。它跟通用ASIC的区别在于:NPU有专门的硬件单元来加速卷积、池化、激活函数这些操作。

华为的昇腾、寒武纪的思元,都是NPU。我去年用昇腾310做过一个边缘AI盒子,8W功耗就能跑ResNet-50,帧率30fps。换成GPU,功耗至少翻三倍。

2.2 关键指标:TOPS、功耗、精度

选芯片不能光看参数,但参数是基础。我一般看三个指标:算力、功耗、精度。

2.2.1 TOPS(万亿次操作每秒)

TOPS是衡量AI芯片算力的标准单位。1 TOPS = 每秒一万亿次操作。

但这里有个坑:不同厂商算TOPS的方式不一样。有的算MAC操作(乘加运算),有的算OP(操作)。我建议你统一看INT8的TOPS,因为边缘端大部分模型都用INT8量化。

芯片 INT8 TOPS 功耗 适用场景
NVIDIA Jetson Orin 275 15-60W 边缘服务器
华为昇腾310 16 8W 智能摄像头
Google Edge TPU 4 2W IoT设备

2.2.2 功耗

功耗是边缘AI的命门。你想想看,一个摄像头模组,散热条件就那么差,功耗超过10W就得加风扇,加了风扇就容易坏。

我做过一个项目,客户要求整机功耗不超过5W。我们试了五六款芯片,最后选了瑞芯微的RK3588,4W功耗跑通了人脸识别。嗯,选芯片就像找对象,合适最重要。

经验之谈:选芯片时,别只看峰值功耗。要看典型功耗,也就是跑实际模型时的功耗。有些芯片峰值功耗很低,但一跑模型就发热降频,实际性能大打折扣。

2.2.3 精度

精度这块,我踩过坑。以前觉得INT8精度够用了,结果做医疗影像分析时,模型精度从FP32的98%掉到了INT8的92%。6个百分点的差距,在医疗领域就是大问题。

所以我的建议是:

  • 对精度要求高的场景(医疗、金融),用FP16或FP32
  • 对功耗敏感的场景(IoT、摄像头),用INT8
  • 实在不行,用混合精度——关键层用FP16,非关键层用INT8

2.3 主流芯片厂商介绍

最后聊聊厂商。我按自己的经验,把主流厂商分成三个梯队:

第一梯队:NVIDIA

生态无敌,CUDA、TensorRT、DeepStream,一套组合拳下来,开发效率极高。但价格贵,功耗高。适合做边缘服务器,不适合做端侧设备。

第二梯队:华为、高通、联发科

华为的昇腾系列,在安防、交通领域用得很多。高通的骁龙系列,手机端AI的王者。联发科的Genio系列,性价比高,适合消费类产品。

第三梯队:寒武纪、地平线、瑞芯微

这些国产厂商,在特定领域很有竞争力。寒武纪的思元系列,适合云端推理。地平线的征程系列,主打自动驾驶。瑞芯微的RK系列,性价比极高,适合小批量产品。

我的选型原则:先看生态,再看性能,最后看价格。生态好的芯片,开发周期能缩短一半。别为了省几百块钱,多花几个月做适配。

好了,这一章就讲到这里。下一章我们聊聊AI芯片的硬件架构,看看这些芯片内部到底是怎么工作的。