#光電混合晶片
中國初創晶片登Nature:比輝達A10快500倍!自研光電混合技術
來自中國的光電混合晶片技術,登上最新頂刊Nature!這次的成果主要聚焦在了自主研發的光子計算處理器——PACE(Photonic Arithmetic Computing Engine)。簡單來說,PACE是一種基於光電混合的架構,它通過光執行矩陣向量乘法,可以實現超低延遲和高能效的計算。根據論文中公開的資料顯示,PACE在解決組合最佳化問題(如伊辛問題和最大割/最小割問題)時,計算延遲低至3納秒,比傳統GPU快了兩個數量級。這一突破的核心在於PACE的高度整合設計。這個系統整合了超過16000個光子元件,並通過創新的2.5D混合先進封裝技術,將光子積體電路(PIC)與電子積體電路(EIC)無縫整合。這種設計不僅解決了大規模光電系統整合中的技術難題,更為商業化落地奠定了基礎。而這個晶片技術,正是來自國內初創企業曦智科技。據瞭解,這是繼八年前曦智科技創始人沈亦晨博士在Nature發表封面論文後,再一次登上這一頂刊。那麼PACE如此的速度,到底是如何做到的呢?首次公開:16000個光子元件的高度整合隨著人工智慧的快速發展,計算需求呈爆發式增長,傳統電子計算面臨著功耗、速度等方面的瓶頸。光子計算憑藉光的獨特性質,如同時進行乘法和累加過程、資料傳輸能耗低、避免電阻損耗和發熱問題等,成為極具潛力的替代方案,受到全球廣泛關注。然而,光子計算在發展過程中面臨諸多挑戰。一方面,整合光子學製造相對不成熟,缺乏先進的封裝解決方案,導致大規模整合光子系統在性能提升、標準設計與驗證以及封裝等方面困難重重。另一方面,光子計算在光學儲存、計算精度(尤其在大規模複雜電路中)以及適配模型和演算法等方面存在不足,限制了其商業化處理程序。為此,曦智科技提出的PACE採用混合架構(首次對外公開),將光子積體電路(PIC)和電子積體電路(EIC)整合在一個系統級封裝(SiP)中。△PACE系統部署PIC 負責執行光矩陣向量乘法(oMAC)操作,EIC則處理控制、迭代邏輯、資料輸入輸出、儲存以及時鐘控制等功能。這種架構設計充分發揮了光子計算在速度和低延遲方面的優勢,以及電子計算在邏輯處理和儲存方面的長處。在PIC中,團隊設計了1×64光學資料模組和64×64權重模組執行oMAC操作。光訊號通過高性能光柵耦合器從外部雷射陣列耦合進入電路,經過向量調製器陣列和權重調製器模組進行調製,最後在光電探測器陣列進行訊號轉換和合併。EIC基於28-nm商業CMOS技術設計,PIC則是基於65-nm矽光子技術建構,單個晶片整合了超過16000個光子元件,實現了高度整合。這種混合架構充分發揮了光計算的平行優勢:光訊號在波導中傳輸時天然完成乘加運算(oMAC),而電子電路則處理邏輯控制與資料儲存。實驗資料顯示,64×64矩陣運算延遲僅3納秒,比傳統GPU快500倍。除此之外,研究團隊創造性地將光學矩陣運算應用於組合最佳化問題。通過設計”噪聲驅動遞迴演算法”,PACE系統能夠高效求解伊辛模型:在求解63節點Max-cut問題時,系統經過平均537次迭代(耗時2.7μs)即可達到92.7%的收斂率,相比NVIDIA A10 GPU提速295倍。更引人注目的是”圖像搜尋”演示,系統能從隨機初始狀態收斂到預設的”貓”圖像目標。Nature審稿人對曦智科技團隊在光子計算工程化方面作出的努力給予了高度肯定:在光子計算領域,通常會通過小規模的演示對大規模系統性能進行樂觀推斷,但本文中的資料均來自整個PACE計算系統的實測性能,作者們工程化地實現了一個超大規模光子矩陣計算系統,可謂“壯舉”。還全球首發了新一代光電計算卡就在前不久的3月25日,曦智科技還正式推出全新一代光電混合計算卡 ——曦智天樞。曦智天樞深度融合了光晶片與電晶片的優勢,採用先進的3D封裝技術,是一款高度可程式設計的光電混合計算卡。與前代產品相比,其在光電整合度、光子矩陣規模、計算精度及可程式設計性等方面均實現了顯著提升。它不僅支援科學計算(如伊辛演算法),還增強了對ResNet50等商業演算法的適配性,進一步拓寬了應用場景。曦智天樞採用非相幹架構設計,具備出色的抗干擾能力和高計算精度。其核心處理器由光學處理單元(OPU)和電學專用積體電路(ASIC)組成,通過3D先進封裝技術實現協同工作,主頻速率達1GHz,輸出精度為8bit。光晶片面積提升至600平方毫米,器件數量超過四萬個,整合度大幅提高。此外,其最大支援128x128矩陣規模,運算能力和靈活性均得到顯著增強。使用者可通過API自由組態計算矩陣係數,實現更高效的最佳化與適配。在軟體方面,產品搭載了曦智光電混合計算軟體棧,支援主流框架如PyTorch和ONNX,使用者可通過曦智編譯器靈活建構高效的應用模型。不僅如此,平台還支援使用者自訂算子,進一步擴展了演算法開發的靈活性。對此,沈亦晨博士表示:曦智天樞首次實現了光電混合計算在複雜商業化模型中的應用,是曦智科技光電混合算力技術在產品化和商業化處理程序中的重要突破。我們堅信,光電混合將會為人工智慧、大語言模型、智能製造等領域帶來算力革新。光+電,會是未來的答案。 (量子位)