IBM推全新AI模擬芯片，能效提升14倍！相變存儲器由旺宏供應

8月23日，IBM研究實(shí)驗室在《自然》期刊雜志上公布了其最新研究成果，研發(fā)出了一種全新的人工智能（AI）模擬芯片，能效可達傳統數字計算機芯片的14倍，可大幅降低AI計算的功耗。

該文章摘要顯示，當前擁有數十億個(gè)參數的 AI 模型可在一系列任務(wù)中實(shí)現較高精度，但也凸顯出傳統通用處理器（包含圖形處理器、中央處理器等）效能低下的問(wèn)題。對此，IBM研究團隊提出“模擬內存計算”的方案，通過(guò)在自身的存儲器上并行執行矩陣-向量乘法，以提供更強的能效。

IBM的研究團隊基于該方案開(kāi)發(fā)出了一個(gè) 14nm 模擬芯片，利用34個(gè)大型相變存儲器（PCM）陣列，結合了數模轉換輸入、模擬外圍電路、模數轉換輸出和大規模并行二維網(wǎng)格路由。每個(gè)14nm芯片上可編碼3500萬(wàn)個(gè)PCM，在每權重對應2-PCMs的方案中，可容納1700萬(wàn)個(gè)參數。將這些芯片組合在一起，便能夠像數字芯片一樣有效地處理真實(shí)AI用例的實(shí)驗。

在測試環(huán)節中，IBM研究團隊先后使用谷歌語(yǔ)音檢測和Librispeech語(yǔ)音識別數據集來(lái)分別測試該芯片語(yǔ)言處理能力的效率。

在關(guān)鍵詞語(yǔ)音檢測方面，IBM團隊提出了一種卷積神經(jīng)網(wǎng)絡(luò )架構，并在包含12個(gè)關(guān)鍵字的谷歌語(yǔ)音命令數據集上進(jìn)行訓練。團隊采用了架構更簡(jiǎn)單的FC（全連接）網(wǎng)絡(luò )結構，最終達到了86.14%的識別精度，且提交速度比MLPerf目前最佳情況快7倍。該模型使用硬件感知訓練在GPU上進(jìn)行訓練，然后部署在團隊的模擬AI芯片上。

在規模更大的 Librispeech 語(yǔ)音識別數據集上，利用5個(gè)模擬AI芯片組合在一起，運行RNN-T（循環(huán)神經(jīng)網(wǎng)絡(luò )轉換器）模型，以逐個(gè)字母地轉錄語(yǔ)音內容。該系統包含5個(gè)芯片上1.4億個(gè)PCM設備的4500萬(wàn)個(gè)權重，能夠采集人們說(shuō)話(huà)的音頻并以非常接近數字硬件設置的精度進(jìn)行轉錄。該實(shí)驗最終達到9.258%的單詞錯誤率，能量效率達6.704TOPS/W（萬(wàn)億次操作每秒/瓦），比MLPerf目前最佳能效提高了14倍。

值得注意的是，IBM的這款AI芯片所搭載的相變存儲器，是由IBM與旺宏共同開(kāi)發(fā)的。

據了解，旺宏與IBM合作淵源深厚，早在2004年，雙方就已經(jīng)有策略合作聯(lián)盟關(guān)系，并共同投入相變存儲器開(kāi)發(fā)十余年，雖然初期還有不少同業(yè)者加入，但隨著(zhù)時(shí)間流逝，不少廠(chǎng)商已經(jīng)退出了共同研發(fā)合作，目前旺宏是IBM相變存儲器唯一合作伙伴，并取得特定AI相關(guān)授權。

目前旺宏與IBM的相變存儲器共同開(kāi)發(fā)計劃，以三年為一期，期滿(mǎn)后雙方視狀況再簽訂新共同開(kāi)發(fā)合約。雙方早已嗅到AI帶來(lái)的商機無(wú)窮，將相變存儲器合作方向瞄準AI應用，最近一次合約于2021年10月簽訂，旺宏還特別通過(guò)重大信息公布了相關(guān)內容。

旺宏當時(shí)公告指出，該公司將繼續參與IBM相變存儲器共同開(kāi)發(fā)計劃，并取得特定模擬AI技術(shù)授權，該次合約從2022年元月22日起至2025年元月21日止，旺宏將有保密義務(wù)并分擔研發(fā)費用，同時(shí)遵守美國出口管制法令等義務(wù)，具體目的為持續投入先進(jìn)技術(shù)開(kāi)發(fā)，對旺宏技術(shù)及競爭力有正面影響。

業(yè)界人士分析，相變存儲器具備速度快、非揮發(fā)性、低功率等特性，讓高密度存儲器可做為儲存應用，填補AI服務(wù)器所需的DRAM、NAND當中的價(jià)格及性能落差。

對于與IBM合作相變存儲器的最新進(jìn)展，旺宏于29日表示，持續順利進(jìn)行中，鎖定AI等先進(jìn)應用發(fā)展。

編輯：芯智訊-林子