隨著越來(lái)越多的聯(lián)網(wǎng)設(shè)備需要越來(lái)越多的帶寬來(lái)完成遠(yuǎn)程辦公和云計(jì)算等任務(wù)，管理所有用戶共享的有限無(wú)線頻譜將變得極具挑戰(zhàn)性。
工程師們正在利用人工智能來(lái)動(dòng)態(tài)管理可用的無(wú)線頻譜，以期減少延遲并提升性能。但大多數(shù)用于分類和處理無(wú)線信號(hào)的人工智能方法都耗電嚴(yán)重，而且無(wú)法實(shí)時(shí)運(yùn)行。
現(xiàn)在，麻省理工學(xué)院的研究人員開(kāi)發(fā)了一種專為無(wú)線信號(hào)處理而設(shè)計(jì)的新型人工智能硬件加速器。他們的光學(xué)處理器能夠以光速執(zhí)行機(jī)器學(xué)習(xí)計(jì)算，在納秒級(jí)時(shí)間內(nèi)對(duì)無(wú)線信號(hào)進(jìn)行分類。
該光子芯片的速度比最佳數(shù)字方案快約100倍，同時(shí)信號(hào)分類的收斂準(zhǔn)確率達(dá)到約95%。新型硬件加速器還具有可擴(kuò)展性和靈活性，因此可用于各種高性能計(jì)算應(yīng)用。同時(shí)，它比數(shù)字AI硬件加速器更小、更輕、更便宜、更節(jié)能。
該設(shè)備在未來(lái)的 6G 無(wú)線應(yīng)用中尤其有用，例如通過(guò)使無(wú)線調(diào)制格式適應(yīng)不斷變化的無(wú)線環(huán)境來(lái)優(yōu)化數(shù)據(jù)速率的認(rèn)知無(wú)線電。
通過(guò)使邊緣設(shè)備能夠?qū)崟r(shí)執(zhí)行深度學(xué)習(xí)計(jì)算，這款新型硬件加速器可以為信號(hào)處理以外的許多應(yīng)用提供顯著的加速。例如，它可以幫助自動(dòng)駕駛汽車(chē)對(duì)環(huán)境變化做出瞬間反應(yīng)，或使智能起搏器能夠持續(xù)監(jiān)測(cè)患者的心臟健康狀況。
“許多應(yīng)用可以通過(guò)能夠分析無(wú)線信號(hào)的邊緣設(shè)備來(lái)實(shí)現(xiàn)。我們?cè)谡撐闹刑岢龅膬?nèi)容可能為實(shí)時(shí)可靠的人工智能推理開(kāi)辟許多可能性。這項(xiàng)工作只是一項(xiàng)可能產(chǎn)生巨大影響的開(kāi)端。”麻省理工學(xué)院電氣工程與計(jì)算機(jī)科學(xué)系教授、量子光子學(xué)與人工智能小組和電子研究實(shí)驗(yàn)室 (RLE) 首席研究員、該論文的資深作者 Dirk Englund 說(shuō)道。
與他共同參與論文撰寫(xiě)的還有：第一作者羅納德·Davis三世（Ronald Davis III），24屆博士生；前麻省理工學(xué)院博士后、現(xiàn)任南加州大學(xué)助理教授的陳在軍（Zaijun Chen）；以及RLE客座科學(xué)家、NTT Research高級(jí)科學(xué)家瑞安·哈默利（Ryan Hamerly）。這項(xiàng)研究發(fā)表在《科學(xué)進(jìn)展》（Science Advances）雜志上。
光速處理
用于無(wú)線信號(hào)處理的先進(jìn)數(shù)字 AI 加速器將信號(hào)轉(zhuǎn)換為圖像，并通過(guò)深度學(xué)習(xí)模型進(jìn)行分類。雖然這種方法非常準(zhǔn)確，但深度神經(jīng)網(wǎng)絡(luò)的計(jì)算密集型特性使其不適用于許多時(shí)間敏感的應(yīng)用。
光學(xué)系統(tǒng)可以通過(guò)利用光來(lái)編碼和處理數(shù)據(jù)，從而加速深度神經(jīng)網(wǎng)絡(luò)，而且其能耗也比數(shù)字計(jì)算更低。然而，研究人員一直在努力最大限度地提高通用光學(xué)神經(jīng)網(wǎng)絡(luò)在信號(hào)處理中的性能，同時(shí)確保光學(xué)設(shè)備的可擴(kuò)展性。
通過(guò)開(kāi)發(fā)一種專門(mén)用于信號(hào)處理的光學(xué)神經(jīng)網(wǎng)絡(luò)架構(gòu)，研究人員將其稱為乘法模擬頻率變換光學(xué)神經(jīng)網(wǎng)絡(luò)（MAFT-ONN），正面解決了這個(gè)問(wèn)題。
MAFT-ONN 通過(guò)對(duì)所有信號(hào)數(shù)據(jù)進(jìn)行編碼并在無(wú)線信號(hào)數(shù)字化之前在所謂的頻域內(nèi)執(zhí)行所有機(jī)器學(xué)習(xí)操作來(lái)解決可擴(kuò)展性問(wèn)題。
研究人員設(shè)計(jì)的光學(xué)神經(jīng)網(wǎng)絡(luò)能夠同時(shí)執(zhí)行所有線性和非線性運(yùn)算。深度學(xué)習(xí)需要這兩種類型的運(yùn)算。
由于這種創(chuàng)新設(shè)計(jì)，整個(gè)光學(xué)神經(jīng)網(wǎng)絡(luò)每層只需要一個(gè) MAFT-ONN 設(shè)備，而其他方法則需要為每個(gè)單獨(dú)的計(jì)算單元或“神經(jīng)元”配備一個(gè)設(shè)備。
Davis說(shuō)：“我們可以將 10,000 個(gè)神經(jīng)元安裝到單個(gè)設(shè)備上，并一次性計(jì)算出必要的乘法。”
研究人員利用一種名為光電倍增的技術(shù)實(shí)現(xiàn)了這一目標(biāo)，該技術(shù)顯著提高了效率。此外，該技術(shù)還能幫助他們構(gòu)建一個(gè)光學(xué)神經(jīng)網(wǎng)絡(luò)，該網(wǎng)絡(luò)可以輕松擴(kuò)展，無(wú)需額外開(kāi)銷，并增加額外的層數(shù)。
以納秒為單位的結(jié)果
MAFT-ONN 以無(wú)線信號(hào)作為輸入，處理信號(hào)數(shù)據(jù)，并將信息傳遞給邊緣設(shè)備執(zhí)行后續(xù)操作。例如，通過(guò)對(duì)信號(hào)的調(diào)制進(jìn)行分類，MAFT-ONN 可以使設(shè)備自動(dòng)推斷信號(hào)類型以提取其攜帶的數(shù)據(jù)。
研究人員在設(shè)計(jì) MAFT-ONN 時(shí)面臨的最大挑戰(zhàn)之一是確定如何將機(jī)器學(xué)習(xí)計(jì)算映射到光學(xué)硬件。
“我們不能直接使用現(xiàn)成的普通機(jī)器學(xué)習(xí)框架。我們必須對(duì)其進(jìn)行定制，使其適應(yīng)硬件，并弄清楚如何利用物理原理，以便它能夠執(zhí)行我們想要的計(jì)算，”Davis說(shuō)。
當(dāng)他們?cè)谀M中測(cè)試其架構(gòu)的信號(hào)分類時(shí)，光學(xué)神經(jīng)網(wǎng)絡(luò)單次即可達(dá)到 85% 的準(zhǔn)確率，并通過(guò)多次測(cè)量快速收斂到 99% 以上的準(zhǔn)確率。MAFT-ONN 僅需約 120 納秒即可完成整個(gè)過(guò)程。
“測(cè)量時(shí)間越長(zhǎng)，精度就越高。由于 MAFT-ONN 以納秒為單位計(jì)算推理，因此無(wú)需犧牲太多速度即可獲得更高的精度。”Davis補(bǔ)充道。
雖然最先進(jìn)的數(shù)字射頻設(shè)備可以在微秒內(nèi)完成機(jī)器學(xué)習(xí)推理，但光學(xué)設(shè)備可以在納秒甚至皮秒內(nèi)完成。
展望未來(lái)，研究人員希望采用所謂的多路復(fù)用方案，以便執(zhí)行更多計(jì)算并擴(kuò)大 MAFT-ONN 的規(guī)模。他們還希望將這項(xiàng)工作擴(kuò)展到更復(fù)雜的深度學(xué)習(xí)架構(gòu)，以運(yùn)行 Transformer 模型或 LLM。
這項(xiàng)工作的部分資金來(lái)自美國(guó)陸軍研究實(shí)驗(yàn)室、美國(guó)空軍、麻省理工學(xué)院林肯實(shí)驗(yàn)室、日本電報(bào)電話公司和國(guó)家科學(xué)基金會(huì)。
