谷歌AI芯片AI设计!端到端闭环,三星联发科都buy in了
开源预训练版本
明敏 发自 凹非寺
量子位 | 公众号 QbitAI
芯片加速AI发展,AI反过来又推动芯片进步?
刚刚,谷歌芯片设计算法AlphaChip正式亮相。
DeepMind CEO哈萨比斯调侃:这下闭环了。
训练最好的芯片设计算法→用它设计更好的AI芯片→用它们训练更好的模型→再设计更好的芯片→……
要知道,AlphaChip只需几个小时就能生成比肩人类水平的芯片设计图。
谷歌目前最强TPU“Trillium”,AlphaChip就参与了设计,而TPU又训练了Gemini、Imagen、Veo等新算法。
以及联发科、三星等芯片厂商,也都将其用到了实际生产中。
这不,最近处在转型逆风期的英特尔深陷并购风波。于是网友给出主意:
要不谷歌把英特尔买了吧。
谷歌表示AlphaChip现在不仅速度更快,而且需要的计算资源更少,已经达到人类专家级水平。
最新发布中,谷歌还开源了一个预训练版本,这样大家都能用AlphaChip更简易地开发芯片(但也还是自己预训练的效果更好)。
围观群众都相当兴奋:
这真的是超酷的贡献,对于工程领域很重要。
几个小时搞定AI芯片设计
向前追溯来看,AlphaChip最早期的成果于2020年以预印本形式发布,2021年登上Nature。
但直到今天谷歌才给它了一个正式的名字。Jeff Dean是作者之一。
AlphaChip和AlphaGo、AlphaZero的原理相似,都是基于深度强化学习。
AlphaZero掌握了围棋、国际象棋的游戏规则,AlphaChip就是将规划芯片布局视为一个游戏。
一个芯片上往往包含数十亿个晶体管,由他们组成的数千万个逻辑门就是标准单元,此外还有数千个存储块,称为宏块。
芯片设计就是要确定它们的位置,如何布局将影响芯片的处理速度和电源效率。
传统方法中,光是放置宏块这一步就很耗时了,为了给标准单元留出更多空间,每一次迭代都要几天甚至几周时间。
AlphaChip基于深度强化学习,从之前的芯片布局中进行学习,生成新的设计方案。
它将宏和标准单元映射到一个平面画布上,形成具有数百万到数十亿节点的“芯片网表”。
从一个空白格开始,一次放置一个电路元件,直到放置完所有元件。算法会对功率、性能和面积(PPA)等进行优化,并输出概率分布。最终将根据布局质量计算奖励。
利用一种新型的基于边缘(edge-based)的图神经网络AlphaChip可以学习互联芯片组件之间的关系,并应用在芯片之间,所以AlphaChip能对设计的每个布局进行改进。
下图中,左图为零样本下AlphaChip布局开源Ariane RISC-V CPU的结果,右图为基于预训练策略(设计20个TPU)微调的效果。
经过迭代,目前AlphaChip设计芯片的效果和速度都较此前有明显提升,达到和人类专家相近水平。
谷歌列举了AlphaChip在参与设计TPU v5e、TPU v5p、Trillium上的效果,它们分别是目前谷歌最先进的三款AI计算芯片。
AI设计芯片正在成为新范式
值得一提的是,AlphaChip团队此前还陷入过打假风波。
谷歌大脑员工Satrajit Chatterjee在内部质疑该团队发表在Nature上的论文中存在一些站不住脚、实验尚未经过充分的测试。2022年3月,该研究员被谷歌解雇。
(一个猜测:这或许也是AlphaChip迟迟没有正式发布的原因之一?)
不过,用AI设计芯片,不只是谷歌在搞。
英伟达的H100,也有AI参与设计。
通过利用深度强化学习agent设计电路,H100中有近13000条电路由AI设计。
更早之前,三星也被曝用新思科技的DSO.ai来设计Exynos处理器。
当时新思还很得意表示,DSO.ai是第一个用于处理器设计的商业AI软件。
以及生成式AI大潮下,Cadence也在力推AI设计芯片,推出了Optimization AI方案。而且还有Cadence Copilot,利用大语言模型(LLM)和其他基础模型,显著提升工程师的生产力。
参考链接:
[1]https://deepmind.google/discover/blog/how-alphachip-transformed-computer-chip-design/
[2]https://x.com/demishassabis/status/1839354651206160563
[3]https://x.com/JeffDean/status/1839308592408834559
- 神秘「蓝莓」登顶文生图竞技场,疑似Flux续作,网友:都来打广告2024-09-29
- 百度智能云晒大模型落地成绩单:5个第一!2024-09-25
- OpenAI连丢3高管,光杆司令奥特曼紧急扶持华人上位!半年流失15员大将2024-09-26
- GPT-4o能玩《黑神话》!精英怪胜率超人类,无强化学习纯大模型方案2024-09-22