安徽省光学学会

2024年11月14日 星期四

行业动态

Light期刊发表矩阵乘法光子加速器的综述论文

文章来源:武汉光电国家研究中心作者:发布时间:2022-02-06


近年来,随着人工智能和第五代通信对数据处理需求的不断增长,数据计算容量和计算功耗快速增加。矩阵计算是科学与工程领域中应用最广泛、必不可少的信息处理工具之一。特别以深度学习为代表的神经网络算法,其主要特征就是包含了繁重的矩阵计算,以卷积神经网络为例,其矩阵计算的开销可以占据总开销的80%以上。加速和优化矩阵计算,可以大幅地改善信号处理和人工智能的计算效率和功耗。光子器件具有超大带宽和超低功耗,具有多个自由物理维度,这使得光子计算成为“后摩尔时代”中高容量、低延迟矩阵信息处理中最有竞争力的候选者之一。最近几年,光子矩阵乘法得到了迅速的发展,并被广泛应用于光信号处理、人工智能和光子神经网络等光子加速度领域。基于矩阵乘法计算的大量应用展示了光子加速器领域巨大的潜能和机会。

图1:矩阵乘法光子加速器概念图。

近期,来自华中科技大学的张新亮教授团队,与香港中文大学、上海理工大学、浙江大学以及曦智科技的多位研究学者合作,以“Photonic matrix multiplication lights up photonic accelerator and beyond”为题,在Light: Science & Applications上发表了矩阵乘法光子加速器主题的综述文章。

光子矩阵-矢量乘法分类

目前主流的光子矩阵-矢量乘法(MVM)主要包含三类,即基于(单/多)平面光转换(PLC)的矩阵计算,基于马赫泽德干涉仪(MZI)网络的矩阵计算,和基于波分复用(WDM)的矩阵计算。其中平面转换的矩阵计算又分为单平面矩阵计算(SPLC)和多平面矩阵计算(MPLC),它们都属于相干计算,已报道的输入向量长度分别可以达到357和490000的量级。MZI和WDM方法的输出向量长度一般在100以下,主要用于集成光子矩阵计算芯片。这些方法主要基于光子的空间维度或者波长维度进行矩阵计算,还可以结合光子多个维度构建超高容量的光子张量核心。

图2:光子矩阵乘法分类。

光子矩阵-矢量乘法加速应用

光子矩阵乘法网络本身可以用作光子信号处理的通用线性光子回路。近年来,MVM已成为多种光子信号处理方法的有力工具。MPLC-MVM得益于其大规模矩阵计算的能力,可以管理大量模式,可以用作通用的模式分类器,其操作的模式规模可以达到几百个,还可以进一步扩展实现光子多个经典物理维度的同时控制,以及用于空间成像加密等应用。MZI-MVM易于集成,并且由于移相器工作速度快,可以实现MZI网格功能的自动配置,用作自适应的模式处理器件,实现多个模式的自由上传、下载、复用解复用以及信道解扰等。WDM-MVM占用空间更小,更容易配置传输矩阵,已经应用于可编程脉冲整形、微环权重库以及信号分量分析等。

图3: 光子矩阵乘法应用。

人工智能技术已广泛应用于各种电子行业,如基于深度学习的语音识别和图像处理。矩阵计算作为神经网络的基本构件,占据了大部分的计算任务,提高矩阵的性能是神经网络加速的最有效手段之一。与电计算相比,光计算在数据存储和流量控制方面较差,而光学非线性的低效率限制了激活函数等非线性计算的应用。然后,通过波长、模式和偏振等复用技术,光学方法在大规模并行计算方面具有显著优势,同时具有极高的数据调制速度和低的延时。因此,光子网络非常适合于矩阵计算。光计算和人工智能的结合有望实现智能光子处理器和光子加速器。近年来,人工智能技术在光学领域也得到了快速发展。各种光子矩阵计算都被验证可用于替代神经网络算法的线性部分,其单核已经验证计算容量可以超过11TOPS,光子矩阵计算的延时普遍在皮秒量级,单次乘加操作的能耗在飞焦量级,信号调制速率可以高达100 GHz。相比电子计算,其在速率、延时、功耗等方面具有明显的优势。

挑战与展望

目前,光子矩阵计算和电子计算相比,两者之间仍然存在着巨大的差距,为了解决这个问题,直接而有效的解决方案之一是制造大规模光子集成回路。此外,通过使用光子的多个自由维度,例如模式、波长等,光学器件可以进行大量的并行计算,这些并行操作可以在一个物理光子计算核心中执行。在全光人工神经网络成熟之前,特别是在光学非线性效应和全光级联成熟前,光电混合人工智能计算仍是一种更实用、更有竞争力的深层人工神经网络的候选架构。因此,开发高效、专用的光电混合人工智能硬件芯片系统是光子人工智能的核心研究路径之一。未来可基于光子矩阵计算和电子控制实现加速器底层硬件,并开发适用于该硬件的各种算法,最后在用户层可以灵活地调用这些算法实现各种加速应用,例如信道解扰器、图像识别等。

图4:光电混合AI计算芯片架构。

论文信息:

该研究成果以" Photonic matrix multiplication lights up photonic accelerator and beyond "为题在线发表在Light:Science&Applications。来自华中科技大学的董建绩教授为论文的通讯作者,周海龙副教授为论文的第一作者。论文的合作作者还包括香港中文大学的黄超然博士、曦智科技的沈亦晨博士,上海理工大学的张启明教授和顾敏院士,浙江大学的钱超博士,陈红胜教授和阮志超教授,以及华中科技大学的成骏伟博士、董文婵博士和张新亮教授。

论文链接:https://www.nature.com/articles/s41377-022-00717-8.pdf



新闻链接:http://wnlo.hust.edu.cn/info/1641/10704.htm

免责声明:

  网站内容来源于互联网,由网络编辑负责审查,目的在于传递信息,提供专业服务,不代表本网站平台赞同其观点和对其真实性负责。如因内容、版权问题存在异议的,请与我们取得联系,我们将协调给予处理(按照法规支付稿费或删除),联系方式:ahos@aiofm.ac.cn 。网站平台将加强监控与审核,一旦发现违反规定的内容,按国家法规处理,处理时间不超过24小时。