本文重點解釋如何使用硬件轉換卷積神經網絡(CNN),并特別介紹使用帶CNN硬件加速器的人工智能(AI)微控制器在物聯網(IoT)邊緣實現人工智能應用所帶來的好處。
AI應用通常需要消耗大量能源,并以服務器農場或昂貴的現場可編程門陣列(FPGA)為載體。AI應用的挑戰在于提高計算能力的同時保持較低的功耗和成本。當前,強大的智能邊緣計算正在使AI應用發生巨大轉變。與傳統的基于固件的AI計算相比,以基于硬件的卷積神經網絡加速器為載體的智能邊緣AI計算具備驚人的速度和強大的算力,開創了計算性能的新時代。這是因為智能邊緣計算能夠讓傳感器節點在本地自行決策而不受5G和Wi-Fi網絡數據傳輸速率的限制,為實現之前難以落地的新興技術和應用場景提供了助力。例如,在偏遠地區,傳感器級別的煙霧/火災探測或環境數據分析已成為現實。這些應用支持電池供電,能夠工作很多年的時間。本文通過探討如何采用帶專用CNN加速器的AI微控制器實現CNN的硬件轉換來說明如何實現這些功能。
采用超低功耗卷積神經網絡加速器的人工智能微控制器
MAX78000是一款有超低功耗CNN加速器的AI微控制器片上系統,能在資源受限的邊緣設備或物聯網應用中實現超低功耗的神經網絡運算。其應用場景包括目標檢測和分類、音頻處理、聲音分類、噪聲消除、面部識別、基于心率等健康體征分析的時間序列數據處理、多傳感器分析以及預測性維護。
圖1為MAX78000的框圖,其內核為帶浮點運算單元的Arm Cortex-M4F內核,工作頻率高達100 MHz。為了給應用提供足夠的存儲資源,MAX78000還配備了512 kB的閃存和128 kB的SRAM。該器件提供多個外部接口,例如I2C、SPI、UART,以及用于音頻的I2S。此外,器件還集成了60 MHz的RISC-V內核,可以作為一個智能的直接存儲器訪問(DMA)引擎從/向各個外圍模塊和存儲(包括閃存和SRAM)復制/粘貼數據。由于RISC-V內核可以對AI加速器所需的
圖1.MAX78000的結構框圖
傳感器數據進行預處理,因而Arm內核在此期間可以處于深度睡眠模式。推理結果也可以通過中斷觸發Arm內核在主應用程序中執行操作,通過無線傳輸傳感器數據或向用戶發送通知。
具備用于執行卷積神經網絡推理的專用硬件加速器單元是MAX7800x系列微控制器的一個顯著特征,這使其有別于標準的微控制器架構。該CNN硬件加速器可以支持完整的CNN模型架構以及所有必需的參數(權重和偏置),配備了64個并行處理器和一個集成存儲器。集成存儲器中的442 kB用于存儲參數,896 kB用于存儲輸入數據。不僅存儲在SRAM中的模型和參數可以通過固件進行調整,網絡也可以實時地通過固件進行調整。器件支持的模型權重為1位、2位、4位或8位,存儲器支持容納多達350萬個參數。加速器的存儲功能使得微控制器無需在連續的數學運算中每次都要通過總線獲取相關參數——這樣的方式通常伴有高延遲和高功耗,代價高昂。CNN加速器可以支持32層或64層的網絡,具體層數取決于池化函數。每層的可編程圖像輸入/輸出大小最多為1024 × 1024像素。
CNN硬件轉換:功耗和推理速度比較
CNN推理是一項包含大型矩陣線性方程運算的復雜計算任務。Arm Cortex-M4F微控制器的強大能力可以使得CNN推理在嵌入式系統的固件上運行。但這種方式也有一些缺點:在微控制器上運行基于固件的CNN推理時,計算命令和相關參數都需要先從存儲器中檢索再被寫回中間結果,這會造成大量功耗和時延。
表1對三種不同解決方案的CNN推理速度和功耗進行了比較。所用的模型基于手寫數字識別訓練集MNIST開發,可對視覺輸入數據中的數字和字母進行分類以獲得準確的輸出結果。為確定功耗和速度的差異,本文對三種解決方案所需的推理時間進行了測量。
表1.手寫數字識別的CNN推理時間和推理功耗,基于MNIST數據集
方案一使用集成Arm Cortex-M4F處理器的MAX32630進行推理,其工作頻率為96 MHz。方案二使用MAX78000的CNN硬件加速器進行推理,其推理速度(即數據輸入與結果輸出之間的時間)比方案一加快了400倍,每次推理所需的能量也僅為方案一的1/1100。方案三對MNIST網絡進行了低功耗優化,從而最大限度地降低了每次推理的功耗。雖然方案三推理結果的準確性從99.6%下降到了95.6%,但其速度快了很多,每次推理只需0.36 ms,推理功耗降也低至僅1.1 μW。兩節AA堿性電池(總共6 Wh能量)可以支持應用進行500萬次的推理(忽略系統其它部分的功耗)。
這些數據說明了硬件加速器的強大計算能力可以大大助益無法利用或連接到連續電源的應用場景。MAX78000就是這樣一款產品,它支持邊緣AI處理,無需大量功耗和網絡連接,也無需冗長的推理時間。
MAX78000 AI微控制器的使用示例
MAX78000支持多種應用,下面本文圍繞部分用例展開討論。其中一個用例是設計一個電池供電的攝像頭,需要能檢測到視野中是否有貓出現,并能夠通過數字輸出打開貓門允許貓進入房屋。
圖2為該設計的示例框圖。在本設計中,RISC-V內核會定期開啟圖像傳感器并將圖像數據加載到MAX78000的CNN加速器中。如果系統判斷貓出現的概率高于預設的閾值,則打開貓門然后回到待機模式。
圖2.智能寵物門框圖
開發環境和評估套件
邊緣人工智能應用的開發過程可分為以下幾個階段:
第一階段:AI——網絡的定義、訓練和量化
第二階段:Arm固件——將第一階段生成的網絡和參數導入C/C++應用程序,創建并測試固件
開發過程的第一階段涉及建模、訓練和評估AI模型等環節。此階段開發人員可以利用開源工具,例如 PyTorch 和 TensorFlow。MAX78000的GitHub網頁也提供全面的資源幫助用戶在考慮其硬件規格的同時使用PyTorch構建和訓練AI網絡。網頁也提供一些簡單的AI網絡和應用,例如面部識別(Face ID),供用戶參考。
圖3顯示了采用PyTorch進行AI開發的典型過程。首先是對網絡進行建模。必須注意的是,MAX7800x微控制器并非都配置了支持所有PyTorch數據操作的相關硬件。因此,必須首先將ADI公司提供的ai8x.py文件包含在項目中,該文件包含MAX78000所需的PyTorch模塊和運算符。基于此可以進入下一步驟構建網絡,使用訓練數據對網絡進行訓練、評估和量化。這一步驟會生成一個檢查點文件,其中包含用于最終綜合過程的輸入數據。最后一步是將網絡及其參數轉換為適合CNN硬件加速器的形式。值得注意的是,雖然任何PC(筆記本、服務器等)都可用于訓練網絡,但如果沒有CUDA顯卡,訓練網絡可能會花費很長的時間——即使對于小型網絡來說也有可能需要幾天甚至幾周的時間。
開發過程的第二階段是通過將數據寫入CNN加速器并讀取結果的機制來創建應用固件。第一階段創建的文件通過#include指令集成到C/C++項目中。微控制器的開發環境可使用Eclipse IDE和GNU工具鏈等開源工具。ADI公司提供的軟件開發套件(Maxim Micros SDK (Windows))也已經包含了所有開發必需的組件和配置,包括外設驅動以及示例說明,幫助用戶簡化應用開發過程。
圖3.AI開發過程
成功通過編譯和鏈接的項目可以在目標硬件上進行評估。ADI開發了兩種不同的硬件平臺可供選用:圖4為 MAX78000EVKIT ,圖5為 MAX78000FTHR ,一個稍小的評估板。每個評估板都配有一個VGA攝像頭和一個麥克風。
圖4.MAX78000評估套件
圖5.MAX78000FTHR評估套件
結論
以前,AI應用必須以昂貴的服務器農場或FPGA為載體,并消耗大量能源。現在,借助帶專用CNN加速器的MAX78000系列微控制器,AI應用依靠單組電池供電就可以長時間運行。MAX78000系列微控制器在能效和功耗方面的性能突破大大降低了邊緣AI的實現難度,使得新型邊緣AI應用的驚人潛力得以釋放。
-
亞德諾
+關注
關注
6文章
4680瀏覽量
15929
原文標題:如何采用帶專用CNN加速器的AI微控制器實現CNN的硬件轉換
文章出處:【微信號:analog_devices,微信公眾號:analog_devices】歡迎添加關注!文章轉載請注明出處。
發布評論請先 登錄
相關推薦
評論