Rolling bearings play a vital role in ensuring the safe operation of rotating machinery. However, in many application scenarios, the collected data has a low signal-to-noise ratio and the samples with faults are rare, which affects the generalization capability of the model, making it impossible to achieve accurate diagnosis. To solve this problem, the selection of time-frequency (TF) maps was considered in this paper through reinforcement learning. The time-frequency maps are built by four classical time-frequency characterization methods such as short-time Fourier transform (STFT) and synchro squeezing transform (SST). And the match-reinforcement learning time frequency selection (MRLTFS) fault diagnosis model is proposed to extract the fault-related features. Experiments show that the proposed MRLTFS method is superior to existing methods in robustness, generalization and feature selection capability.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.