中科院推出國際上首個稀疏深度學(xué)習(xí)處理器
中科院計算所與寒武紀(jì)公司提出了國際上首個稀疏深度學(xué)習(xí)處理器Cambricon-X,相關(guān)工作于近日被計算機體系結(jié)構(gòu)領(lǐng)域頂級國際會議MICRO 2016(International Symposium on Microarchitecture)接收。該處理器由中國科大少年班陳云霽、陳天石兄弟開創(chuàng)。
深度學(xué)習(xí)是一類借鑒生物的多層神經(jīng)網(wǎng)絡(luò)處理模式所發(fā)展起來的智能處理技術(shù)?;谏疃葘W(xué)習(xí)的圍棋程序AlphaGo甚至已經(jīng)達到了職業(yè)棋手的水平。稀疏深度學(xué)習(xí)可大幅度削減神經(jīng)網(wǎng)絡(luò)中連接數(shù)量,因此被業(yè)界廣泛采用。但其計算和訪存模式非常不規(guī)則,給相應(yīng)的芯片設(shè)計帶來了巨大的挑戰(zhàn)。寒武紀(jì)團隊針對稀疏深度神經(jīng)網(wǎng)絡(luò)特殊的計算模式和訪存特性,提出采用一維稀疏表示、專用的數(shù)據(jù)訪問引擎和動態(tài)計算調(diào)度策略來實現(xiàn)稀疏深度學(xué)習(xí)的專用處理器Cambricon-X,能高效處理各種類型的稀疏深度神經(jīng)網(wǎng)絡(luò)(包括卷積神經(jīng)網(wǎng)絡(luò)和遞歸神經(jīng)網(wǎng)絡(luò)等)。在TSMC 65nm工藝下,Cambricon-X峰值性能可達0.5Tops/s,面積開銷僅為6.38平方毫米,稀疏深度學(xué)習(xí)速度相較于主流高端GPU提升性能 10倍,能耗僅為其3.4%。相關(guān)工作已于2016年1月申請國內(nèi)和PCT專利,此后UCSB也參與了Cambricon-X的進一步學(xué)術(shù)探索。
此前,中科院計算所提出了國際上首個深度學(xué)習(xí)處理器結(jié)構(gòu)(與法國Inria合作完成)、國際上首個多核深度學(xué)習(xí)處理器結(jié)構(gòu)(與法國Inria合作完成),研制了國際上首個深度學(xué)習(xí)處理器芯片,提出了國際上首個深度學(xué)習(xí)指令集,多次獲得計算機體系結(jié)構(gòu)頂級會議最佳論文獎。Cambricon-X的工作進一步夯實了中科院與寒武紀(jì)在深度學(xué)習(xí)處理器領(lǐng)域的國際領(lǐng)導(dǎo)性地位。