谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计
本文经AI新媒体量子位(公众号ID:QbitAI)授权转载,转载请联系出处。
设计一块AI芯片有多难?
这么说吧,围棋的复杂度10360,而芯片则是102500,你感受一下……
一般来说,工程师们设计一块芯片,少则需要几周,多则好几个月。
现在,AI生产力来了!
AI自己动手,竟然用6小时就设计出一块芯片。
最近,这项谷歌的研究登上了Nature杂志。
![谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计](https://s6.51cto.com/oss/202106/10/e80535e4aed43c5c1ef6bb23a87d667a.jpg)
布局时间缩短数倍
小小的一块芯片包括了数十亿个晶体管,由它们组成的数千万个逻辑门就是标准单元,此外还有数千个存储块,称为宏块。
确定它们的位置,也就是布局规划,对芯片设计至关重要。
![谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计](https://s3.51cto.com/oss/202106/10/d5f797b5e0f44b5cb38ca03805ed6798.jpg)
因为这直接关系到如何布线,进而影响着芯片的处理速度和电源效率。
但是,光是放置宏块这一步就非常耗时,为了给标准单元留出更多空间,每一次迭代都需要几天或几周时间。
△人类设计和AI设计芯片的平面图(灰色块为宏块)
完成整个布局,则要花费数周甚至数月。
现在,谷歌的研究人员提出了一种具有泛化能力的芯片布局方法。
它能够基于深度强化学习,从之前的布局中进行学习,然后生成新的设计方案。整体架构是这样的:
![谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计](https://s2.51cto.com/oss/202106/10/22d4aea3d1ce9e2f8f1156db9b069c68.jpg)
由于AI模型需要学习10万个芯片布局,为了保证速度,研究人员设计了一种奖励机制,基于线路长度和布线拥塞的近似代价函数进行计算。
![谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计](https://s4.51cto.com/oss/202106/10/664ade78684072c730cbd133a24beb6c.jpg)
具体来说,需要将宏和标准单元映射到一个平面画布上,形成具有数百万到数十亿节点的「芯片网表」。
然后,AI模型会对功率、性能和面积(PPA) 等进行优化,并且输出概率分布。
下图分别是零样本生成和基于预训练策略微调的效果,其中每个小矩形代表一个宏块。在预训练策略中,中间留出了用于放置标准单元的空间。
![谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计](https://s6.51cto.com/oss/202106/10/51929ffed7fa4dca9d654dc852b2e627.jpg)
与其他方法相比,谷歌的新方法大大减少了设计时间,只需不到6小时,就能实现性能优化的布局。
![谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计](https://s2.51cto.com/oss/202106/10/e94b0444f1ea054cd225414ad81dabda.jpg)
谷歌:效果不错,已经用上了
研究团队对不同策略下的布局效果进行了可视化展示,从图中可以看到,预训练策略微调的结果要明显优于零样本生成。
![谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计](https://s3.51cto.com/oss/202106/10/26017835de794c8a2ffd7e59fad16d04.jpg)
并且,从不同训练时长的效果对比可以看到,在训练2-12小时的情况下,预训练策略要优于零样本生成。
![谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计](https://s4.51cto.com/oss/202106/10/12c364b5489d2e78e6c2d7f0a3075e2e.jpg)
在不同规模的数据集上进行测试,研究人员发现,随着数据集规模的扩大,生成布局的质量和收敛时间的结果更优。
![谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计](https://s4.51cto.com/oss/202106/10/cdae63a20398a08b8ead32d01b1464b4.jpg)
![谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计](https://s4.51cto.com/oss/202106/10/084bcf5facd279f63d185d4b7563578b.jpg)
谷歌表示,
这一方法适用于任何类型的芯片。
目前已经被用于生产下一代Google TPU(加速器芯片)。
![谷歌让AI芯片学会“下崽”,下一代TPU就让AI自己设计](https://s6.51cto.com/oss/202106/10/9ed5132b13f6a67ffe90310797bfadad.jpg)
相关文章
- 2021年中国基础云服务行业市场规模已经达到2256亿元
- 扬开源之帆,借官方认证考试之势,在NGINX领域脱颖而出
- 计算机视觉帮助解决业务挑战的五种方法
- 重磅!白山云全新发布 Baishan Canvas——极速、安全、敏捷的一体化边缘云平台
- Game Over?强AI与弱AI之争
- AI 成为增长新引擎,来听听亚马逊云科技的技术方法论
- 智能化仓库管理将成趋势 慧策成企业数字化转型加速器
- IDC:2020年全球电信市场规模为1.53万亿美元
- AI落地制造业:智能机器人应具备这四种能力
- LG公司宣布在美国开设新的云呼叫中心
- 等保关保双制度下,瑞数动态安全超融合助力企业落地“三化六防”
- 谁将在容器软件市场上获胜
- 人工智能的环境成本和承诺
- Cloudera:上云不是终点,统一有效的数据战略才是关键
- 云计算的宿命:全球合并成一台计算机,支持无服务器运行
- 从运营商实践,看“智联多云”为何成为云化战略成功关键
- 你真的了解3D打印吗
- 云原生系统之弹性模式
- 5G技术应用于未来美军事网络的优势
- 全球芯片短缺已开始在现实世界中产生重大影响