本文目录导读:

这是一个很有深度的问题,NPU(神经网络处理单元)的架构正处于快速演进期,核心目标是在更低的功耗和延迟下,高效处理Transformer等大模型,同时兼顾传统CNN(卷积神经网络)任务。
主流的NPU新架构趋势可以从宏观架构设计和微观计算单元两个层面来看:
宏观架构层面的核心变革:从“计算中心”到“数据流中心”
传统的NPU(如早期手机SoC中的)主要针对CNN设计,采用脉动阵列(Systolic Array),但Transformer模型带来了巨大的挑战:
- 大模型与“内存墙”: 模型参数动辄数十亿,对片外带宽(DDR/HBM)和片上缓存(SRAM)提出了极高要求。
- 复杂的注意力机制: 需要灵活的非线性计算和数据搬运,脉动阵列的固定数据流模式效率不高。
新架构的宏观设计发生了几大变化:
-
存内计算(In-Memory Computing, IMC):
- 思想: 直接在存储单元(如SRAM、RRAM、MRAM)内完成矩阵乘法运算,而不是把数据搬来搬去。
- 优势: 极大减少“冯·诺依曼瓶颈”导致的数据搬运功耗和时间。
- 代表: 许多AI芯片初创公司(如国内的知存科技、国外的Mythic)都在采用此技术,三星的HBM-PIM也属于一种3D封装下的存内计算,但目前工艺成熟度和精度控制仍是挑战。
-
近存计算(Near-Memory Computing):
- 思想: 将计算单元(MAC阵列)和存储单元(SRAM)通过先进的3D堆叠技术(如TSV、混合键合)紧密耦合。
- 优势: 比传统分离式设计带宽提升数十倍,功耗大幅下降,是目前更务实的解决方案。
- 代表: AMD的XDNA(源于赛灵思) 架构、NVIDIA的Grace Hopper超芯片、英特尔Meteor Lake的NPU都采用了类似思路。
-
数据流架构(Dataflow Architecture):
- 思想: 不再依赖于中央控制,而是由数据本身触发计算,计算单元按流水线方式排列,数据在单元间“流过”。
- 优势: 硬件利用率高,非常适合Transformer这种层数深、结构相对固定的模型。
- 代表: Google的TPU v1/v4、Groq的LPU(语言处理单元)、Cerebras的晶圆级芯片(WSE-2/3)、以及国内寒武纪的某些系列。
微观计算单元的革新:从“MAC”到“多样算力”
在具体的计算单元(PE,Processing Element)设计上,也开始变得多样化:
-
Transformer专属引擎(Transformer Engine):
- 核心: 对Attention机制中的矩阵乘法(QK.V)进行硬件加速,并引入FP8(8-bit浮点) 作为主要的训练和推理格式,通过硬件自动缩放精度,在保证精度的同时大幅提升计算速度。
- 代表: NVIDIA的Hopper (H100) 和 Blackwell (B200) GPU上的Transformer Engine。
-
可重构计算单元:
- 核心: 单元内集成了乘法器、加法树、以及专用的非线性函数单元(如Softmax、LayerNorm、GELU的近似计算),这些单元可以根据不同的算子需求(卷积、全连接、注意力)动态配置。
- 代表: 华为的达芬奇架构(昇腾系列)、苹果的Neural Engine。
-
稀疏计算支持:
- 核心: 对权重或激活值的稀疏性(大量零值)进行加速,通过硬件检测并跳过零值计算,或者使用结构化稀疏。
- 代表: NVIDIA的Ampere及之后的GPU、高通Snapdragon的Adreno/NPU、英特尔Gaudi系列。
具体芯片和架构实例(截至2025年5月)
以下是当前市场上或学术界最具代表性的新架构:
| 架构/芯片 | 核心创新 | 适用场景 | 关键特点 |
|---|---|---|---|
| NVIDIA Blackwell GPU | Transformer Engine + 第二代稀疏 | 云端训练/推理 | FP8/FP4支持;第五代NVLink (900GB/s);专为万亿参数模型设计。 |
| AMD Ryzen AI (XDNA) | 近存计算 + 自适应数据流 | PC/笔记本端推理 | 基于赛灵思AI引擎,2000 TOPS(目前仅宣传,实际算力在46亿参数模型)。 |
| 英特尔 Meteor Lake/Lunar Lake NPU | 可扩展的数据流架构 | PC端推理 | 支持动态的INT8/FP16计算,低延迟,芯片级内存一致。 |
| Google TPU v5e/v5p | 大规模2D网格数据流 + 矩阵单元 | 云端推理/训练 | 针对Transformer优化,高效的大模型部署。 |
| 高通Snapdragon X Elite / 8 Gen (Hexagon NPU) | 混合精度 + 张量加速器 + 向量加速器 | 移动端/笔记本端推理 | 独立的标量、向量、张量单元,支持INT8/FP16/BF16。 |
| 华为昇腾910B/910C (达芬奇架构V2) | Cube + Vector + Scalar三维指令集 | 云端推理/训练 | 可重构计算单元,强调通用性和高性能。 |
| Groq LPU | 确定性数据流架构 | 云端推理 | 无缓存一致性开销,极低延迟,但DDR带宽是瓶颈。 |
| Cerebras WSE-3 | 晶圆级芯片 + 片内全部SRAM | 云端训练 | 不存在显存墙和通信墙,支持全精度。 |
未来的关键方向
- Tiling & 3D封装: 成为主流,将计算、存储、I/O通过3D方式堆叠,打破带宽瓶颈。
- 混合精度计算: 支持FP8、甚至FP4/INT2,用更低的精度换取更高的吞吐和效率。
- 可编程性增强: 不再仅仅是固定的矩阵乘法器,而是支持Transformer、MoE(混合专家)、扩散模型、Mamba等新型架构的通用加速器。
- 片上网络(NoC): 支持更复杂的片内数据通信,例如针对MoE的门控路由、专家间的灵活通信。
一句话总结: 新一代NPU不再是单纯的“矩阵乘法器”,而是“数据流导向 + 近存/存内计算 + 混合精度 + 可重构算子”的智能数据处理器。 苹果、高通、AMD、英特尔、华为等厂商都在朝这个方向全速前进。