AlphaChip:人工智能如何革新芯片设计

Gábor Bíró 2024年9月23日
4 阅读时间

谷歌DeepMind近期发布了AlphaChip,这是一款旨在革新计算机芯片设计的开源人工智能系统。AlphaChip能够在短短数小时内生成优化的芯片布局,这一关键步骤被称为布局规划,而传统的人工方法可能需要数月才能完成。这项突破性进展在2021年《自然》杂志的一篇论文中得到了详细阐述。

AlphaChip:人工智能如何革新芯片设计
来源:

据谷歌DeepMind称,AlphaChip已在谷歌张量处理器(TPU) последних 三代的设计中发挥了重要作用,加速了人工智能的进步,并正在改变芯片制造和设计的格局。

AlphaChip人工智能系统概述

AlphaChip采用复杂的强化学习方法来优化芯片设计,将复杂的布局规划任务视为一个错综复杂的难题。该系统的主要特点包括:

  1. 利用一种新颖的“基于边”的图神经网络来学习芯片组件之间的关系。
  2. 将芯片布局规划视为一个游戏,将电路组件按顺序放置到网格(芯片画布)上。
  3. 通过从过去的布局中学习,随着经验的积累变得更快、更准确。
  4. 能够推广到不同的芯片设计,使其应用范围广泛。
  5. 在处理当前设计之前,先在先前几代芯片模块上进行预训练。
  6. 在数小时内生成布局(布线规划图),而人工则需要数周或数月的时间。
  7. 在优化线长和组件放置方面达到超人的性能,这是衡量芯片效率和性能的关键指标。

AlphaChip的方法与AlphaGo和AlphaZero相似,将强化学习技术应用于复杂的现实工程问题。这种创新方法不仅加速了芯片设计过程,还为创建更高效、更强大的计算机芯片以用于各种应用开辟了新的可能性。

对谷歌TPU设计的影响

AlphaChip对谷歌张量处理器(TPU)的设计产生了重大影响,在优化 последних 三代人工智能加速器中发挥了关键作用:

  • 对于TPU v5e,AlphaChip放置了10个模块,与人类专家相比,线长减少了3.2%。
  • 在当前的第六代TPU(名为Trillium)中,AlphaChip放置了25个模块,线长减少了6.2%。

Trillium是谷歌的第六代TPU,其单芯片的峰值计算性能几乎是上一代的五倍,内存带宽翻倍,能源效率提高了67%。这些进步直接促进了谷歌强大的生成式人工智能系统的发展,包括大型语言模型(如Gemini)以及图像和视频生成器(如Imagen和Veo)。

行业影响

AlphaChip的影响力已扩展到谷歌之外,在整个半导体行业引发了人工智能辅助芯片设计的创新浪潮:

  • 联发科等主要厂商已采用并扩展了AlphaChip的功能,以加速其最先进芯片的开发,包括用于三星智能手机的Dimensity旗舰5G芯片。
  • SEMI预测,在全球范围内,300mm晶圆厂设备的支出可能在2025年至2027年间达到创纪录的4000亿美元,部分原因是数据中心和边缘设备对人工智能芯片的需求不断增长——这些芯片的设计越来越多地借助人工智能。

这种投资激增反映出人们越来越认识到人工智能在革新芯片设计和制造工艺方面的潜力。

开源和未来潜力

谷歌DeepMind已发布了AlphaChip的全面开源资源,使研究人员和开发人员能够探索和构建这项突破性技术。开源软件包包括:

  1. 一个软件存储库,完全复现了最初《自然》论文中描述的方法。
  2. 一个在20个TPU模块上训练的预训练模型检查点。
  3. 一个详细的教程,解释了如何使用开源存储库执行预训练。

这些资源可在GitHub上获取,允许外部研究人员在不同的芯片模块上预训练系统,并将其应用于新的设计。

展望未来,谷歌的目标是让AlphaChip优化芯片设计周期的每个阶段,从计算机架构探索到制造。这可能会促成开发出速度更快、成本更低、能效更高的芯片,用于从智能手机到医疗设备和农业传感器的各种设备。

总结与未来展望

AlphaChip的推出标志着人工智能和芯片设计领域的一个重要里程碑。该系统不仅显著加快了设计过程,特别是布局规划,而且还有潜力产生更高质量、更高效的芯片。预计这一发展将对从消费电子产品到自动驾驶汽车等众多行业产生深远影响。

AlphaChip创造了一个强大的反馈循环:人工智能设计的芯片能够支持更先进的人工智能模型,而这些模型反过来又可以设计出更好的芯片。这种循环可能会极大地加速芯片设计和人工智能本身的进步。

随着这项技术的发展和普及,我们将饶有兴致地看到它不仅如何改变科技行业,还将如何改变我们的日常生活。AlphaChip的意义超越了芯片制造——它代表着人工智能在设计和开发未来技术方面发挥基础性作用迈出了一步。

Gábor Bíró 2024年9月23日