清華大學(xué)電子工程系與自動化系的研究團(tuán)隊(duì)聯(lián)合取得了一項(xiàng)突破性進(jìn)展——在國際上首創(chuàng)了“分布式廣度光計(jì)算架構(gòu)”,并基于此成功研制出“太極(TaiChi)”光芯片。這一重大成果不僅為人工智能(AI)計(jì)算范式帶來了全新思路,更在計(jì)算能效上實(shí)現(xiàn)了數(shù)量級的跨越,為解決AI大模型訓(xùn)練與推理所面臨的“能耗墻”瓶頸提供了極具潛力的中國方案。
突破傳統(tǒng):從深度到廣度的范式革新
當(dāng)前,以深度學(xué)習(xí)為代表的AI計(jì)算主要依賴于電子芯片(如GPU)的串行、分層處理。隨著模型參數(shù)爆炸式增長,這種“深度”計(jì)算模式面臨著功耗激增、散熱困難、傳輸延遲等嚴(yán)峻挑戰(zhàn)。清華大學(xué)團(tuán)隊(duì)另辟蹊徑,提出“分布式廣度光計(jì)算”新架構(gòu)。其核心思想在于,將復(fù)雜的計(jì)算任務(wù)拆解并映射到一個大規(guī)模并行處理的廣域空間中,利用光的物理特性進(jìn)行瞬時并行計(jì)算。這類似于從“一條深巷走到頭”轉(zhuǎn)變?yōu)椤耙粋€廣場同時處理”,從根本上改變了計(jì)算的執(zhí)行方式。
“太極”光芯:融合智慧,駕馭光速
以這一創(chuàng)新架構(gòu)為基石研制的“太極”光芯片,是其理念的物理載體。芯片取名“太極”,寓意著陰陽共濟(jì)、動態(tài)平衡,體現(xiàn)了其融合光計(jì)算與智能算法、協(xié)調(diào)并行與精準(zhǔn)控制的特性。該芯片的關(guān)鍵創(chuàng)新在于:
- 大規(guī)模并行與高維度操作:通過精妙的集成光子學(xué)設(shè)計(jì),芯片能在單個時鐘周期內(nèi)完成高維度矩陣運(yùn)算(如卷積、變換),這是傳統(tǒng)電子芯片需要數(shù)千乃至數(shù)萬周期才能完成的,從而極大提升了計(jì)算吞吐量。
- 超低功耗與超高能效:光計(jì)算在傳輸與處理過程中幾乎無電阻熱損耗。“太極”芯片在處理特定AI任務(wù)(如圖像識別、自然語言處理的關(guān)鍵算子)時,其能效比(單位能量完成的運(yùn)算量)相比當(dāng)前最先進(jìn)的專用電子芯片(如TPU)可提升1-2個數(shù)量級,為構(gòu)建綠色數(shù)據(jù)中心奠定了硬件基礎(chǔ)。
- 與AI框架的深度融合:研究團(tuán)隊(duì)并非止步于硬件,還開發(fā)了配套的軟件棧與編譯器,能夠?qū)⒅髁鞯腁I模型(如Transformer、CNN)高效編譯并映射到光芯片上執(zhí)行,降低了使用門檻。
賦能未來:驅(qū)動AI與公共數(shù)據(jù)的深度融合
“太極”光芯片的高效能特性,使其在處理海量、多模態(tài)的人工智能公共數(shù)據(jù)方面具有天然優(yōu)勢。例如,在智慧城市管理中,需要實(shí)時分析全市的交通監(jiān)控視頻流;在生物醫(yī)藥領(lǐng)域,需快速篩選海量的分子結(jié)構(gòu)或醫(yī)學(xué)影像數(shù)據(jù)。傳統(tǒng)計(jì)算架構(gòu)受限于能耗與速度,難以實(shí)現(xiàn)實(shí)時、低成本的大規(guī)模分析。而光計(jì)算架構(gòu)能夠以極低能耗對這類高并發(fā)、高帶寬的公共數(shù)據(jù)進(jìn)行即時處理與特征提取,從而加速AI模型的訓(xùn)練與迭代,讓公共數(shù)據(jù)資源的價(jià)值得到更充分、更快速的釋放。
****
清華大學(xué)“分布式廣度光計(jì)算架構(gòu)”與“太極”光芯片的誕生,標(biāo)志著我國在下一代人工智能計(jì)算基礎(chǔ)設(shè)施的競爭中占據(jù)了重要的先發(fā)位置。它不僅僅是一塊芯片,更代表了一種打破馮·諾依曼架構(gòu)束縛、探索物理智能計(jì)算新路徑的深遠(yuǎn)嘗試。隨著后續(xù)工藝優(yōu)化與生態(tài)建設(shè),這項(xiàng)技術(shù)有望從實(shí)驗(yàn)室走向產(chǎn)業(yè)應(yīng)用,為我國的數(shù)字經(jīng)濟(jì)、智能制造、科學(xué)發(fā)現(xiàn)等領(lǐng)域提供強(qiáng)大的、綠色的算力引擎,助力全球人工智能可持續(xù)發(fā)展。