Bearing Fault Diagnosis Based on VMD-WPT-ViT for Wind Turbine

Abstract

As an important part of wind turbine, the fault diagnosis of rolling bearings is helpful to maintain the stable operation of the fan. However, most of the existing fault diagnosis methods focus on one-dimensional time series data, which is not only difficult to extract multi-level feature information, but also fails to consider the noise problem in the signal. Therefore, to solve the above problems, a noise reduction algorithm combining variational mode decomposition (VMD) and wavelet packet threshold noise reduction (WPT) is proposed, and Vision-Transformer (ViT) model is combined to realize the fault diagnosis of rolling bearings of wind turbines. VMD is used to decompose the original signal and find out the noisy component. WPT is used to de-noise the component, and the de-noised component and other components are reconstructed to form a pure signal and converted into a two-dimensional image. Finally, a diagnostic model is built based on Vision-transformer. The experimental results show that the proposed method can effectively improve the diagnostic accuracy by extracting the advanced global features of the signal.