Research on Intelligent Fault Diagnosis of Rolling Bearing Based on Improved ShufflenetV2-LSTM

Table 4

Hyperparameter settings of each model.

Model	LSTM	Optimizer	Activation function	Dropout
Improved Alexnet	Add	RMSprop	LeakyRelu	Delete the original
Improved MobilenetV2	Add	RMSprop	LeakyRelu	Without
Improved Squeezenet	Add	RMSprop	LeakyRelu	Keep the original
Improved Resnet-18	Add	Adam	LeakyRelu	Without
Improved Xception	Add	Adam	LeakyRelu	Without