mh5_anomaly_detector repository

Repository Summary

Checkout URI https://github.com/narayave/mh5_anomaly_detector.git
VCS Type git
VCS Version master
Last Updated 2018-08-19
Dev Status MAINTAINED
Released UNRELEASED

Packages

Name Version
mh5_anomaly_detector 0.1.6

README

ROS Anomaly Detector package

Overview

The ROS Anomaly Detector Module (ADM) is designed to execute alongside industrial robotic arm tasks to detect unintended deviations at the application level. The ADM utilizes a learning based technique to achieve this. The process has been made efficient by building the ADM as a ROS package that aptly fits in the ROS ecosystem. While this module is specific to anomaly detection for an industrial arm, it is extensible to other projects with similar goals. This is meant to be a starting point for other such projects. <!-- Figure \ref{fig:ros_i_overview} presents the ROS-I architecture. The red box around ROS ADM in the ROS Layer shows where the ADM exists within the ROS ecosystem. -->

The crux of anomaly detection within this module relies on a three step process. The steps includes creating datasets out of the published messages within ROS, training learning models from those datasets, and deploying it in production. Appropriately, the three modes that the ROS ADM can be executed in are Collect, Learn, and Operate. The following image presents the high level workflow that is meant to be followed.

Image

Collect Mode

The ADM is first run in the Collect mode after the task has been programmed and assessed in the commissioning phase of the robot. This mode runs AND collects data as a node by subscribing to the '/joint_states' topic, of message type 'sensor_msgs/JointState,' within ROS.

Learn Mode

The goal of this mode is submitting the user with trained classifiers. In order to accomplish this the user is depended on to supply not only the datasets for training, but also the classifiers of interest. These are supplied to an internal library via a script. The internal library semi-automates the learning process. Particularly, the user is only required to supply their script to the Interface component. It in turn, pre-processes the data (dimensionality reduction if specified), and the set of classifiers are all trained with their appropriate datasets. After training, the classifiers are saved in storage.

At a high level, the user's responsibilities includes writing a simple script, and supplying the names of datasets in a structured manner. From a small set of functions calls to the Interface class, the classifiers are quickly trained and are efficiently prepared for deployment. There are four key, programmatic steps to accomplish in order to get training predictions and its associated accuracies. First, the Interface class needs to be instantiated by passing a YAML input file as an argument. YAML is a human-friendly data serialization method, that is interoperable between programming languages and systems. Next, the user creates instances of the classifiers they want trained. The user then passes off the classifiers to the Interface class for training, by calling the genmodel_train() function. Lastly, testing predictions can be gathered for all classifiers, by calling get_testing_predictions() .

The following python script presents an example of a script utilizing the learning library.

if __name__ == "__main__":
    unsupervised_models = []
    supervised_models = []

# Step 1
    input_file = sys.argv[1] # YAML input
    interface = Interface(input_file)

# Step 2
    ocsvm = svm.OneClassSVM(nu=0.5, kernel="rbf", gamma=1000)
    unsupervised_models.append(('ocsvm', clf_ocsvm))

    svm1 = svm.SVC(kernel='rbf', gamma=5, C=10)
    supervised_models.append(('rbfsvm1', svm1))
    svm2 = svm.SVC(kernel='rbf', gamma=50, C=100)
    supervised_models.append(('rbfsvm2', svm2))

# Step 3
    interface.genmodel_train(unsupervised_models, supervised_models)

# Step 4
    unsup, sup = interface.get_testing_predictions()

Learning library

  • Interface
  • Data Process
  • Data Persist
  • Anomaly Detector
  • Output Matrix

The following image shows the design of the different components of the library, and how they interact with each other. Image

Operate Mode

In the Operate mode, the classifier that was determined the best is put into effect as the robot is deployed to execute its task, along with the other saved preprocessing models. It is worth noting that the task the robot executes is the same one that was initially programmed in the commissioning phase. The Operate mode runs as a node, so it subscribes to the '/joint_states' topic. The data is preprocessed and fed to the determined classifier for a prediction. The user is warned of anomalous data points, and is encourages to take action when the anomaly score surpasses the threshold.


Dependencies

ROS needs to be first installed

catkin build ros related items needed

ros-core

moveit - ros_moveit_core - moveit_setup_assistant

ros industrial core motoman driver

Dependencies independet of ROS


Build

setup.py should run from home directory specify the 'learn_lib' package to build within the src/ directory

then go back and build the whole ROS workspace Then scripts should be able to run

-- Build entire workspace!!!!!

    python setup.py build


Usage

The ADM can be included and initiated by placing each of the codes in the appropriate project. Items within the < > brackets need to be modified by user.

Collect Mode

<include file="$(find mh5_anomaly_detector)/launch/module.launch">
  <arg name="mode_arg" value="-collect" />
  <arg name="other_args" value="<tpc_name>'
    `$(find pkg)/data/<output_file>.csv' <out_label>" />
</include>

Learn Mode

<include file="$(find mh5_anomaly_detector)/launch/module.launch">
  <arg name="mode_arg" value="-train" />
  <arg name="other_args" value="<script>.py" />
  <arg name="pkg_name" value="pkg" />
  <arg name="file_name" value="'$(find pkg)/scripts/<yaml_input_file>.yaml'" />
</include>

Operate Mode

<include file="$(find mh5_anomaly_detector)/launch/module.launch">
  <arg name="mode_arg" value="-operate" />
  <arg name="other_args" value="'<tpc_name>' <classifier_name> <threshold_value>" />
</include>

The ADM can also be initiated from the command line, following the bash command. It's important that all arguments are properly supplied.

roslaunch mh5_anomaly_detector module.launch arg_1:=arg1_value ...


Gotchas (Limitations and things to keep in mind)

  1. Make sure to always supply full paths, instead of relative paths
  2. To properly use in production, learning models should be trained and tested to be precise.
  3. ROS ADM needs to be protected itself. Not only should be there be preventive mechanisms, but also ways to hinder attackers from exploiting the ADM.

4. ADM works in the application space, and other methods must be explored to protect lower levels as well.

TODOs

  1. Decrease the large overhead involved when making changes to task.

2. ADM needs to be flexible to account for multiple tasks.

License

BSD-3