《電子技術應用》
您所在的位置:首頁 > 人工智能 > 解決方案 > 卷積神經(jīng)網(wǎng)絡的硬件轉換:什么是機器學習?——第三部分

卷積神經(jīng)網(wǎng)絡的硬件轉換:什么是機器學習?——第三部分

2023-06-14
作者:Ole Dreessen,現(xiàn)場應用工程師
來源:ADI

  摘要

  本系列文章由三部分組成,主要探討卷積神經(jīng)網(wǎng)絡(CNN)的特性和應用。CNN主要用于模式識別和對象分類。作為系列文章的第三部分,本文重點解釋如何使用硬件轉換卷積神經(jīng)網(wǎng)絡(CNN),并特別介紹使用帶CNN硬件加速器的人工智能(AI)微控制器在物聯(lián)網(wǎng)(IoT)邊緣實現(xiàn)人工智能應用所帶來的好處。系列文章的前兩篇文章為《卷積神經(jīng)網(wǎng)絡簡介:什么是機器學習?——第一部分》和《訓練卷積神經(jīng)網(wǎng)絡:什么是機器學習?——第二部分》。

  簡介

  AI應用通常需要消耗大量能源,并以服務器農(nóng)場或昂貴的現(xiàn)場可編程門陣列(FPGA)為載體。AI應用的挑戰(zhàn)在于提高計算能力的同時保持較低的功耗和成本。當前,強大的智能邊緣計算正在使AI應用發(fā)生巨大轉變。與傳統(tǒng)的基于固件的AI計算相比,以基于硬件的卷積神經(jīng)網(wǎng)絡加速器為載體的智能邊緣AI計算具備驚人的速度和強大的算力,開創(chuàng)了計算性能的新時代。這是因為智能邊緣計算能夠讓傳感器節(jié)點在本地自行決策而不受5G和Wi-Fi網(wǎng)絡數(shù)據(jù)傳輸速率的限制,為實現(xiàn)之前難以落地的新興技術和應用場景提供了助力。例如,在偏遠地區(qū),傳感器級別的煙霧/火災探測或環(huán)境數(shù)據(jù)分析已成為現(xiàn)實。這些應用支持電池供電,能夠工作很多年的時間。本文通過探討如何采用帶專用CNN加速器的AI微控制器實現(xiàn)CNN的硬件轉換來說明如何實現(xiàn)這些功能。

  采用超低功耗卷積神經(jīng)網(wǎng)絡加速器的人工智能微控制器

  MAX78000是一款有超低功耗CNN加速器的AI微控制器片上系統(tǒng), 能在資源受限的邊緣設備或物聯(lián)網(wǎng)應用中實現(xiàn)超低功耗的神經(jīng)網(wǎng)絡運算。其應用場景包括目標檢測和分類、音頻處理、聲音分類、噪聲消除、面部識別、基于心率等健康體征分析的時間序列數(shù)據(jù)處理、多傳感器分析以及預測性維護。

  圖1為MAX78000的框圖,其內核為帶浮點運算單元的Arm? Cortex?-M4F內核,工作頻率高達100 MHz。為了給應用提供足夠的存儲資源,MAX78000還配備了512 kB的閃存和128 kB的SRAM。該器件提供多個外部接口,例如I?C、SPI、UART,以及用于音頻的I?S。此外,器件還集成了60 MHz的RISC-V內核,可以作為一個智能的直接存儲器訪問(DMA)引擎從/向各個外圍模塊和存儲(包括閃存和SRAM)復制/粘貼數(shù)據(jù)。由于RISC-V內核可以對AI加速器所需的

293.JPG

  圖1.MAX78000的結構框圖

  傳感器數(shù)據(jù)進行預處理,因而Arm內核在此期間可以處于深度睡眠模式。推理結果也可以通過中斷觸發(fā)Arm內核在主應用程序中執(zhí)行操作,通過無線傳輸傳感器數(shù)據(jù)或向用戶發(fā)送通知。

  具備用于執(zhí)行卷積神經(jīng)網(wǎng)絡推理的專用硬件加速器單元是MAX7800x系列微控制器的一個顯著特征,這使其有別于標準的微控制器架構。該CNN硬件加速器可以支持完整的CNN模型架構以及所有必需的參數(shù)(權重和偏置),配備了64個并行處理器和一個集成存儲器。集成存儲器中的442 kB用于存儲參數(shù),896 kB用于存儲輸入數(shù)據(jù)。不僅存儲在SRAM中的模型和參數(shù)可以通過固件進行調整,網(wǎng)絡也可以實時地通過固件進行調整。器件支持的模型權重為1位、2位、4位或8位,存儲器支持容納多達350萬個參數(shù)。加速器的存儲功能使得微控制器無需在連續(xù)的數(shù)學運算中每次都要通過總線獲取相關參數(shù)——這樣的方式通常伴有高延遲和高功耗,代價高昂。CNN加速器可以支持32層或64層的網(wǎng)絡,具體層數(shù)取決于池化函數(shù)。每層的可編程圖像輸入/輸出大小最多為1024 × 1024像素。

  CNN硬件轉換:功耗和推理速度比較

  CNN推理是一項包含大型矩陣線性方程運算的復雜計算任務。Arm Cortex-M4F微控制器的強大能力可以使得CNN推理在嵌入式系統(tǒng)的固件上運行。但這種方式也有一些缺點:在微控制器上運行基于固件的CNN推理時,計算命令和相關參數(shù)都需要先從存儲器中檢索再被寫回中間結果,這會造成大量功耗和時延。

  表1對三種不同解決方案的CNN推理速度和功耗進行了比較。所用的模型基于手寫數(shù)字識別訓練集MNIST開發(fā),可對視覺輸入數(shù)據(jù)中的數(shù)字和字母進行分類以獲得準確的輸出結果。為確定功耗和速度的差異,本文對三種解決方案所需的推理時間進行了測量。

292.JPG

  方案一使用集成Arm Cortex-M4F處理器的MAX32630進行推理,其工作頻率為96 MHz。方案二使用MAX78000的CNN硬件加速器進行推理,其推理速度(即數(shù)據(jù)輸入與結果輸出之間的時間)比方案一加快了400倍,每次推理所需的能量也僅為方案一的1/1100。方案三對MNIST網(wǎng)絡進行了低功耗優(yōu)化,從而最大限度地降低了每次推理的功耗。雖然方案三推理結果的準確性從99.6%下降到了95.6%,但其速度快了很多,每次推理只需0.36 ms,推理功耗降也低至僅1.1μW。兩節(jié)AA堿性電池(總共6 Wh能量)可以支持應用進行500萬次的推理(忽略系統(tǒng)其它部分的功耗)。

  這些數(shù)據(jù)說明了硬件加速器的強大計算能力可以大大助益無法利用或連接到連續(xù)電源的應用場景。MAX78000就是這樣一款產(chǎn)品,它支持邊緣AI處理,無需大量功耗和網(wǎng)絡連接,也無需冗長的推理時間。

  MAX78000 AI微控制器的使用示例

  MAX78000支持多種應用,下面本文圍繞部分用例展開討論。其中一個用例是設計一個電池供電的攝像頭,需要能檢測到視野中是否有貓出現(xiàn),并能夠通過數(shù)字輸出打開貓門允許貓進入房屋。

  圖2為該設計的示例框圖。在本設計中,RISC-V內核會定期開啟圖像傳感器并將圖像數(shù)據(jù)加載到MAX78000的CNN加速器中。如果系統(tǒng)判斷貓出現(xiàn)的概率高于預設的閾值,則打開貓門然后回到待機模式。

295.JPG

  圖2.智能寵物門框圖

  開發(fā)環(huán)境和評估套件

  邊緣人工智能應用的開發(fā)過程可分為以下幾個階段:

  第一階段:AI——網(wǎng)絡的定義、訓練和量化

  第二階段:Arm固件——將第一階段生成的網(wǎng)絡和參數(shù)導入C/C++應用程序,創(chuàng)建并測試固件

  開發(fā)過程的第一階段涉及建模、訓練和評估AI模型等環(huán)節(jié)。此階段開發(fā)人員可以利用開源工具,例如PyTorch 和TensorFlow。MAX78000的GitHub網(wǎng)頁也提供全面的資源幫助用戶在考慮其硬件規(guī)格的同時使用PyTorch構建和訓練AI網(wǎng)絡。網(wǎng)頁也提供一些簡單的AI網(wǎng)絡和應用,例如面部識別(Face ID),供用戶參考。

  圖3顯示了采用PyTorch進行AI開發(fā)的典型過程。首先是對網(wǎng)絡進行建模。必須注意的是,MAX7800x微控制器并非都配置了支持所有PyTorch數(shù)據(jù)操作的相關硬件。因此,必須首先將ADI公司提供的ai8x.py文件包含在項目中,該文件包含MAX78000所需的PyTorch模塊和運算符?;诖丝梢赃M入下一步驟構建網(wǎng)絡,使用訓練數(shù)據(jù)對網(wǎng)絡進行訓練、評估和量化。這一步驟會生成一個檢查點文件,其中包含用于最終綜合過程的輸入數(shù)據(jù)。最后一步是將網(wǎng)絡及其參數(shù)轉換為適合CNN硬件加速器的形式。值得注意的是,雖然任何PC(筆記本、服務器等)都可用于訓練網(wǎng)絡,但如果沒有CUDA顯卡,訓練網(wǎng)絡可能會花費很長的時間——即使對于小型網(wǎng)絡來說也有可能需要幾天甚至幾周的時間。

  開發(fā)過程的第二階段是通過將數(shù)據(jù)寫入CNN加速器并讀取結果的機制來創(chuàng)建應用固件。

294.JPG

  圖3.AI開發(fā)過程

  第一階段創(chuàng)建的文件通過#include指令集成到C/C++項目中。微控制器的開發(fā)環(huán)境可使用Eclipse IDE和GNU工具鏈等開源工具。ADI公司提供的軟件開發(fā)套件(Maxim Micros SDK (Windows))也已經(jīng)包含了所有開發(fā)必需的組件和配置,包括外設驅動以及示例說明,幫助用戶簡化應用開發(fā)過程。

  成功通過編譯和鏈接的項目可以在目標硬件上進行評估。ADI開發(fā)了兩種不同的硬件平臺可供選用:圖4為MAX78000EVKIT,圖5為MAX78000FTHR,一個稍小的評估板。每個評估板都配有一個VGA攝像頭和一個麥克風。

290.JPG

  圖4.MAX78000評估套件

291.JPG

  圖5.MAX78000FTHR評估套件

  結論

  以前,AI應用必須以昂貴的服務器農(nóng)場或FPGA為載體,并消耗大量能源。現(xiàn)在,借助帶專用CNN加速器的MAX78000系列微控制器,AI應用依靠單組電池供電就可以長時間運行。MAX78000系列微控制器在能效和功耗方面的性能突破大大降低了邊緣AI的實現(xiàn)難度,使得新型邊緣AI應用的驚人潛力得以釋放。欲了解更多信息,請訪問超低功耗人工智能(AI) MCU。

  參考資料

  “研討會2 - 邊緣人工智能:Maxim Integrated MAX78000 AI加速器的實踐介紹”。ADI公司。

  視頻系列:了解人工智能。ADI公司。

  PyTorch徽標經(jīng)知識共享 版權歸屬-相同方式共享 4.0 國際 公共許可證許可使用。



更多精彩內容歡迎點擊==>>電子技術應用-AET<<

mmexport1621241704608.jpg

本站內容除特別聲明的原創(chuàng)文章之外,轉載內容只為傳遞更多信息,并不代表本網(wǎng)站贊同其觀點。轉載的所有的文章、圖片、音/視頻文件等資料的版權歸版權所有權人所有。本站采用的非本站原創(chuàng)文章及圖片等內容無法一一聯(lián)系確認版權者。如涉及作品內容、版權和其它問題,請及時通過電子郵件或電話通知我們,以便迅速采取適當措施,避免給雙方造成不必要的經(jīng)濟損失。聯(lián)系電話:010-82306118;郵箱:aet@chinaaet.com。