清华“太极”光芯片:采用最古朴思路,“苦熬”而成

近日,清华大学电子工程系副教授方璐带领课题组成员,与中国工程院院士、清华大学自动化系戴琼海院士课题组组成交叉研究团队,在智能光计算芯片领域实现了新的突破。相关成果已于近日发表在Science期刊。

近日,清华大学电子工程系副教授方璐带领课题组成员,与中国工程院院士、清华大学自动化系戴琼海院士课题组组成交叉研究团队,在智能光计算芯片领域实现了新的突破。相关成果已于近日发表在Science期刊。

他们首创干涉-衍射分布式广度光计算架构,研制了全球首款大规模通用智能光计算芯片——太极(Taichi),其系统级能效为每秒每焦耳160万亿次运算,超越主流商用AI芯片3个数量级,为后摩尔时代高性能智能计算开辟了新路径。

太极芯片首次赋能智能光计算实现超过1000个类别的自然场景图像分类,以及跨模态内容生成等智能任务,可为AI大模型、智能无人系统、通用人工智能(AGI)等提供强有力的算力支持。

在这项研究初期,团队一度遭遇很大的困难。令研究者感慨的是,他们最终通过回顾上世纪80、90年代甚至更早期的经典成果,突破了研究瓶颈。他们表示,正是这种追求初心的态度使得研究团队能够不受时代局限,不受潮流影响,始终保持对科学问题本质的关注和热情,从而产生新的突破。图片

Science论文截图

一加一大于二

光具备传播速度快、表征维度多、计算功耗低等物理特性,智能光计算成为新一代人工智能的国际交叉前沿,拥有广泛的应用前景。研究团队针对大规模通用智能光计算难题,摒弃了现有光计算沿用传统电子深度计算的范式,提出分布式广度计算架构,构建深度浅但宽度广的光神经网络,整体架构可重构、可复用。

与传统的深度计算层层堆叠的方法不同,太极通过将复杂智能任务化繁为简,拆分为多通道、高并行的子任务,同时为子任务单独组织集群、分配计算资源,从而实现复杂任务的高效处理。受“易有太极,是生两仪”的中国传统哲学理念启发,研究团队以光的干涉和衍射表征“两仪”,建立干涉-衍射片上联合传播模型。

“通过融合干涉的灵活可重构特性和衍射的大规模高并行特性,以辩证统一的理念实现干涉-衍射智能光计算。这种统一使得太极光芯片不仅具有可重构通用计算能力,也具备高通量并行计算能力,实现一加一大于二的效果。”论文共同一作、清华大学电子工程系博士生徐智昊解释道。

图片

太极芯片。受访者供图

最古朴的思路

研究初期,团队沿用了电子计算的深度学习架构以构建大规模智能光计算,然而,推进了半年就遇到了瓶颈:随着层数的增加,计算规模与计算精度存在不可调和的矛盾。“过去,我们构建网络结构大多是沿用电计算架构,却发现光的优势和潜力无法在电的架构中发挥出来,如同笼中困兽。通过理论建模和分析发现,是电的架构‘囚禁’了光的能力,也就是说现有的深度神经网络的架构并不适合智能光计算。”方璐在接受《中国科学报》采访时说。为了挣脱瓶颈,研究团队决定走出固有电架构思维的舒适区,寻求新的架构突破。他们将目光投向了上世纪80、90年代甚至更早期有关机器学习、神经网络的经典研究成果。在这些可能被遗忘的想法和经典智慧中,通过重新审视和借鉴,他们找到了突破当前困境的关键:回到最古朴的思路——做宽、做浅。在方璐看来,回顾经典也是回归科研的初心,放下对潮流的盲从。这种追求初心的态度使得研究团队能够不受时代局限,不受潮流影响,始终保持对科学问题本质的关注和热情,从而发现新的思路、提出新的理论。

太极光芯片的诞生是交叉学科合作的结晶,脑科学研究也为太极光芯片的架构研发提供了重要思路。有脑科学的研究提出“浅脑理论”,即大脑以浅层扁平架构形成大规模的并行计算单元。从感知到运动,甚至意识,各脑区都在这个浅层网络中发挥着重要作用。“脑科学的系列成果带给我们的研究很多启发。”方璐补充道。

“苦熬”芯片

然而,推翻构架仅仅只是个开始。接踵而至的是另一个艰巨的挑战:芯片研制。太极光芯片是研究团队历时3年不懈努力,历经无数次失败和挑战后,所收获的成果。在芯片研发的漫长征程中,流片是一个至关重要的里程碑,它标志着研究团队的理论概念将要转化为实际可制造的芯片。对于方璐和她的研究团队来说,这个阶段充满期待和焦虑。“流片周期通常需要3到6个月,等待芯片加工完成是一件很焦灼的事情。团队希望芯片快点回来,进入下一阶段的测试;同时又担心流片的效果不好。这意味着要从头再来一遍,时间成本很高。”方璐说。研究团队第一次流片时,等待了4个月,结果不尽如人意。他们不得不从零开始,重新审视每一个细节,寻找可能存在的问题。经过2个月的调整和优化,又等了6个月,第二次流片结果送回时,团队成员们露出来了开心的笑容。在这项研究研究中,太极光芯片的实验成果是团队共同“苦熬”出来的。“为了让实际实验结果达到理论仿真的预期,我们不断调整和优化实验系统。每一次实验都是一个漫长而繁琐的过程。类似的过程重复‘上演’了百余次,研究团队的目标是将千分类智能任务的准确率做到90%,最终以超出预期的结果结束了这场‘持久战’。”徐智昊告诉《中国科学报》

图片

方璐(左一)和课题组学生(中间为徐智昊)在实验室。杜珊妮摄

挑战传统

2023年9月,团队第一时间投稿给了Science编辑部。论文在3天后送审,1个多月后,研究团队就收到了第一轮审稿意见。“审稿人对太极光芯片的架构思路产生了意见分歧。深度学习发展至今,深度神经网络已经成为主流的智能计算架构。”方璐说。研究团队坚持自己的观点,用更多的理论和实验证据说服审稿人。第二轮审稿结束后,他们的论文被顺利接收了。今年是方璐从事科学研究的第17年:她本科毕业于中国科学技术大学,博士毕业于香港科技大学,现任清华大学电子工程系长聘副教授。

大规模光电智能计算是方璐团队一以贯之的研究目标,整个团队为实现这一目标构建了纵向并行、横向联网的路径规划。

“团队成员都有各自独立的研究问题,在并行探索路径的同时,彼此之间也会进行交叉合作,开展原创思想的交流碰撞。‘路漫漫其修远兮’,研究团队将会不忘初心,在大规模智能光计算的路径上持之以恒地求索。目前我们正在搭建芯片的应用系统,为产业化提供可行的方案。”方璐补充道。

论文链接:

https://www.science.org/doi/10.1126/science.adl1203

THE END
免责声明:本站所使用的字体和图片文字等素材部分来源于互联网共享平台。如使用任何字体和图片文字有冒犯其版权所有方的,皆为无意。如您是字体厂商、图片文字厂商等版权方,且不允许本站使用您的字体和图片文字等素材,请联系我们,本站核实后将立即删除!任何版权方从未通知联系本站管理者停止使用,并索要赔偿或上诉法院的,均视为新型网络碰瓷及敲诈勒索,将不予任何的法律和经济赔偿!敬请谅解!
相关阅读
  • 国外品种占九成市场?国产“猪芯片”打破种猪垄断

    国外品种占九成市场?国产“猪芯片”打破种猪垄断

    一头优秀的商品猪应该是怎样的?吃得少、长得快、瘦肉多……在全球,这样的猪被极少数品种占据绝大部分市场,在我国也是如此。我国每年出栏生猪7亿头左右,其中90%的品种,是来自国外的三元猪。

    2分钟前
  • 古尔曼:苹果 Mac Pro / Studio 产品升级 M4 芯片要等到明年年中

    古尔曼:苹果 Mac Pro / Studio 产品升级 M4 芯片要等到明年年中

    芯物联 5 月 20 日消息,彭博社记者马克・古尔曼在今天更新的 Power On 时事通讯中预测,搭载 M4 芯片的 Mac Pro、Mac Studio 发布时间要等到明年年中。古尔曼表示:“这意味着 Mac Pro、Mac Studio 产品线今年依然停留在 M2 芯片上”。

    3分钟前
  • 算力芯片三巨头,有人欢喜有人愁

    算力芯片三巨头,有人欢喜有人愁

    近日,寒武纪发布了2023年年度业绩预亏公告,交上了连续7年累计亏损近50亿元的业绩答卷。同时,距离2023年4月份公司股价最高271.47元每股,目前寒武纪股价已下跌超六成,市值腰斩。

    2024年5月14日 11:16

栏目精选