
想象一下,在一个极其繁忙的超级厨房里,有一位切菜速度快如闪电的顶级大厨(CPU/GPU)。然而,这个厨房的布局有个致命缺陷:食材仓库(内存)建在离厨房几公里外的地方。结果就是,大厨绝大部分的时间和体力,都浪费在了去仓库来回搬运白菜和土豆的路上。
这并不是什么荒诞喜剧,而是现代计算机正在上演的真实困境。随着AI大模型时代的到来,这种”把数据搬来搬去”的传统模式已经濒临崩溃。今天,我们就来深度硬核科普一项试图彻底重构计算机大脑的前沿技术——存算一体(Compute-in-Memory)。
冯诺依曼架构的隐痛:一堵难以逾越的”内存墙”
自1945年以来,几乎所有的现代计算机都在运行冯诺依曼架构。这种架构将处理单元(CPU或GPU)与存储单元(内存)严格分离,指令和数据需要共享同一条总线进行传输。
这种设计在过去几十年非常成功,但它也带来了一个致命后果:内存墙(Memory Wall)。过去40年里,处理器的算力飙升,而内存的带宽和延迟改善却远远跟不上步伐——过去二十年,内存容量提升了128倍,带宽提升了20倍,但延迟仅仅改善了30%。处理器被迫在空转中浪费大量时钟周期,苦苦等待数据进出内存。
更令人窒息的是能耗。就像拓宽高速公路并不能彻底解决堵车一样,在芯片和内存之间穿梭数据,会产生极高的电能损耗。访问芯片外内存的能耗,是一次浮点运算能耗的200倍。在谷歌TPU的能耗中,高达80%的电量是花在电信号传输上,而不是用于逻辑计算。在一些系统中,单单DRAM(动态随机存取存储器)就占到了系统总功耗的40%。数据搬运,正在榨干数据中心的电力。
AI大模型时代:算力狂飙下的访存窒息
如果说过去”内存墙”只是隐隐作痛,那么深度学习的爆发则让它变成了重症。
从2012年至今,行业顶级AI模型的规模增长了数十万倍。从GPT-3的1750亿参数,到如今动辄万亿参数的巨型模型,AI硬件的内存容量和读取能力被逼到了极限。
运行神经网络,本质上就是进行海量的矩阵乘法运算(MAC,乘加运算)。在这个过程中,大部分的时间和能量都消耗在从内存中提取权重数据上,而不是真正的数学计算上。以一个万亿参数的模型为例,即便使用最先进的并行技术,估计也需要320张拥有80GB内存的英伟达A100 GPU才能装下。按每张卡3.2万美元计算,光是硬件采购成本就高达1000万美元。如果不能打破冯诺依曼瓶颈,先进的AI将永远是极少数超级科技巨头或政府才玩得起的”烧钱游戏”。
存算一体的三路出击:SRAM、DRAM与RRAM
为了推翻这堵”内存墙”,科学家们提出了一个大胆的构想:既然搬运数据太慢太贵,为什么不干脆让内存自己做计算?
这并不是一个新概念,早在20世纪60年代斯坦福大学教授就提出过”内存逻辑”,但受限于当时内存和逻辑芯片难以融合的制造工艺(逻辑工艺追求速度,内存工艺追求高密度和低漏电,两者水火不容),这一理念长期沉寂。直到今天,面对摩尔定律的放缓,业界开始在三个层面上发起猛攻:
- 系统级(近存计算): 这是目前最主流的折中方案。利用2.5D或3D芯片堆叠封装技术,将多个DRAM内存裸片直接堆叠在CPU之上。它们之间通过成千上万条被称为”硅通孔(TSVs)”的微小通道连接,从而获得海量的内部带宽。
- 电路级(SRAM/DRAM存内计算): 科学家通过修改外围电路,让计算直接在SRAM或DRAM存储阵列内部(in situ)发生。例如,微软、英伟达等机构提出的Ambit加速器架构,通过同时激活DRAM内部的三个字线行(两行作为输入,一行作为输出),直接在内存子阵列中实现AND(与)和OR(或)等基本逻辑运算。
- 器件级(新型非易失性存储): 这一层寄希望于完全不同于电容原理的新型存储硬件,最典型的是RRAM(阻变存储器)。传统DRAM靠电容器里的电荷存数据,而RRAM通过改变材料的电阻状态来存储信息。这种特殊的物理结构使得RRAM能够直接在存储单元内计算逻辑函数。
模拟计算复兴:当矩阵乘法遇见存储阵列
存算一体最激动人心的突破,在于它唤醒了一个沉睡数百年的古老技术——模拟计算(Analog Computing)。
几十年来,我们习惯了数字计算机里绝对精确的”0″和”1″。但AI神经网络其实根本不需要那么高的精度——AI有96%还是98%的把握认出一只鸡,那都是一只鸡。这种对误差的容忍度,让模拟计算迎来了完美的复出时机。
在模拟计算存算一体架构中,矩阵乘法变成了奇妙的初中物理题:
初创公司Mythic AI巧妙地将数字闪存单元(Flash cells)当作可变电阻来使用。他们将神经网络的权重写入存储单元,表现为通道的电导(电阻的倒数)。然后,将输入数据(激活值)转化为电压输入。根据欧姆定律(I = V × G),产生的电流自然就是电压与电导的乘积,也就是”激活值×权重”。
更绝的是,只要把这些存储单元连在同一条线上,各个单元产生的电流就会在物理线路上自动汇加在一起。瞬间,庞大的矩阵乘法(乘加运算)就在读取存储器的一瞬间,靠着电流的物理定律自动完成了!没有数据的来回搬运,没有0和1的繁琐转换,只有极其暴力的物理直出。
破局与产业化:台积电、三星与初创公司的角逐
当前,这项革命性技术正在加速从实验室走向商业化:
- 初创公司的突围: 像Mythic AI这样的公司已经推出了基于模拟闪存阵列的AI芯片。这款小小的芯片能以仅仅3瓦的功耗,实现每秒25万亿次(25 TOPS)的数学运算!相比之下,完成同等任务的传统数字系统可能要喷出50到100瓦的惊人热量,并且体积庞大。
- 晶圆代工巨头的基石: 在近存计算(系统级)方面,芯片代工巨头台积电(TSMC)正在大显身手。例如,AMD为其处理器增加缓存的”3D V-Cache”技术,正是建立在台积电先进的3D堆叠封装技术之上的。这为未来将数百GB内存直接与AI ASIC芯片深度融合铺平了道路。在存储领域,三星也正凭借HBM-PIM等近存计算技术构筑大厂的护城河。
总结与展望:终点还是起点?
真正的智能,或许并不在纯粹的数字世界里。人类的大脑也是计算精度较低,但却将记忆存储、计算和通信紧密融合在一起的终极”存算一体”系统。
过去50年,我们把音乐、图像和视频统统数字化,认为数字计算就是信息处理的终点。但也许在100年后,我们会发现数字计算机只是一个起点。面对依然强大的英伟达GPU和固若金汤的冯诺依曼生态,存算一体芯片目前主要在边缘计算、AR/VR等低功耗场景展露头角。
留给我们的开放问题是: 随着摩尔定律逼近原子极限,在未来那个需要极其庞大算力支撑的”超级通用人工智能(AGI)”时代,模拟存算一体技术,究竟是会仅仅作为数据中心里的边缘协处理器,还是会掀起一场颠覆性的革命,彻底终结数字GPU的霸权?