[新闻] 受《周易》启发!北京清华团队发布陆AI光芯片“太极”

楼主: horrorghost (恐怖复苏)   2024-04-14 19:59:14
https://www.chinatimes.com/realtimenews/20240413000997-260409?chdtv
受《周易》启发!北京清华团队发布陆AI光芯片“太极”
08:57 2024/04/13 中时 李文辉
据微信公众号“清华大学”12日消息,北京清华大学电子工程系方璐副教授课题组、自动
化系戴琼海院士课题组摒弃传统电子深度计算范式,另辟蹊径首创分布式广度智慧光计算
架构,研制全球首款大规模干涉衍射异构集成芯片太极(Taichi),实现160 TOPS/W的
通用智慧计算。该研究成果于12日凌晨以〈大规模光芯片“太极”赋能160 TOPS/W通用
人工智能〉为题发表在最新一期的《科学》(Science)上。
方璐、戴琼海为论文的通讯作者,电子工程系博士生徐智昊、博士后周天贶(清华大学水
木学者)为论文第一作者。
随着各类大模型和深度神经网络涌现,制造出满足人工智能发展、兼具大算力和高能效的
下一代AI芯片已成为国际前沿热点。
作为人工智能的三驾马车之一,算力,是训练AI模型、推理任务的关键。光计算,是将计
算载体从电变为光,利用光在芯片中的传播进行计算,以其超高的并行度和速度,被认为
是未来颠覆性计算架构的最有力竞争方案之一。光芯片,具备高速高平行计算优势,被寄
予希望用来支撑大模型等先进人工智能应用。
智慧光计算作为新兴计算模态,在后摩尔时代展现出有望超越硅基电子计算的潜力。然而
其计算任务局限于简单的字符分类、基本的影像处理等。其痛点是光的计算优势被困在了
不适合的电架构中,计算规模受限,无法支撑亟需高算力与高能效的复杂大模型智慧计算

协助光计算“挣脱”算力瓶颈,另辟蹊径,“从0到1”重新设计适合光计算的新架构,是
清华团队迈出的关键一步。
方璐将这次科研历程比拟为浪漫的“双向奔赴”:从算法架构上自顶向下探索,在硬件
芯片设计上自底向上推演。
相异于电子神经网络依赖网络深度以实现复杂的计算与功能,“太极”光芯片架构源自光
计算独特的“全连接”与“高并行”属性,化深度计算为分布式广度计算,为实现规模易
扩展、计算高并行、系统强鲁棒的通用智慧光计算探索了新路径。
论文第一作者、电子系博士生徐智昊指出,在“太极”架构中,自顶向下的编码拆分-解
码重构机制,将复杂智慧任务化繁为简,拆分为多通道高并行的子任务,构建的分布式“
大感受野”浅层光网络对子任务分而治之,突破物理模拟器件多层深度级联的固有计算误
差。
团队以周易典籍“易有太极,是生两仪”为启发,建立干涉-衍射联合传播模型,融合衍
射光计算大规模并行优势与干涉光计算灵活重构特性,将衍射编解码与干涉特征计算进行
部分/整体重构复用,以时序复用突破通量瓶颈,自底向上支撑分布式广度光计算架构,
为片上大规模通用智慧光计算探索了新路径。
通俗来讲,干涉-衍射的组合方式仿佛就是在拼乐高玩具。乐高积木可以通过一个模组凹
槽与另一个模组凸起的契合来完成两个组件的拼接。在科研团队眼中,一旦把干涉、衍射
变成基础模组,进行重构复用,可以凭借丰富的想像力搭建出变化无穷的造型。
据论文报导:“太极”光芯片具备879 T MACS/mm2 的面积效率与160 TOPS/W的能量效
率,首次赋能光计算实现自然场景千类物件识别、跨模态内容生成等人工智能复杂任务。
“太极”光芯片有望为大模型训练推理、通用人工智能、自主智慧无人系统提供算力支撑

方璐表示,“之所以将光芯片命名为‘太极’,也是希望可以在如今大模型通用人工智能
蓬勃发展的时代,以光子之道,为高性能计算探索新灵感、新架构、新路径 。”
北京资讯科学与技术国家研究中心的光电智慧技术交叉创新团队由来自电子系、自动化系
、积体电路学院、软件学院的领域学者和专门研究人员组成。
方璐认为,这次突破性科研成果的成功取得,是北京清华大学深入推进有组织科研的一次
生动实践。
该课题受到科技部2030“新一代人工智能”重大项目、国家自然科学基金委杰青项目、基
础科学中心专案,清华大学-之江实验室联合研究中心支持。
在合作者中,有来自各个学科、不同背景的成员。集思广益,多学科、多角度地探索更多
解决途径。
作者: observer0117 (ob)   2024-04-15 06:59:00
光学逻辑闸芯片,用光取代电子?https://www.technice.com.tw/uncategorized/30121/理论上光比电子快,但现阶段光学元件很难缩小到奈米等级
作者: kevin0308030 (幸福微光)   2024-04-15 14:47:00
看不懂

Links booklink

Contact Us: admin [ a t ] ucptt.com