One-Class Fault Detection Using Multi-Layer Elm-Based Auto-Encoder

Abstract

A new approach for one-class fault detection trained only by normal samples has been proposed in this paper. The approach contains multi-anterior-layers for feature extraction and one post-layer for one-class classification. The multi-anterior-layers are based on extreme learning machine-based auto-encoder (ELM-AE). Multi-ELM-AEs are stacked in the front hidden layers to extract abstract features from the raw input. The post-layer is based on the reconstruction error-based ELM-AE (Re-ELM-AE) to act as one-class classifier. As the extension of ELM-AE, the decision threshold and function are given in the Re-ELM-AE, which are utilized to identify whether the test sample is faulty. The efficacy of the presented algorithm is demonstrated on a mathematic example and fault dataset from motor bearing. The method has been compared with shallow learning methods such as one-class support vector machine (OCSVM), the Re-ELM-AE, and one multi-layer neural network named stacked auto-encoder (SAE). The experiment results show that the proposed method outperforms OCSVM and Re-ELM-AE in classification accuracy. Though the classification accuracy of the proposed method and SAE is similar, the training and testing time of the proposed method is much lower than SAE.