Research Article

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

Table 3

Settings of SAE.

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

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

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

Softmax classifierThe number of input layer nodes10
The number of output layer nodes4