上海交大科研團隊在光學智能計算芯片研制上取得突破性進展

  近日，上海交通大學電子信息與電氣工程學院電子工程系區(qū)域光纖通信網(wǎng)與新型光通信系統(tǒng)國家重點實驗室鄒衛(wèi)文教授團隊在光學智能計算領域取得突破性進展，研究成果以Optical coherent dot-product chip for sophisticated deep learning regression(面向復雜深度學習回歸任務的光學相干點積核芯片)為題在光學領域權威期刊Light: Science & Applications上發(fā)表。該工作成功研制了一款光學相干點積核計算芯片，具有運行復雜神經網(wǎng)絡的能力，率先在光學智能計算芯片上實現(xiàn)了高精度的醫(yī)學圖像重構任務。鄒衛(wèi)文教授為該論文的通訊作者，博士研究生徐紹夫為該論文的第一作者。

團隊研究布局

  隨著智能應用的普及，以神經網(wǎng)絡為代表的智能算法復雜度呈現(xiàn)爆發(fā)性增長，龐大的算力需求給現(xiàn)有的數(shù)字處理器帶來了極大的壓力，亟待開辟新型計算模式來緩解摩爾定律增速與算力需求增速之間的矛盾關系。近年來，光學智能計算技術受到國內外學術界廣泛關注。光學系統(tǒng)的計算時鐘頻率可以超過數(shù)十GHz(109 Hz)，同時具備靜態(tài)無功耗的特性，被認為是實現(xiàn)下一代高速低功耗智能計算加速器的潛在途徑。學術界報道的多種光學智能計算方案雖成功驗證了高速低功耗計算的能力，但所能實現(xiàn)的任務均是簡單的分類任務，與實際應用在復雜度上存在較大差距。

  2016年末，鄒衛(wèi)文教授團隊著力開展智能光學信號處理技術研究，率先驗證了智能算法可以有效提升微波光子信號處理系統(tǒng)的性能，相關成果發(fā)表于2019年Light: Science & Applications期刊上(論文鏈接：https://www.nature.com/articles/s41377-019-0176-4)，并受到微波光子學領域的廣泛關注。在此基礎上，團隊瞄準利用光學系統(tǒng)實現(xiàn)高速高效智能計算這一目標，先后提出光學點積核、光學卷積分塊技術等新型系統(tǒng)架構，分別解決了算法執(zhí)行規(guī)模與輸入端口能量效率等問題，為光學智能計算芯片的研制與應用奠定了原理基礎。

光學相干點積核芯片及其封裝模塊

  此次，鄒衛(wèi)文教授團隊與國內合作單位(北京大學、中科院半導體所)設計并研制一款光學相干點積核計算芯片。該芯片突破了陣列化光學器件的相干調控關鍵技術，成功地實現(xiàn)了實數(shù)域計算。借助于新型片上反饋控制算法，大幅提升了光學計算的數(shù)值精度。與此前的工作相比，該芯片在數(shù)域完整度與數(shù)值精度上的突破，使其具備了執(zhí)行復雜智能任務的能力。

  光學相干點積核芯片重構結果與計算機結果的對比

  本研究中，鄒衛(wèi)文教授團隊利用醫(yī)學圖像重構任務作為驗證，在芯片上成功地運行了AUTOMAP(用于通用圖像重構)神經網(wǎng)絡模型，圖像重構的質量接近了32位計算機的理想水平。該工作不僅推動光學神經網(wǎng)絡研究領域攻克實際應用難題，更為下一代智能計算技術提供了新思路。后續(xù)進一步提高芯片的器件集成規(guī)模，有望實現(xiàn)更高速、更低功耗的光學神經網(wǎng)絡處理器，緩解智能算力需求劇增與傳統(tǒng)硬件算力受限的矛盾。 該工作由上海交通大學、北京大學與中科院半導體所合作完成。上海交通大學為第一完成單位，博士研究生徐紹夫為第一作者，鄒衛(wèi)文教授為通訊作者。

  該工作受國家重點研發(fā)計劃(Program No. 2019YFB2203700)與國家自然科學基金項目(Grant No. 61822508)支持。

  Shaofu Xu, Jing Wang, Haowen Shu, Zhike Zhang, Sicheng Yi, Bowen Bai, Xingjun Wang, Jianguo Liu, and Weiwen Zou, Optical coherent dot-product chip for sophisticated deep learning regression, Light: Science & Applications, vol. 10, 221, 2021. Doi: 10.1038/s41377-021-00666-8.

  論文鏈接：https://www.nature.com/articles/s41377-021-00666-8