Compound feature selection and parameter optimization of ELM for fault diagnosis of rolling element bearings

ISA Trans. 2016 Nov:65:556-566. doi: 10.1016/j.isatra.2016.08.022. Epub 2016 Sep 9.

Abstract

This paper proposes a hybrid system named as HGSA-ELM for fault diagnosis of rolling element bearings, in which real-valued gravitational search algorithm (RGSA) is employed to optimize the input weights and bias of ELM, and the binary-valued of GSA (BGSA) is used to select important features from a compound feature set. Three types fault features, namely time and frequency features, energy features and singular value features, are extracted to compose the compound feature set by applying ensemble empirical mode decomposition (EEMD). For fault diagnosis of a typical rolling element bearing system with 56 working condition, comparative experiments were designed to evaluate the proposed method. And results show that HGSA-ELM achieves significant high classification accuracy compared with its original version and methods in literatures.

Keywords: Ensemble empirical mode decomposition; Extreme learning machine; Fault diagnosis; Feature selection; Gravitational search algorithm; Parameter optimization.