|author||alexander <firstname.lastname@example.org>||Fri Mar 26 21:42:19 2021 +0000|
|committer||Kshitij Sisodia <email@example.com>||Mon Mar 29 16:29:55 2021 +0100|
Opensource ML embedded evaluation kit Change-Id: I12e807f19f5cacad7cef82572b6dd48252fd61fd
This repository is for building and deploying Machine Learning (ML) applications targeted for Arm® Cortex®-M and Arm® Ethos™-U NPU. To run evaluations using this software, we suggest using an MPS3 board or a fixed virtual platform (FVP) that supports Ethos-U55 software fast model. Both environments run a combination of the new Arm® Cortex™-M55 processor and the Arm® Ethos™-U55 NPU.
The purpose of the evaluation kit is to allow the user to develop software and test the performance of the Ethos-U55 NPU and Cortex-M55 CPU. The Ethos-U55 NPU is a new class of machine learning(ML) processor, specifically designed to accelerate computation for ML workloads in constrained embedded and IoT devices. The product is optimized to execute mathematical operations efficiently that are commonly used in ML algorithms, such as convolutions or activation functions.
The evaluation kit adds value by providing ready to use ML applications for the embedded stack. As a result, you can experiment with the already developed software use cases and create your own applications for Cortex-M CPU and Ethos-U NPU. The example application at your disposal and the utilized models are listed in the table below.
|ML application||Description||Neural Network Model|
|Image classification||Recognize the presence of objects in a given image||Mobilenet V2|
|Keyword spotting(KWS)||Recognize the presence of a key word in a recording||DS-CNN-L|
|Automated Speech Recognition(ASR)||Transcribe words in a recording||Wav2Letter|
|KWS and ASR||Utilise Cortex-M and Ethos-U to transcribe words in a recording after a keyword was spotted||DS-CNN-L Wav2Letter|
|Anomaly Detection||Detecting abnormal behavior based on a sound recording of a machine||Coming soon|
|Generic inference runner||Code block allowing you to develop your own use case for Ethos-U55 NPU||Your custom model|
The above use cases implement end-to-end ML flow including data pre-processing and post-processing. They will allow you to investigate embedded software stack, evaluate performance of the networks running on Cortex-M55 CPU and Ethos-U55 NPU by displaying different performance metrics such as inference cycle count estimation and results of the network execution.
The evaluation kit is based on the Arm® Corstone™-300 reference package. Arm® Corstone™-300 helps you build SoCs quickly on the Arm® Cortex™-M55 and Arm® Ethos™-U55 designs. Arm® Corstone™-300 design implementation is publicly available on an Arm MPS3 FPGA board, or as a Fixed Virtual Platform of the MPS3 development board.
The Ethos-U NPU software stack is described here.
All ML use cases, albeit illustrating a different application, have common code such as initializing the Hardware Abstraction Layer (HAL). The application common code can be run on x86 or Arm Cortex-M architecture thanks to the HAL. For the ML application-specific part, Google® TensorFlow™ Lite for Microcontrollers inference engine is used to schedule the neural networks models executions. TensorFlow Lite for Microcontrollers is integrated with the Ethos-U55 driver and delegates execution of certain operators to the NPU or, if the neural network model operators are not supported on NPU, to the CPU. CMSIS-NN is used to optimise CPU workload execution with int8 data type. Common ML application functions will help you to focus on implementing logic of your custom ML use case: you can modify only the use case code and leave all other components unchanged. Supplied build system will discover new ML application code and automatically include it into compilation flow.
To run an ML application on the Cortex-M and Ethos-U55 NPU, please, follow these steps:
To get familiar with these steps, you can follow the quick start guide.
For more details: