如何實現嵌入式視覺和機器學習

目前嵌入式視覺領域最熱門的話題之一就是機器學習。機器學習涵蓋多個行業大趨勢，不僅在嵌入式視覺 (EV) ，而且在工業物聯網 (IIoT) 和云計算中均發揮著極為顯赫的作用。對不熟悉機器學習的人來說，很多時候機器學習是通過神經網絡創建和訓練來實現的。神經網絡一語極為寬泛，包含相當數量的截然不同的子類別。這些子類別的名稱一般用于識別被實現的網絡的具體類型。這些網絡均在大腦皮層上建模，在大腦皮層中每個神經元接收輸入、處理輸入并將其傳達給另一個神經元。因此神經元一般由輸入層、多個隱藏內部層和一個輸出層組成。

圖1：簡單的神經網絡

在最簡單的層面上，神經元獲得輸入，施加權重給輸入，然后在加權輸入總和上執行傳遞函數。結果隨后傳遞給隱藏層中的另一個層，或傳遞給輸出層。將一級的輸出傳遞給另一級且不構成循環的神經網絡被稱為前饋神經網絡 (FNN)，而那些有反饋、內含定向循環的神經網絡則被稱為遞歸神經網絡 (RNN)。在眾多機器學習應用中極為常用的神經網絡是深度神經網絡 (DNN)。這類神經網絡擁有多個隱藏層，能實現更復雜的機器學習任務。為確定每層使用的權重和偏差值，需要對神經網絡進行訓練。在訓練過程中，該神經網絡施加有一定數量的正確輸入和錯誤輸入，并使用誤差函數教授網絡所需的性能。訓練深度神經網絡可能需要龐大的數據集來正確訓練所需性能。

機器學習最重要的應用之一是嵌入式視覺領域，各類系統正在從視覺使能系統演進為視覺引導自動化系統。嵌入式視覺應用與其他更簡單的機器學習應用的區別在于它們采用二維輸入格式。因此，在機器學習實現方案中，通過使用稱為卷積神經網絡 (CNN) 的網絡結構，因為它們能夠處理二維輸入。CNN 是一類前饋網絡，內置多個卷積層和子采樣層以及一個單獨的全連通網絡，以執行最終分類。鑒于 CNN 的復雜性，它們也歸屬深度學習類別。在卷積層中，輸入圖像被細分為一系列重疊的小模塊。在進行進一步的子采樣和其它階段之前，該卷積的結果先通過激活層創建激活圖，然后應用到最終的全連通網絡上。CNN 網絡的具體定義因實現的網絡架構而異，但它一般會包含至少下列元：

卷積 – 用于識別圖像中的特征

修正線性單元（reLU）- 用于在卷積后創建激活圖的激活層

最大池化 – 在層間進行子采樣

全連通 - 執行最終分類

這些元中每一個元的權重通過訓練決定，同時 CNN 的優勢之一在于訓練網絡相對容易。通過訓練生成權重需要龐大的圖像集，其中既有需要檢測的對象，也有偽圖像。這樣能讓我們為 CNN 創建所需的權重。由于訓練流程中所涉及的處理要求，訓練流程一般運行在提供高性能計算的云處理器上。

框架

機器學習是一個復雜的課題，尤其是在每次不得不從頭開始，定義網絡、網絡架構和生成訓練算法的時候。為幫助工程師實現網絡和訓練網絡，有一些行業標準框架可供使用，例如 Caffe 和 Tensor Flow。Caffe 框架為機器學習開發人員提供各種庫、模型和 C++ 庫內的預訓練權重，同時提供 Python 和 Matlab 綁定。該框架能讓用戶無需從頭開始即能創建網絡并訓練網絡，以開展所需的運算。為便于重復使用,Caffe 用戶能通過 model zoo 共享自己的模型。Model Zoo 提供多種能根據所需的專門任務實現和更新的模型。這些網絡和權重定義在 prototxt 文件中。在用于機器學習環境時，prototxt 文件是用于定義推斷引擎的文件。

圖2：定義網絡的 Prototxt 文件實例

實現嵌入式視覺和機器學習

基于可編程邏輯的解決方案，例如異構賽靈思 All Programmable Zynq -7000 SoC（片上系統）和 Zynq UltraScale+ MPSoC 等多處理器片上系統 (MPSoC) 越來越廣泛地用于嵌入式視覺應用。這些器件將可編程邏輯 (PL) 架構與處理系統 (PS) 中的高性能ARM 內核完美組合在一起。這種組合形成的系統擁有更快的響應速度，極為靈活便于未來修改，并且提供了高能效解決方案。對許多應用來說低時延決策與響應循環極為重要。例如視覺引導自主機器人，響應時間對避免給人員造成傷害、給環境造成破壞至關重要。縮短響應時間的具體方法是使用可編程邏輯實現視覺處理流水線和使用機器學習推斷引擎實現機器學習。在這方面使用可編程邏輯，與傳統解決方案相比可減少系統瓶頸問題。在使用基于 CPU/GPU 的方法時，運算每一階段都需要使用外部 DDR，因為圖像不能在有限內部緩存內的功能間傳遞。可編程邏輯方法使用內部 ARM 按需提供緩存，允許采用流媒體方法。避免在 DDR 內存儲中間元不僅可降低圖像處理的時延，而且還能降低功耗，甚至提高確定性，因為無需與其他系統資源共享訪問。

圖 3：可編程邏輯實現的好處

機器學習推斷引擎實現方案中使用的數值表示系統也對機器學習性能有重要作用。機器學習應用正越來越多地使用更高效的降精度定點數值系統，例如 INT8 表達式。與傳統的浮點 32 (FP32) 方法相比，使用定點降精度數值系統不會造成顯著的精度下降。因為與浮點相比，定點數學在實現難度上也明顯更低，轉而采用 INT8 能在一些實現中提供更高效、更快速的解決方案。使用定點數值系統對在可編程邏輯解決方案中的實現方案而言相當理想，reVISION 在可編程邏輯中能與 INT8 表達式協同工作。這些 INT8 表達式方便在可編程邏輯中使用專門的 DSP 模塊。在使用相同的內核權重時，這些 DSP 模塊架構能實現最多兩個并發 INT8 乘法累加運算供執行。這樣不僅能提供高性能實現方案，而且還能夠降低功耗。只要采用，可編程邏輯的靈活性也還便于實現進一步降精度定點數值表達系統。