NanoEdge AI Library for 1-class classification (1CC)

1. What is NanoEdge AI Library for 1-class classification (outlier detection)?

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

When embedded on microcontrollers, it gives them the ability to easily "identify" outliers from 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 for 1-class classification is the code that contains an AI model (for example, as a bundle of signal treatment, Machine Learning model, and optimally tuned hyperparameters) designed to identify if a sensor pattern is inside the class or outside the class (outlier/anomaly). The class under consideration is defined by the user in NanoEdge AI Studio (NanoEdgeAIStudio), also called the Studio, and are used during the training process of the AI model.

2. Install / Getting started

The main functions available via the library are:

oneclass_init() run first before the classification process to load the knowledge
oneclass() run a classification iteration to identify if the pattern is inside or outside the class (inference)

2.1. Static library

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

2.2. NanoEdge AI Library functions

Most NanoEdge AI function return the status of the library in the following enum, neai_state: 

enum neai_state { 
    NEAI_OK = 0,
    NEAI_INIT_FCT_NOT_CALLED = 123,
    NEAI_BOARD_ERROR,
    NEAI_KNOWLEDGE_BUFFER_ERROR,
    NEAI_NOT_ENOUGH_CALL_TO_LEARNING,
    NEAI_UNKNOWN_ERROR};

Here are the possible statuses:

NEAI_OK: the library is working as expected
NEAI_INIT_FCT_NOT_CALLED: the learn or detect function has been called without running the init function before. Initialize your library.
NEAI_BOARD_ERROR: the board detected is not authorized. For instance, it may happen if you are trying to use a library (for instance obtained from the free version of NanoEdge AI Studio) with a non-supported board.
NEAI_KNOWLEDGE_BUFFER_ERROR: the knowledge loaded is not compatible with this library. Make sure that the knowledge being used is the one obtained with this exact library.
NEAI_NOT_ENOUGH_CALL_TO_LEARNING: this is a fail-safe to prevent users from running an insufficient (only one or a few) number of iterations of the learning function. Run more learning iterations.
NEAI_UNKNOWN_ERROR: there is an unknown error with the library.
Info white.png Information

Although it is present in all NanoEdge AI header files, the NEAI_NOT_ENOUGH_CALL_TO_LEARNING status is only relevant for anomaly detection libraries.

2.2.1. Initialization

enum neai_state neai_oneclass_init(const float knowledge_buffer[]);

Initialization must be called at the beginning to load the knowledge.

  • Input:
const float knowledge_buffer[], this buffer is defined in the header file knowledge.h provided in the .zip file containing the static NanoEdge AI Library.
  • Output:
the neai_state enum (NEAI_OK == 0, in case of success).

For more details on the output of the initialization function, refer to the header file NanoEdgeAI.h provided in the .zip file containing the static NanoEdge AI Library, or in the code examples below.

2.2.2. 1-class classification

enum neai_state neai_oneclass(float input_buffer[], uint8_t *is_outlier);

This function returns the result of outlier detection, whether or not the detected buffer is an outlier.

  • Input:
float input_buffer[], the length of the buffer is DATA_INPUT_USER * AXIS_NUMBER.
  • Output:
1 if the input pattern is detected as an outlier (outside the class considered), and 0 if it is not an outlier (it belongs to the class).
The neai_state enum (NEAI_OK == 0, in case of success).
Info white.png Information

The uint8_t is_outlier variable must be defined prior to calling the extrapolation function, and pointed to using &is_outlier when passed as an argument (see code example below).

Warning white.png Warning

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

2.3. Example "Hello World!"

Header files:

NanoEdgeAI.h and knowledge.h (provided in the .zip file that you download by clicking Compile in NanoEdge AI Studio (on the "Deploy" screen after obtaining a library)

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) 2023, STMicroelectronics

All rights reserved.

Redistribution and use in source and binary forms, with or without modification, are permitted provided that
the following conditions are met:

* Redistributions of source code must retain the above copyright notice, this list of conditions and the
following disclaimer.

* Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the
following disclaimer in the documentation and/or other materials provided with the distribution.

* Neither the name of the copyright holders nor the names of its contributors may be used to endorse or promote
products derived from this software without specific prior written permission.

*THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER / OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.*
*/

#ifndef __NANOEDGEAI_H__
#define __NANOEDGEAI_H__

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

/* Define */

#define AXIS_NUMBER 3
#define DATA_INPUT_USER 512

#ifndef __NEAI_STATE__
#define __NEAI_STATE__
enum neai_state {
NEAI_OK = 0,
NEAI_INIT_FCT_NOT_CALLED = 123,
NEAI_BOARD_ERROR,
NEAI_KNOWLEDGE_BUFFER_ERROR,
NEAI_NOT_ENOUGH_CALL_TO_LEARNING, //This is a fail-safe to prevent users from learning one or even no signals.
NEAI_MINIMAL_RECOMMENDED_LEARNING_DONE,
NEAI_UNKNOWN_ERROR};
#endif

/* Function prototypes */
#ifdef __cplusplus
extern "C" {
#endif
    /**
     * @brief  Initialization must be called at the beginning to load the knowledge.
     *         This buffer is defined in the header file knowledge.h provided in the .zip file
     * @retval NEAI_OK in case of success.
     */
    enum neai_state neai_oneclass_init(const float knowledge_buffer[]);

    /**
     * @brief  The result of outlier detection, whether or not the detected buffer is an outlier.
     * @param  data_input[] [IN]: Signal to be classified AXIS_NUMBER * DATA_INPUT_USER
     * @param  is_outlier[] [OUT]: 1 if the input pattern is detected as an outlier
     *                             0 if it is not
     * @retval NEAI_OK in case of success.
     */
    enum neai_state neai_oneclass(float data_input[], uint8_t *is_outlier);
#ifdef __cplusplus
}
#endif

#endif

/* =============
Here some sample declaration added in your main program for the use of the NanoEdge AI library.
You can directly copy this declaration or modify the names.
* WARNING: respect the size of the buffer.

uint8_t oneclass_result = 0; // Point to the result (see argument of neai_oneclass fct)
float input_user_buffer[DATA_INPUT_USER * AXIS_NUMBER]; // Buffer of input values
*/
Warning DB.png Important
  • The input buffer collected from your sensor is already declared in the header file NanoEdgeAI.h, float input_user_buffer[DATA_INPUT_USER * AXIS_NUMBER];, feel free to comment this line, and then rename and declare your input buffer in the main program.
  • The knowledge buffer knowledge is declared in the header file knowledge.h.


Main program: main.c

This program must be completed by the user (depending 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) 2020, STMicroelectronics

All rights reserved.

Redistribution and use in source and binary forms, with or without modification, are permitted provided that
the following conditions are met:

* Redistributions of source code must retain the above copyright notice, this list of conditions and the
  following disclaimer.

* Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the
  following disclaimer in the documentation and/or other materials provided with the distribution.

* Neither the name of the copyright holders nor the names of its contributors may be used to endorse or promote
  products derived from this software without specific prior written permission.

*THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER / OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.*
*/
/**
  **************************************************************************
  * Demo: NanoEdge AI process to include in main program body
  *
  * @note  This program must be completed and customized by the user
  **************************************************************************
  */

/* Includes --------------------------------------------------------------------*/
#include "NanoEdgeAI.h"
#include "knowledge.h"
/* Private define --------------------------------------------------------------*/
/* Private variables defined by user -------------------------------------------*/
float input_user_buffer[DATA_INPUT_USER * AXIS_NUMBER]; // Buffer of input values

/* Private function prototypes 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
 * @note   If AXIS_NUMBER = 3 (cf NanoEdgeAI.h), the buffer must be
 *         ordered as follow:
 *         [x0 y0 z0 x1 y1 z1 ... xn yn zn], where xi, yi and zi
 *         are the values for x, y and z axes, n is equal to
 *         DATA_INPUT_USER (cf NanoEdgeAI.h)
 */
void fill_buffer(float sample_buffer[])
{
    /* USER BEGIN */
    /* USER END */
}
/* -----------------------------------------------------------------------------*/
int main(void)
{
    /* Initialization ----------------------------------------------------------*/
    enum neai_state error_code = neai_oneclass_init(knowledge);
    if (error_code != NEAI_OK) {
        /* This happens if the knowledge does not correspond to the library or if the library works into a not supported board. */
    }
    /* Oneclass ----------------------------------------------------------------*/
    uint8_t oneclass_result = 0;
    while (1) {
        fill_buffer(input_user_buffer);
        neai_oneclass(input_user_buffer, &oneclass_result);
        /* USER BEGIN */
        /*
        * e.g.: Trigger functions depending on oneclass_result
        * (blink LED, ring alarm, etc.).
        */
        /* USER END */
    }
}

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.



Retrieved from "https://wiki.stmicroelectronics.cn/stm32mcu/index.php?title=AI:NanoEdge_AI_Library_for_1-class_classification_(1CC)&oldid=55622"