算力需求爆炸:大模型时代对AI芯片的终极拷问

各位同学,咱们开门见山。

大模型这波浪潮,说白了就是一场「算力军备竞赛」。我做了十几年芯片,从早期的CNN加速器一路跟到今天的千亿参数大模型,感受最深的一个字就是——「饿」。模型越做越大,算力永远不够用。

今天这堂课,我想跟你聊聊一个核心问题:
大模型到底把AI芯片逼到了什么份上?

嗯,咱们从三个维度来拆解。


一、算力需求的「三级跳」

先看一组数字。你想想看,2018年的BERT有3.4亿参数,训练一次大概需要几天时间。到了GPT-3,1750亿参数,训练一次要花几个月。再到GPT-4,虽然OpenAI没公开具体参数,但业内普遍认为已经突破万亿级别。

这背后意味着什么?

算力需求不是线性增长,而是指数爆炸。

我给大家算一笔账:

模型 参数量 训练算力需求(PFLOPS-day) 相比上一代增长
BERT-Large 3.4亿 ~0.5 基准
GPT-3 1750亿 ~3,640 ~7,000x
GPT-4(估算) ~1.8万亿 ~100,000+ ~30x

看到了吗?从BERT到GPT-3,算力需求暴涨了7000倍。这还只是训练阶段。推理阶段的压力同样巨大——用户每问一次ChatGPT,背后都是数十亿甚至上千亿参数在跑。

核心结论:大模型时代,算力需求每3-4个月翻一番。摩尔定律?早就追不上了。


二、AI芯片的「三大瓶颈」

算力需求爆炸,芯片厂商当然想接招。但现实很骨感——芯片本身遇到了物理极限。

我个人习惯把问题归纳为三个瓶颈:

1. 存储墙

大模型最要命的问题是什么?参数太多,放不进片上SRAM。

1750亿参数,就算用FP16存,也需要大约350GB。而目前最先进的AI芯片,片上SRAM也就几十MB到几百MB。剩下的怎么办?只能反复去HBM(高带宽内存)里搬数据。

我在项目中遇到过这种情况:芯片的算力明明够用,但90%的时间都花在等数据上。说白了,计算不是瓶颈,搬运才是。

2. 带宽墙

HBM的带宽已经做到很高了——HBM3e能做到819GB/s。但跟大模型的需求比起来,还是杯水车薪。

你想想看,一个1750亿参数的模型,每次推理都要把全部参数从HBM搬到计算单元。就算819GB/s的带宽,光搬一次参数就要0.4秒。而用户期望的响应时间是毫秒级。

这中间的差距,不是靠堆料能解决的。

3. 功耗墙

这是最头疼的。我见过一个数据中心,部署了上万张A100,满负荷运行时,一个月的电费够买几辆跑车。

大模型训练一次,碳排放量相当于5辆汽车终身排放。这还不算推理阶段的持续消耗。

注意:功耗问题不只是成本问题,更是物理散热问题。芯片的功率密度已经逼近核反应堆的水平。再往上堆算力,散热方案会先崩溃。


三、架构层面的「终极拷问」

好,问题摆在这了。那AI芯片该怎么破局?

我个人认为,单纯靠制程工艺进步已经不够了。3nm、2nm,每代提升也就20-30%。但算力需求是10倍、100倍地涨。所以必须从架构层面做根本性变革。

这里我画了一张图,帮你理解当前AI芯片面临的核心矛盾:

大模型时代AI芯片核心矛盾 算力需求 • 千亿参数模型 • 毫秒级响应 • 持续训练迭代 • 低成本部署 供需失衡 芯片供给 • 存储墙限制 • 带宽墙限制 • 功耗墙限制 • 工艺放缓 出路:架构创新 + 系统协同 + 新计算范式

这张图想表达的核心思想是:需求端在指数增长,供给端却在逼近物理极限。中间的鸿沟,必须靠架构创新来填补。


四、几个值得关注的「破局方向」

讲完了问题,咱们聊聊解法。我个人比较看好的几个方向:

  1. 存算一体——把计算搬到存储附近,甚至直接在存储单元里做计算。这能从根本上解决存储墙问题。不过目前工艺还不成熟,精度和寿命都有挑战。
  2. 稀疏计算——大模型里很多参数其实是冗余的。如果能动态跳过那些不重要的计算,算力效率能提升好几倍。我曾经带过一个项目,用稀疏化技术把推理速度提升了4倍,精度只掉了0.1%。
  3. 3D堆叠——把计算层和存储层垂直堆叠在一起,用TSV(硅通孔)互联。这样带宽能提升几个数量级。但散热是个大问题,中间层的热量很难散出去。
  4. 光计算/模拟计算——这个更前沿一些。用光信号代替电信号做矩阵乘法,理论上能实现超低功耗、超高速度。但离商用还有距离。

我的建议:如果你是做芯片设计的,别只盯着算力指标。多想想怎么把数据搬得更快、存得更近、算得更省。大模型时代,系统效率比峰值算力更重要。


五、一个真实的「避坑指南」

最后,分享一个我踩过的坑。

几年前,我们团队设计了一款AI芯片,峰值算力标到了1000TOPS,当时觉得无敌了。结果一跑大模型,实际吞吐量只有理论值的20%。

为什么?

因为计算单元一直在等数据。HBM带宽被多个计算核共享,一旦并发上去,带宽就成瓶颈了。我们当时把算力堆得很高,但存储子系统根本没跟上。

所以,千万别只看算力数字。一个芯片好不好用,要看三个指标:

  • 算力利用率(实际跑模型时能发挥多少)
  • 带宽利用率(HBM带宽能不能喂饱计算单元)
  • 能效比(每瓦能跑多少有效计算)

这三个指标,才是大模型时代衡量AI芯片的真正标尺。


好了,这一章的内容就到这里。算力需求爆炸这件事,既是挑战也是机遇。下一章我们会深入聊聊,现有的AI芯片架构到底有哪些「硬伤」,以及为什么说Transformer正在倒逼芯片设计范式变革。

咱们下节课见。

专注资料整理