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