王兴军教授课题组
首页 研究方向 课题组成员 科研成果 组内动态 加入我们 English

联系方式

王兴军教授邮箱:xjwang@pku.edu.cn

王晓红助理邮箱:xhwang@pku.edu.cn

相关链接

中心在Nature Communications期刊发表光计算方面的研究成果

2023年1月5日,中心在Nature Communications杂志在线发表文章“Microcomb-based integrated photonic processing unit”。文章首次报道了一种基于光子神经网络的超高算力密度硅基集成光子处理器。研究团队历时3年,攻克了多波长并行光计算系统校准和超高精度权重加载两个世界性技术难题。

EE1A

图1  论文截图

人工智能(AI)技术此前已在数据密集型计算任务中得到广泛应用。后摩尔时代,为满足AI大算力和低能耗的巨大需求,光子神经网络应运而生。在特定的光学结构中,利用光子可以实现神经网络中的矩阵乘法、卷积等基本数学运算。得益于光子超高速、大带宽和低能耗的天然优势,计算速度和能效大幅提升。近年来,光子神经网络已经可以通过成熟的光电子集成平台实现。由于其小尺寸、可扩展和低能耗的巨大优势,这种新型的光子处理单元有望突破传统微电子处理器的算力和能效瓶颈,同时彻底改变人工智能的硬件部署.

为了进一步挖掘光计算潜力,光子神经网络需要通过并行的方式提升计算吞吐量。在光子神经网络中引入波分复用是一种有效手段。然而这种先进的计算架构在集成方面面临巨大的挑战。除了实现集成多波长光源的难度大以外,另一个重要的难点是多波长光子神经网络相比单波长方案系统更复杂,需要更精细的校准方法和更有效的控制手段。因此,长期以来,芯片化一直是制约并行光子神经网络发展的瓶颈,阻碍了相关技术的推广和普及。

1FDC7

图2 高算力密度集成光子处理器

该工作中,研究团队实现了一个由多波长光源、高速数据加载/接收单元和计算核心构成的硅基集成光子处理器。针对多波长计算系统开发的校准方法能够实现对各部分光电子器件的准确调控,为目前最高权重控制精度(9 bits)的实现奠定了基础。得益于极高的系统集成度和超高速的信息加载/接收速率,该光电子处理器的算力密度高达1.04 TOPS/mm2。在图像边缘检测和手写数字识别测试任务中,边缘检测质量和识别的准确率(96.6%)与电子计算机水平相当(97.0%)。该工作为全集成多波长光计算系统的实现指明了方向,并有望在未来为整个人工智能领域算力突破提供新的解决方案。

该论文的共同第一作者为中心博雅博士后白博文、2020级博士研究生杨其鹏、博雅博士后舒浩文,电子学院常林助理教授。王兴军教授,美国加州大学圣塔巴巴拉John E. Bowers教授和电子学院常林助理教授为论文的通讯作者。主要合作者还包括上海交通大学邹卫文教授,张江实验室杨丰赫副研究员,加州大学圣塔巴巴拉分校谢卫强博士(现为上海交通大学副教授),电子学院胡薇薇教授,中心2019级博士研究生沈碧涛2020级博士研究生陶子涵。该工作由北京大学电子学院区域光纤通信网与新型光通信系统国家重点实验室作为第一单位完成,也是国家重点实验室北京分校区和上海分校区合作的重要成果。

论文原文链接:

https://www.nature.com/articles/s41467-022-35506-9


发表日期:2023年01月06日