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

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

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

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

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

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

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

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

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