天天看點

UIUC陳德銘教授:「萬能晶片」FPGA與深度學習

人工智能的風潮從技術一路蔓延到硬體,讓「晶片」成為今年行業裡盛極一時的主題。人們關注通用晶片領域裡 CPU 和 GPU 不斷重新整理的基準(benchmark),更對專用晶片(ASIC)領域不同場景下不斷問世的解決方案表示出了空前的熱情。

而衆所周知,在專用晶片與通用晶片中間,還有一個更為靈活,也更為神秘的領域:FPGA。無論是英特爾天價的收購還是微軟與 IBM 雄心勃勃的計劃,都讓人對其更加好奇。而「萬能晶片」的名稱,以及多樣化的職責範圍:它可以是智能手機裡不起眼的一個小元件,也可以是數千美金一塊的開發闆,也讓人對其真面目更加疑惑。

FPGA 與深度學習的關系究竟是什麼?它适合加速何種類型的運算?适合應用于何種場景?6 月,機器之心采訪了美國伊利諾伊大學香槟分校電氣與計算機工程系(UIUC ECE)教授、深度學習算法硬體加速解決方案創業公司 Inspirit IoT 創始人陳德銘,與他聊了聊「萬能晶片」FPGA 在深度學習領域的用法。陳教授于近期宣布兼任國内計算機視覺領域創業公司觸景無限的首席科學家。

以下為采訪實錄。

機器之心:從算法的角度,FPGA 适合于加速哪些運算?

FPGA 的優勢在于可定制化的并行。它可以為了特定算法專門定制邏輯單元。例如,算法中需要完成一個開三次方的運算,在通用晶片上需要一長串指令實作它,但是在 FPGA 上可以為這個特定運算設計一個單元。

定制化的一個「極限」的例子是,如果能夠把神經網絡中的參數全部二值化、特征圖(feature map)也全部二值化,那麼甚至可以不用乘法器,而是用異或門(XOR gate)完成計算,會有異常強大的表現。即使不全部二值化,因為全部二值化會讓準确度明顯下降,隻要合理使用各種低位寬的運算,仍可以讓 FPGA 大放異彩。比如在我們最近和 IBM 合作的 ELB-NN 項目中,我們可以在功耗低于 5 瓦的情形下,讓神經網絡在 FPGA 上的算力達到 10.3 TOPS,同時依然維持非常高的準确度。

另一個例子是微軟的 Brainwave FPGA 項目,通過使用特殊的 8-bit 的浮點運算,可以在高容量的 FPGA 上達到 40 TOPS。

值得一提的是,LSTM 等神經網絡每一步操作都很相似,這種整體運算過程是序列的算法也非常适用于用 FPGA 進行加速。FPGA 非常擅長處理流資料,可以設計流水線機制,使得在運算過程中不需要将中間資料存入記憶體而是直接送到下一步進行操作。

機器之心:相比于其他晶片,FPGA 在實作深度學習模型時有哪些結構帶來的優勢?

FPGA 的引腳比一般的晶片多,同時它的邏輯單元可以完全根據算法要求來重組進而産生定制化的資料通路(datapath),是以非常适合處理大批量流資料。它不像 GPU 一樣需要反複調取片外存儲中的資料,理想狀态時,隻要資料一次性流進去再流出來,算法就完成了,是以,FPGA 非常适合具有低延遲需求的任務。

另外 FPGA 的速度快、功率低的特點也讓它在能耗方面相比于 CPU、GPU 也有很大優勢。

比如我們在 FPGA 上實作了長期循環卷積網絡(Long-term Recurrent Convolutional Network),它的速度 ( latency ) 比 GPU 快 3.1 倍,FPGA 的功率比 GPU 小 5.6 倍,最終整體能耗降低 17 倍左右。

機器之心:從行業的角度,FPGA 都能夠用于哪些機器學習與深度學習任務的加速?

在終端,有些自動駕駛公司的車上系統中就采用了 FPGA 完成一些實時檢測識别任務。例如,我曾經代表 UIUC 與小鵬汽車的副總裁、機器學習專家谷俊麗女士交流,她提到,FPGA 可以憑借信号處理和低能耗方面的優勢,成為一個可靠的第二系統,在汽車的第一系統出現失誤的時候,作為安全模式介入進來。IoT 方面的應用就更多了,無論是安防還是智能家居,FPGA 都會有機會進入市場。

在雲端,也有結合 FPGA 的巨大空間。之前,微軟的 Project Catapult 就搭建起了大規模的 FPGA 雲伺服器。現在 Catapult 正在運作搜尋和資料存儲方面的運算,同時,他們也在積極地思考如何把機器學習運算部署在上面,包括是否能将 FPGA 作為其 Azure 雲的後端支援。亞馬遜 AWS 的 FPGA 雲計算已經很有規模了,并且一直在大力推動之中。最近,國内創業公司深鑒也開始結合華為的雲端 FPGA 來進行深度學習算法的加速,特别在基于 LSTM 的語音識别上非常有優勢。

機器之心:能否介紹一下使用高層次綜合(HLS)自動生成寄存器級(RTL)實作與手工用硬體描述語言(HDL)進行實作的差別?

對于深度學習相關算法來說,HLS 是有優勢的。

當你閱讀用高層語言描述的深度學習模型時,你會發現模型是有規律的。比如,卷積操作在 C 語言裡就是一個六層的嵌套循環。此時我就可以針對這個循環開發一個 IP 核,确定如何進行流水線設計(pipelining),如何配置設定存儲等,然後再将 IP 核用在 FPGA 上,進而讓神經網絡中的所有卷積層來共享 IP 核即可。

此時設計的優化要點是 IP 核的大小、是否需要多個 IP 核并行運算等。這是一個非常适用 HLS 的場景,因為 HLS 可以自動化地、快速地探索不同的解決方案。如果采用 RTL,則需要很多工程師,組成多個團隊,每個團隊根據工程師過去的設計經驗着重探索一個方向,這是一個成本非常高的過程。

RTL 是一種描述硬體的空間的關系和邏輯時序等邏輯關系的硬體描述語言。這是一門難學的語言,能夠熟練掌握的從業者也不多,開發和測試都很慢。誇張點說,如果 HLS 開發周期以天和周來計算,那麼 RTL 通常是以月和年。

從最終結果角度講,RTL 完全手工調制的解決方案是很可能比 HLS 自動生成的方案效果好的。我們曾經做過比較,在深度學習的情境下,HLS 和 RTL 方法的差距在 10% 左右。

是以,問題在于是否願意花 3-5 倍的時間獲得額外 10% 的效果。

機器之心:在簡化 FPGA 開發過程這一任務上,研究者做了哪些工作?

我們的團隊開發了一個研究工具,是一個 C 語言的模闆庫,能夠根據 Caffe 和 TensorFlow 的代碼自動生成 C 語言的表達。然後研究者可以将 C 語言的代碼再交給 HLS 工具生成 RTL 代碼。

C 語言是軟體工程師普遍掌握的語言,這樣的工具讓軟體工程師也具有 FPGA 開發能力,而且讓開發者更專注于算法,而不用過多考慮底層硬體。

而且,這個過程中的中間結果都可以導出,專業人員也可以在自動生成的基礎上再進行優化,總而言之,整個開發的效率都得到了極大的提高。

這隻是舉一個例子。還有其它很多的研究人員在做相關的提升深度學習設計效率和品質的各類工具和平台。工業和學術屆的都有。學術屆比如 UCLA 的叢京生老師,以及 Cornell 的張志汝老師等。工業屆比較領先的是賽靈思和英特爾自己設計的一些工具。

機器之心:各個大公司在 FPGA 的應用方面都進行了哪些工作?拓展市場的思路都有哪些?

IBM 和賽靈思正在 Power 系列上進行合作,緻力于在 FPGA 上加更多的存儲,在最新的原型機上,把存儲加到了 64G,相比于市面上現在常見的最大 8 或 16G,極大擴充了資料帶寬,來進行和大資料相關的處理。

英特爾收購了 Altera,并且緻力于幫助 Altera 進行編譯等工具的開發。這是一個 FPGA 商品化的思路,就像 GPU 一樣,說不定未來的 PC 裡就會有一個 FPGA 協處理器在 CPU 旁邊完成工作。或者在晶片層面上,CPU 片上就有一塊 FPGA。總而言之,一旦工具做好,整合的機會會有很多。英特爾在智能城市、智慧制造方面都有很多的構想,并且希望借由 FPGA 實作。

再就是微軟。它的思路主要就是雲計算,除了 Bing 之外,也希望在進行機器學習任務的 Azure 雲中使用 FPGA。

機器之心:能否介紹一下 Inspirit IoT 通過哪些産品實作了深度學習算法在 FPGA 上的加速?

UIUC陳德銘教授:「萬能晶片」FPGA與深度學習

我們公司主要有三個産品,按照層次分别是 DNN 優化器、高層次綜合加速編譯器(HLS compiler),以及整體 FPGA 解決方案。

DNN 優化器的目标是定制一個 FPGA 友好的模型,把訓練好的模型中對于推理來說備援的部分去掉。例如,訓練時使用的浮點數可以被替換為定點數,訓練時每層有非常多的單元,但是在推理的時候可以進行剪枝(Pruning),等等。DNN 優化器就是以各種高層架構的模型為基礎,根據硬體的特性和場景的特性進行自動的優化。

高層次綜合編譯器讓開發者得以繞開寫 Verilog 硬體描述語言,将 C、 C++、CUDA 等代碼直接編成 FPGA,旨在降低硬體開發的門檻,提高軟體工程師的生産力。我們的特色是這個編譯器可以自然結合深度學習算法中的各種 IP 核,進而可以最好的優化各種深度學習算法在 FPGA 上的高速及高品質的實作。

除此之外,我們也有嵌入 FPGA 的開發闆,用于聲音相關的應用場景。其上嵌入了 FPGA 和 4 個麥克風,可以檢測聲音來源、提高聲音品質,也可以和攝像頭等其他裝置連用。例如,捕捉到車輛撞擊聲、尖叫聲等特定聲音之後,能夠調轉攝像頭,收錄相關的視訊等,希望獲得一個「1+1>3」的效果。

特别鳴謝:北京大學高性能計算機研究中心博士生 王碩

本文為機器之心原創,轉載請聯系本公衆号獲得授權。

繼續閱讀