清华团队发布AI光芯片“太极”:灵感来自周易

作为人工智能的三驾马车之一,算力是训练AI模型、推理任务的关键。

清华大学科研团队的新成果发布在了4月12日凌晨的最新一期《科学》上,首创分布式广度智能光计算架构,研制出全球首款大规模干涉衍射异构集成芯片“太极(Taichi)”,实现了160 TOPS/W的通用智能计算。

清华团队发布AI光芯片“太极”:灵感来自周易

据介绍,“太极”光芯片架构开发的过程中,灵感来自典籍《周易》,团队成员以“易有太极,是生两仪”为启发,建立了全新的计算模型,实现了光计算强悍性能的释放。

光计算,顾名思义是将计算载体从电变为光,利用光在芯片中的传播进行计算,以其超高的并行度和速度,被认为是未来颠覆性计算架构的最有力竞争方案之一。

光芯片具备高速高并行计算优势,被寄予希望用来支撑大模型等先进人工智能应用。

清华团队发布AI光芯片“太极”:灵感来自周易

据论文第一作者、电子系博土生徐智吴介绍,在“太极”架构中,自顶向下的编码拆分-解码重构机制,将复杂智能任务化繁为简,拆分为多通道高并行的子任务,构建的分布式'大感受野’浅层光网络对子任务分而治之,突破物理模拟器件多层深度级联的固有计算误差。

论文报道:“太极”光芯片具备879T MACS/mm的面积效率与160 TOPS/N的能量效率。首次赋能光计算实现自然场景千类对象识别、跨模态内容生成等人工智能复杂任务。

“太极”光芯片有望为大模型训练推理、通用人工智能、自主智能无人系统提供算力支撑。

清华团队发布AI光芯片“太极”:灵感来自周易

清华团队发布AI光芯片“太极”:灵感来自周易