隨著人工智慧的落地和大規模應用,AI晶片也成為了常見的晶片品類。AI晶片相比傳統晶片來說,主要的競爭優勢就在於高算力和高能效比。高算力是指能夠比傳統晶片更快地完成AI計算,而高能效比則是指能比傳統晶片用更少的能量完成計算。
要理解記憶體牆,還需要從傳統的馮諾伊曼架構說起。馮諾伊曼架構是計算機的經典體系結構,同時也是之前處理器晶片的主流架構。在馮諾伊曼架構中,計算與記憶體是分離的單元:計算單元根據從記憶體中讀取資料,計算完成後存回記憶體。
存內計算:翻越記憶體牆
存內計算,顧名思義就是直接在儲存內做計算。其具體實現方式有若干條技術路徑。首先,最直接的就是在現有儲存的基礎上做一些電路上的改動。這類實現方法最簡單,例如2018年MIT Chandrakasan研究組在ISSCC上發表的研究就是這類技術的例子。該研究中,存內計算的主要用途是加速卷積計算,而卷積計算從數學上可以展開成帶權重的累加計算,或者說是多個數的加權平均。因此,存內計算的做法是把權重(1-bit)儲存在SRAM中,輸入資料經過DAC成為模擬訊號,並根據SRAM中的對應權重相乘,然後在模擬域做平均,最後由ADC讀出成為數字訊號。這類存內計算往往只是修改現有儲存的譯碼器/放大器模組,並不涉及儲存器件的重新設計,優勢是比較容易和現有工藝整合,但是缺點是能夠帶來的效能提升較為有限,尤其是基於SRAM的方案,一方面SRAM的整合度是有限的,另一方面單位元精度的權重也成為了其應用的限制。
另一種存內計算的技術路線是透過引入新的儲存器件來完成存內計算。如果說前一種存內計算的概念是“在SoC裡引入特殊的SRAM並在SRAM附近高效完成計算”,那麼這一種存內計算就是直接“在儲存陣列內完成計算”了。這種基於新儲存器件的存內計算往往要利用新儲存器件的一些特徵,並且在一塊特殊的儲存陣列內整合計算功能,在要做計算的時候主處理模組只需要給儲存陣列傳送輸入資料,過若干時鐘週期之後儲存陣列會把計算好的結果返回給主處理模組。與傳統的馮諾伊曼架構相比,傳統馮諾伊曼架構中處理器晶片給記憶體發請求,並讀回資料;而在這類存內計算晶片的計算正規化中,主處理模組給儲存陣列傳送請求和輸入資料,而儲存陣列則直接返回計算結果,這樣就省去了計算過程中主處理器和記憶體之間的大量資料搬運。通常來說,存內計算在處理人工智慧相關任務的時候,會把神經網路權重儲存在陣列中,而主處理模組直接給儲存陣列發去神經網路的輸入即可開始計算。
近年來,存內計算已經逐漸成為業界和學界公認的趨勢。拿半導體積體電路領域的“奧林匹克”——ISSCC為例,從2018年開始ISSCC開始設立與存內計算相關的專門session並收錄五篇相關論文,此後存內計算在ISSCC上的相關論文錄用勢頭一直不減,到2020年的ISSCC與存內計算相關的論文數量上升到了七篇。除此之外,半導體器件領域的頂級會議IEDM今年也給了存內計算足夠重視,有三個專門的session共二十多篇相關論文。有趣的是,ISSCC和IEDM上相關存內計算的論文正好對應了前文所說的存內計算的兩種技術路線——ISSCC對應從電路側做技術革新,而IEDM則主要對應器件方向的技術更新換代,透過引入新的儲存器件並基於其新特性來開發高效能的存內計算。其中,IEDM中顯示的正規化轉換更引人關注。今年,IEDM的一大看點就是對於摩爾定律到頭之後下一步方向的預測,有一個專門的panel session更是直接以“摩爾定律已死,但是AI永生”為名字,可見業界對於後摩爾定律時代的發展,最看好的是基於AI的新器件。而在AI相關的新器件/新正規化中,存內計算可謂是最有希望的一種,由此可見今年IEDM的關於後摩爾定律的主題和錄用數十篇存內計算相關的論文之間存在著緊密的聯絡。
IEDM上的來自中國的論文可能成為解決存內計算瓶頸的關鍵
在今年的IEDM上,我們就看到了這樣的技術突破。一家初創公司閃億半導體,與浙江大學、北京大學、華虹宏力合作發表的論文《Programmable Linear RAM: A New Flash Memory-based Memristor for Artificial Synapses and Its Application to Speech Recognition System》恰恰解決了這個矛盾。該論文巧妙地利用電晶體線上性區的特性製備了新型儲存器PLRAM,併成功地設計出了一款可以用在移動終端的超低功耗存內計算晶片,並實現了8-bit精度操作。
該研究讓我們看到了中國半導體行業的崛起,因為在IEDM這樣強手如林的頂尖半導體器件會議上發表文章本身就是對相關技術的肯定。更可喜的是,該研究已經在閃億進行商業化,我們認為閃億擁有的技術首先能克服存內計算的計算精度和應用場景之間的矛盾,可以把存內計算低功耗的優勢發揮到極致,而同時其高精度計算又保證了可以相容大多數神經網路。同時,閃億選擇的IoT和可穿戴式市場也是一個正在蓬勃發展的市場,這些市場非常適合用全球領先的新技術去撬動新的應用,從而讓存內計算真正落地走向千家萬戶。我們希望能看到更多像閃億這樣的高精尖半導體技術商業化的案例,而當市場上出現眾多這樣的充滿活力的高新技術半導體公司時,中國半導體的春天也就到了。