Optimization of a Coupled Neuron Model Based on Deep Reinforcement Learning and Application of the Model in Bearing Fault Diagnosis

• Published in: Sensors (Basel, Switzerland) Vol. 25; no. 12; p. 3654
• Main Authors: Wang, Shan, Li, Jiaxiang, Xu, Xinsheng, Wu, Ruiqi, Qiu, Yuhang, Chen, Xuwen, Qiao, Zijian
• Format: Journal Article
• Language: English
• Published: Switzerland MDPI AG 11.06.2025; MDPI
• Subjects: Accuracy; Algorithms; coupled neuron; Data mining; Deep learning; deep reinforcement learning; Efficiency; Fault diagnosis; Fourier transforms; Machine learning; Neurons; Neurophysiology; Optimization algorithms; parameter optimization; Signal to noise ratio; Wavelet transforms; China; coupled neuron; fault diagnosis; deep reinforcement learning; parameter optimization
• ISSN: 1424-8220; 1424-8220
• DOI: 10.3390/s25123654

Bearings are critical yet vulnerable components in mechanical equipment, with potential failures that can significantly impact system performance. As stochastic resonance methods effectively convert noise energy into fault characteristic energy within bearing vibration signals, they remain a research focus in bearing fault diagnosis. This study proposes a coupled neuron model based on biological stochastic resonance effects for processing bearing vibration signals. To enhance parameter optimization, we develop an improved deep reinforcement learning algorithm that incorporates a prioritized experience replay buffer into the network architecture. Using the SNR as the evaluation metric, the algorithm performs data screening on the replay buffer parameters before training the deep network for predicting coupled neuron model performance. In terms of experimental content, the study performed data processing on simulated signals and vibration signals of gearbox bearing faults collected in the laboratory environment. By comparing the coupled neuron model optimized with a reinforcement learning algorithm, particle swarm algorithm, and quantum particle swarm algorithm, the experimental results show that the coupled neuron model optimized with a deep reinforcement learning algorithm has the optimal signal-to-noise ratio of the output signal and recognition rate of the bearing faults, which are −13.0407 dB and 100%, respectively. The method shows significant performance advantages in realizing the energy enhancement of the bearing fault eigenfrequency and provides a more efficient and accurate solution for bearing fault diagnosis, which has important engineering application value.