The invention provides a rolling bearing fault diagnosis method based on
deep learning, and relates to the technical field of mechanical fault diagnosis. According to the method,
global optimization is carried out on a parameter combination in an ICEEMDAN
decomposition process through a CPO optimization
algorithm, the optimal amplitude of
white noise and the number of times of adding the
white noise are determined, an intrinsic mode function is screened and denoised in combination with a
correlation coefficient, reconstruction of a vibration
signal of a rolling bearing is completed, and the vibration
signal of the rolling bearing is reconstructed. The problem of mode
aliasing is solved while the reliability and the stability of a vibration
signal of the rolling bearing are improved; a K nearest neighbor method is used to construct a
graph model, an edge connection relation between nodes can be effectively established, so that a foundation is laid for subsequent deep analysis and
processing of vibration signals based on a graph structure, a graph attention neural network is input, topological structure information of the nodes and edges is fully utilized, extracted node features are classified and identified, and the accuracy of vibration analysis is improved. Therefore, fault diagnosis of the rolling bearing is realized.