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

## TL;DR

This paper introduces a deep reinforcement learning-optimized coupled neuron model for improving bearing fault diagnosis by enhancing signal-to-noise ratios and recognition accuracy.

## Contribution

The novel contribution is an improved deep reinforcement learning algorithm with prioritized experience replay for optimizing a coupled neuron model in fault diagnosis.

## Key findings

- The deep reinforcement learning-optimized model achieved a signal-to-noise ratio of −13.0407 dB.
- The model achieved a 100% recognition rate for bearing faults.
- The method outperformed particle swarm and quantum particle swarm algorithms in fault diagnosis performance.

## Abstract

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.