Self-adaptive selection graph pooling based fault diagnosis method under few samples and noisy environment

Abstract

Neural network (NN)-based methods are extensively used for intelligent fault diagnosis in industrial systems. Nevertheless, due to the limited availability of faulty samples and the presence of noise interference, most existing NN-based methods perform limited diagnosis performance. In response to these challenges, a self-adaptive selection graph pooling method is proposed. Firstly, graph encoders with sharing parameters are designed to extract local structure-feature information (SFI) of multiple sensor-wise sub-graphs. Then, the temporal continuity of the SFI is maintained through time-by-time concatenation, resulting in a global sensor graph and reducing the dependency on data volume from the perspective of adding prior knowledge. Subsequently, leveraging a self-adaptive node selection mechanism, the noise interference of redundant and noisy sensor-wise nodes in the graph is alleviated, allowing the networks to concentrate on the fault-attention nodes. Finally, the local max pooling and global mean pooling of the node-selection graph are incorporated in the readout module to get the multi-scale graph features, which serve as input to a multi-layer perceptron for fault diagnosis. Two experimental studies involving different mechanical and electrical systems demonstrate that the proposed method not only achieves superior diagnosis performance with limited data, but also maintains strong anti-interference ability in noisy environments. Additionally, it exhibits good interpretability through the proposed self-adaptive node selection mechanism and visualization methods.