谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计 / 头条资讯

13天前 阅读 / 4 来源 / 原创 文 / 越读党订阅

月石一 发自 凹非寺量子位 报道 | 公众号QbitAI 设计一块AI芯片有多难? 这么说吧,围棋的复杂度10360,而芯片则是102500,你感受一下 △围棋的复杂度 一般来说,工程师们设计一块芯片,少则需要几周,多则好几个月。 现...

月石一 发自 凹非寺
量子位 报道 | 公众号QbitAI

设计一块AI芯片有多难?

这么说吧,围棋的复杂度10360,而芯片则是102500,你感受一下



△围棋的复杂度

一般来说,工程师们设计一块芯片,少则需要几周,多则好几个月。

现在,AI生产力来了!

AI自己动手,竟然用6小时就设计出一块芯片。

最近,这项谷歌的研究登上了Nature杂志。


小小的一块芯片包括了数十亿个晶体管,由它们组成的数千万个逻辑门就是标准单元,此外还有数千个存储块,称为宏块。

确定它们的位置,也就是布局规划,对芯片设计至关重要。


因为这直接关系到如何布线,进而影响着芯片的处理速度和电源效率。

但是,光是放置宏块这一步就非常耗时,为了给标准单元留出更多空间,每一次迭代都需要几天或几周时间。


△人类设计和AI设计芯片的平面图(灰色块为宏块)

完成整个布局,则要花费数周甚至数月。

现在,谷歌的研究人员提出了一种具有泛化能力的芯片布局方法。

它能够基于深度强化学习,从之前的布局中进行学习,然后生成新的设计方案。整体结构是这样的:


由于AI模型需要学习10万个芯片布局,为了保证速度,研究人员设计了一种奖励机制,基于线路长度和布线拥塞的近似代价函数进行计算。


具体来说,需要将宏和标准单元映射到一个平面画布上,形成具有数百万到数十亿节点的「芯片网表」。

然后,AI模型会对功率、性能和面积(PPA) 等进行优化,并且输出概率分布。

下图分别是零样本生成和基于预训练策略微调的效果,其中每个小矩形代表一个宏块。在预训练策略中,中间留出了用于放置标准单元的空间。



与其他方法相比,谷歌的新方法大大减少了设计时间,只需不到6小时,就能实现性能优化的布局。


研究团队对不同策略下的布局效果进行了可视化展示,从图中可以看到,预训练策略微调的结果要明显优于零样本生成。


并且,从不同训练时长的效果对比可以看到,在训练2-12小时的情况下,预训练策略要优于零样本生成。


在不同规模的数据集上进行测试,研究人员发现,随着数据集规模的扩大,生成布局的质量和收敛时间的结果更优。



谷歌表示,

这一方法适用于任何类型的芯片。
目前已经被用于生产下一代Google TPU(加速器芯片)。


参考链接:

https://www.nature.com/articles/s41586-021-03544-w

https://www.nature.com/articles/d41586-021-01515-9

https://ai.googleblog.com/2020/04/chip-design-with-deep-reinforcement.html

https://arxiv.org/abs/2004.10746


特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

0

相关阅读
#头条资讯

自主品牌承压明显 上半年乘用车累销770.4万辆

版权声明:本文版权为网易汽车所有,转载请注明出处。 网易汽车7月8日报道 7月8日,全国乘联会发布全国乘用车销量数据。6月,乘用车零售销量达到165.4万辆,同比下降6.2%,环比增长2.9%。至此,从3月以来,乘用车市场已经持续4个月...

#头条资讯

高合HiPhi X首批试生产下线 预计于明年上半年交付

版权声明:本文版权为网易汽车所有,转载请注明出处。网易汽车10月31日报道10月30日,华人运通新能源汽车项目暨高合HiPhi X(参数丨图片)首批试生产车顺利下线,并实现全球首创生产下线无人驾驶。此前,高合HiPhi X已于北京车展首发亮...

#杂志封面

《NBA特刊》杂志_NBA特刊2020年第02期杂志封面

《NBA特刊》杂志_NBA特刊2020年第02期杂志封面

#头条资讯

存安全隐患 特斯拉将召回共4.86万辆进口电动汽车

日前,特斯拉汽车(北京)有限公司根据《缺陷汽车产品召回管理条例》和《缺陷汽车产品召回管理条例实施办法》的要求,向国家市场监督管理总局备案了召回计划,自即日起召回部分进口Model S、Model X电动汽车。具体信息如下:...

#杂志封面

《时代金融》杂志_时代金融2020年第03期杂志封面

《时代金融》杂志_时代金融2020年第03期杂志封面

#头条资讯

加拿大涉及特斯拉奇事:车辆"看见"警灯后自动加速

日前,据外媒报道,加拿大联邦警察报告了一起涉及特斯拉汽车的奇怪事件。加拿大阿尔伯塔省皇家骑警接到投诉,“一辆汽车在Ponoka附近的2号公路上超速行驶。这辆车似乎当时正在自动驾驶,时速超过140公里,两个前排座椅完全倾斜...