TPU v4芯片科技之星照亮人工智能的未来天空
Google I/O开发者大会在疫情的影响下取消了去年的线上形式，这年份它以强势回归的姿态展开。在没有开发者亲临场地的Google园区中，CEO桑达尔·皮查伊宣布了一系列令人瞩目的新技术，其中包括能够实现“空间瞬移”的全息视频聊天技术Project Starling，以及最新一代AI芯片TPU v4。
这项科技被描述为“我们部署到Google系统中的最快系统之一，对我们来说是一个具有历史意义的里程碑。”皮查伊对此进行了介绍。TPU v4不仅速度提升2倍，更是性能提升10倍，它主要与Pod相连发挥作用，每个Pod中包含4096个单独的TPU v4芯片，通过其独特互连技术，将数百独立处理器转化为一个高效运行的系统。每个Pod都能达到1 exaFlOP级算力，即全球最快超级计算机“富岳”性能的一半。
今年MLPerf结果显示，Google TPU v4在图像分类训练测试方面表现出色，在使用ImageNet数据集时，只需1.82分钟完成任务，而其他竞争对手如Nvidia A100、AMD Epyc 7742和Intel Xeon Platinum 8168需要更多时间才能完成同样的任务。此外，当用于大型维基百科语料库上的BERT模型训练时，TPU v4也展示出了其卓越性能。
除了这些成就之外，Google还展示了多种利用TPU v4进行AI应用实例，如MUM模型（Multitask Unified Model）和专为对话设计的LaMDA。这两种模型分别比BERT强大1000倍，并且能够同时处理网页、图像等多种数据类型。LaMDA则可以与人类无缝交流，这些都是依赖于TPU v4支持的大型项目。
自2016年首次宣布以来，Google已经发布了五代自研AI芯片，其更新频率每隔一年一次。从第一代采用28nm工艺制程到第二代增加推理能力，再到第三代提高浮点运算能力，现在第四代更是进一步加速这一进程。而未来看来，无论何种形态的人类社会发展，都将离不开这种不断进步的人工智能技术支撑。
