Cross domain fault diagnosis method based on MLP-mixer network

Abstract

The quality of rolling bearings determines the safety of mechanical equipment operation, and bearings with more precise structures are prone to damage due to excessive operation. Therefore, cross domain fault diagnosis of bearings has become a research hotspot. To better improve the accuracy of bearing cross domain fault diagnosis, this study proposes two models. One is a cross domain feature extraction model constructed using a mixed attention mechanism, which recognizes and extracts high-level features of bearing faults through channel attention and spatial attention mechanisms. The other is a bearing cross domain fault diagnosis model based on multi-layer perception mechanism. This model takes the feature signals collected by the attention mechanism model as input to identify and align the differences between the source and target domain features, facilitating cross domain transfer of features. The experimental results show that the mixed attention mechanism model has a maximum accuracy of 97.3 % for feature recognition of different faults, and can successfully recognize corresponding signal values. The multi-layer perception model can achieve the highest recognition accuracy of 99.5 % in bearing fault diagnosis, and it can reach a stable state when it iterates to 26, and the final stable loss value is 0.28. Therefore, the two models proposed in this study have good application value.