Research Article

Diagnosis of Retinal Diseases Based on Bayesian Optimization Deep Learning Network Using Optical Coherence Tomography Images

Table 2

Hyperparameters and their search space.

ParameterSearch spaceDescription

OptimizerAdam, RMSProp, SGD, AdaDeltaTo optimize the input weights by comparing the prediction and the loss function
Learning rate1e − 3, 1e − 4, 1e − 5, 1e − 6To determine the step size at each iteration while minimizing the loss function
Activation functionReLu, Elu and Tanh, Leaky ReLuTo introduce nonlinearity into the output of neurons
Number of neurons in customized layers64,128, 56, 512,1024To compute the weighted average of the input
Batch size32,64,128Number of training examples utilized in one iteration