Research Article
Industrial Robot Vibration Anomaly Detection Based on Sliding Window One-Dimensional Convolution Autoencoder
Algorithm 1
Training and testing of the SW1DCAE model.
| Step 1: unsupervised training | | (0) Sliding window processing training set | | (1) Build the SW1DCAE model. Set learning rate, training epochs, and other hyperparameters | | (2) Randomly initialize the weights and biases of the network model | | (3) Input training set | | (4) Cycle training times N | | (5) Encoding, convolution, pooling | | (6) Decode, compute upsampling, deconvolution | | (7) Calculation error | | (8) Gradient descent, update the gradient parameters of each layer | | (9) End loop | | (10) Save model structure, parameters | | Output: Reconstructed data , approximately equal to input data X. | | Step 2: exception test | | (0) Sliding window processing test set | | (1) Input the test set into the trained model saved in step 1. | | (2) Set the loss threshold, and the reconstruction error of the samples in the test set exceeds the threshold, which is judged as abnormal | | Output: test sample for anomalies. |
|