SWR2 Net: a fault diagnosis framework for rotating machine under limited samples and noise interference
摘要
In the industrial environment, strong noise interference and insufficient training samples bring severe challenges to mechanical fault diagnosis. A fault diagnosis framework based on a novel multimodal data fusion method and an improved multiscale residual network is proposed to overcome the above issues. Firstly, a multimodal data fusion method using principal component analysis and feature map concatenation is introduced. It significantly enhances the feature retention of different modal signals and improves their adaptability to various rotating machinery types. Secondly, by integrating a shifted window attention mechanism, the proposed multiscale residual network demonstrates an improved feature extraction and global information perception, exhibiting a superior fault classification capability than comparison models. Finally, three cases are utilized to test the applicability of the framework for bearings and motors, showcasing its robustness under high noise and limited training sample scenarios. Besides, the framework shows a generalization capability with over 99% accuracy in multiple working conditions, reflecting its potential for industrial applications.