Research Article

Rolling Bearing Fault Diagnosis Based on STFT-Deep Learning and Sound Signals

Table 8

Settings of SAE.

Layer 1The number of input neurons128
The number of hidden neurons64
Sparsity0.1
Sparsity penalty factor of the loss function3
Weight decay factor of the loss function0.003

Layer 2The number of input neurons64
The number of hidden layer neurons32
Sparsity0.1
Sparsity penalty factor of the loss function3
Weight decay factor of the loss function0.003

Layer 3The number of input neurons32
The number of hidden neurons16
Sparsity0.1
Sparsity penalty factor of the loss function3
Weight decay factor of the loss function0.003

Softmax classifierThe number of input neurons16
The number of output neurons4