NanoEdge AI Library for anomaly detection (AD)

1 What is NanoEdge AI Library?

NanoEdge™ AI Library is an Artificial Intelligence (AI) static library originally developed by Cartesiam, for embedded C software running on Arm® Cortex® microcontrollers.

When embedded on microcontrollers, it gives them the ability to easily "learn" and "understand" sensor patterns, by themselves, without the need for the user to have additional skills in Mathematics, Machine Learning, or data science.

The NanoEdge AI static library is the code that contains an AI model (for example, as a bundle of signal treatment, Machine Learning model, or optimally tuned hyperparameters). It is designed to gather knowledge incrementally during a learning phase to become able to detect potential anomalous machine behaviors, and possibly predict them.

Info white.png Information
  • The library is not available in the free version of NanoEdgeAIStudio.
  • However, if you select one of the authorized development boards available, you can download a specific library, which works only on such targets.

2 Install / Getting started

The main functions available via the library are:

initialize() run first before learning/detecting, or to reset the knowledge of the library/emulator
set_sensitivity() adjust the preset, internal detection sensitivity (does not affect learning, only returned similarity scores)
learn() start a number of learning iterations (to establish an initial knowledge, or enrich an existing one)
detect() start a number detection iterations (inference), once a minimum knowledge base has been established


Warning DB.png Important

When building a smart device, the final features heavily depend on the way those functions are called. It is entirely up to the developer to design relevant learning and detection strategies, depending on the project specificities and constraints.

NanoEdgeAI ild.png

For example for a hypothetical machine, one possible strategy is to:

  1. initialize the model
  2. establish an initial knowledge base by calling learn() every minute for 24 hours on that machine
  3. switch to the inference mode by calling detect() 10 times every hour (and averaging the returned scores), each day
  4. blink a LED and ring alarms whenever detect() returns any anomaly (average score < 90%)
  5. run another learning cycle to enrich the existing knowledge, if the temperature rises above 60°C (and the machine is still OK)
  6. send a daily report (average number of anomalies per hour, with date, time, and machine ID for instance) using Bluetooth® or LoRa®.

In summary, those smart functions can be triggered by external data (for example from sensors, buttons, to account for and adapt to environment changes).
The scores returned by the smart functions can trigger all kinds of behaviors on your device.
The possibilities are endless.

2.1 Static library

The static NanoEdge AI Library is only available via NanoEdge AI Studio in its full version.

  • In NanoEdge AI Studio, after obtaining a library, click Compile (on the "Deploy" screen, which follows the "Optimize and Benchmark" and "Emulator" screens)
  • Open the .zip obtained
  • Select and copy the static library libneai.a
  • Link this static library to your project code

2.2 NanoEdge AI Library functions

2.2.1 Initialization

uint8_t NanoEdgeAI_initialize(void);

Initialization can be run at the beginning to initialize the model and/or later to initialize a new model and reset all knowledge.
Returns 0 in case of success.

Info white.png Information
  • If you are using a limited library for authorized development boards, make sure that you board is compatible. Otherwise, error code 123 is returned.

2.2.2 Learning

uint8_t NanoEdgeAI_learn(float data_input[]);

This function is used to learn patterns in your data. It can be used at any time, in the beginning to build the original knowledge base of the AI model, but also later, as an additional learning phase to complement the existing knowledge.

  • Input:
float data_input[], the length of the data is BUFFER_SIZE * NB_AXES .
  • Output:***
1 means success (the input signal has been learned), other values mean that learning has failed.
Info white.png Information
The learning function can be called:
  1. initially, before any inference, to establish some reference knowlege base
  2. subsequently, whenever needed, to complete the existing knowledge and enrich it (for exemple, to take into account some new nominal environment conditions)
Warning white.png Warning

NanoEdge AI Library uses float data types instead of int. If you are using int data types, convert (cast) them into float.

2.2.3 Detection

uint8_t NanoEdgeAI_detect(float data_input[]);

This function returns returns a similarity percentage, measure of the mathematical distance between the incoming signal and the existing knowledge, learned by the library.

  • Input:
float data_input[], the length of the data is BUFFER_SIZE * NB_AXES.
  • output:
The percentage of similarity [0-100] between the new signal and learned patterns.
"100" means completely similar, and "0" completely dissimilar.
Info white.png Information
  • The recommended threshold percentage is 90. Values under this threshold reveal a behavior that differs from the usual behavior learned by the AI model. This threshold can be defined by the user, depending on the final application sensitivity requirements.
  • If you are using a limited library for authorized development boards, make sure that you board is compatible. Otherwise, detection returns error code 123.

2.2.4 Setting sensitivity

void NanoEdgeAI_set_sensitivity(float sensitivity);

This function sets the sensitivity of the model in detection mode. It can be tuned at any time without having to go through a new learning phase. This sensitivity has no influence on the knowledge acquired during the learning steps. It only plays a role in the detection step, by influencing the similarity percentages that are returned by the detect function (acts as a linear scaler of the preset internal sensitivity).

  • Input: float sensitivity
  • Output: None
Info white.png Information
  • The default sensitivity value is 1. A sensitivity value between 0 and 1 (excluded) decreases the sensitivity of the model, while a value in between 1 and 100 increases it. We recommend increasing or decreasing sensitivity by steps of 0.1.
  • Sensitivity 1.1 - 100 tends to decrease the percentages of similarity returned (the algorithm is more sensitive to perturbations), while sensitivity 0 - 0.9 tends to increase them (the algorithm is less sensitive to perturbations).

2.2.5 Getting sensitivity

float NanoEdgeAI_get_sensitivity(void);

This function returns the current sensitivity of the model.
The recommended/default sensitivity is 1. However, this value can be defined by the user depending on his application sensitivity requirements.

  • Input: None
  • Output: The sensitivity of model.

2.2.6 Getting status

uint8_t NanoEdgeAI_get_status(void);

This function returns the current status of the library.

  • Input: None
  • Output: The status of the library.
0: the library is successfully deployed and is ready to be used.
123: the library is being used on a board that is not authorized (only for libraries that are limited to partner/authorized development boards).

2.3 Backing up and restoring the library knowledge

2.3.1 Creating backups

When using NanoEdge AI Library, knowledge is created on the go. It means that after each learning iteration, the Machine Learning model is incrementally getting richer and richer.

For performance reasons, this knowledge is saved into the microcontroller RAM. Since RAM is volatile, better copy this knowledge into non-volatile memory and prevent its loss (for example, in case of power failure).

NanoEdge AI Library attributes a specific memory section called .neai for the knowledge variables (model hyperparameters). To use it, you need to create a memory section in your linker script according to your microcontroller architecture.


Here is a general outline of the procedure:

  1. The idea is to create a section called .neai (for NanoEdge AI) in your RAM, using the linker script that is specific to the board / MCU that you use (it is a .ld file).

    For example, using Arm® Mbed™ for a NUCLEO-L432KC (Cortex®-M4), modify the file STM32L432XX.ld and add the following RAM section:
    .neai 0x0000000020004000  :                                                                            
    {                                                                                                      
       KEEP(*(.neai))          /* keep my variable even if not referenced */
    } >RAM
    

    Make sure that the RAM address (in this example, 0x0000000020004000) corresponds to a valid section according to the board and MCU technical documentation. Here it corresponds to the beginning of the RAM0 (Data Space) section, see below:

    NanoEdgeAI memory size.png
  2. Then, create two new functions in your main code (containing the NanoEdge AI functions and to be compiled and programmed to the MCU): one function to save the knowledge from the RAM to the Flash memory (dump), and another one to restore the knowledge from the Flash memory to the RAM (load).

    How to create these two functions, which write to the RAM or Flash memory, highly depends on the board and MCU used (see their technical documentation).
  3. For the dump function (RAM => Flash memory), indicate the RAM address to start reading from (the .neai section address, such as 0x0000000020004000 in the example above), the Flash memory address to start writing to, and the total size of the knowledge saved in the NanoEdge AI Library (libneai.a).

    You can obtain the size of this knowledge by means of the arm-none-eabi-size tool:
    $ arm-none-eabi-size -A libneai.a | grep neai
    
  4. For the load function (Flash memory => RAM), it is the reverse operation. You need the Flash memory address to start reading from, the RAM address to start writing to (.neai section), and the size of the knowledge to restore.

2.3.2 Restoring backups

To restore a previously saved backup, copy the entire knowledge from the Flash memory back into the RAM .neai section, as explained in step #4 of the section "2.3.1 Creating backups" above.

Then, you are able to use your NanoEdge AI Library functions again.

2.4 Example "Hello World!"

Header file: NanoEdgeAI.h

Example of NanoEdge AI Library header file:

This snippet is provided AS IS, and by taking it, you agree to be bound to the license terms that can be found here for the component: Application.
/**
* Copyright (c) 2021 STMicroelectronics.
* All rights reserved.
*
* This software component is licensed by ST under Ultimate Liberty license
* SLA0044, the "License"; You may not use this file except in compliance with
* the License. You may obtain a copy of the License at:
*                             www.st.com/SLA0044
*
******************************************************************************
*/

/* Includes */
#include <stdint.h>

/* Define */
#define NEAI_ID "5d5fd41ec8e0327c934"
#define NB_AXES 3
#define BUFFER_SIZE 256

/* Function prototypes */
#ifdef __cplusplus
extern "C" {
#endif
   uint8_t NanoEdgeAI_initialize(void);
   uint8_t NanoEdgeAI_learn(void);
   uint8_t NanoEdgeAI_detect(void);
   void NanoEdgeAI_set_sensitivity(float sensitivity);
   float NanoEdgeAI_get_sensitivity(void);
   uint8_t NanoEdgeAI_get_status(void);
#ifdef __cplusplus
}
#endif

Main program: main.c
This program must be completed by the user (depending for instance on the applications or the desired features).

This snippet is provided AS IS, and by taking it, you agree to be bound to the license terms that can be found here for the component: Application.
/**
* Copyright (c) 2021 STMicroelectronics.
* All rights reserved.
*
* This software component is licensed by ST under Ultimate Liberty license
* SLA0044, the "License"; You may not use this file except in compliance with
* the License. You may obtain a copy of the License at:
*                             www.st.com/SLA0044
*
******************************************************************************
*/

/* Includes ------------------------------------------------------------------*/
#include "NanoEdgeAI.h"

/* Private define ------------------------------------------------------------*/
#define LEARNING_ITERATIONS 100 /* e.g. this value is set by user */  

/* Private variables defined by user -----------------------------------------*/
float data_buffer[BUFFER_SIZE * NB_AXES];

/* Private function prototype defined by user --------------------------------*/
/*
 * @brief  Collect data process
 *
 * This function is defined by user, depends on applications and sensors
 *
 * @param  sample_buffer: [in, out] buffer of sample values 
 * 
 * @retval None
 */

void get_buffer(float sample_buffer[]) 
{
   /* USER BEGIN */

   /* USER END */
}

/* ---------------------------------------------------------------------------*/
int main()
{
   /* Initialization --------------------------------------------------------*/
   uint8_t error_code = NanoEdgeAI_initialize();

   if (error_code != 0) {
      /* This happen if the library is used in an unauthorized board.
      /* Free version only
   }

   /* Learning --------------------------------------------------------------*/
   int iteration = 0;
   uint8_t learn_status = 0;
   /* e.g.: Stopping criterion for learning process. */
   /* This stopping criterion could be modified by user */
   while (iteration < LEARNING_ITERATIONS) {
      get_buffer(data_buffer);
      learn_status = NanoEdgeAI_learn(data_buffer);
      iteration++;
   }  

   /* Set sensitivity of AI model (optional, Input value is set by user)-----*/
   /* NanoEdgeAI_set_sensitivity(1.1); */

   /* Detection -------------------------------------------------------------*/
   uint8_t similarity_percentage  = 0;
   while (true) {
      get_buffer(data_buffer);
      similarity_percentage = NanoEdgeAI_detect(data_buffer);
      /* BEGIN USER */
      /* e.g.: Trigger function depending on similarity_percentage 
      /*       (print output, blink LED, ring alarm...).
      /* END USER */
   }     
}

3 Resources

Documentation
All NanoEdge AI Studio documentation is available here.

Tutorials
Step-by-step tutorials, to use NanoEdge AI Studio to build a smart device from A to Z.