Pretrained Neural network ALEX-NET training process.

Is the modelshown below is overfitting or not? sometimes it seems that mini-batch accuracy is less than validation accuracy why?.

Answer 1

[Srivardhan Gadila]

I think the model is not overfitting. The validation loss normally decreases during the initial phase of training, as does the training loss. However, when the network begins to overfit the data, the loss on the validation set typically begins to rise and clearly there is not much difference between the training loss and validation loss. You can refer to Improve Shallow Neural Network Generalization and Avoid Overfitting for more understanding of Overfitting and steps to avoid overfitting.

The Validation accuracy can be higher than the training (mini-batch) accuracy, one possible situtation is when the network has layers that behave differently during prediction than during training for example, dropout layers. It also depends on how the training & validation data are split.

[Deepika B]

Thank you for your kind response sir.

[Deepika B]

sir, I have another question.. Is pretrained neural network is used for any new task or particularly some specific task? Is vgg19 is suitable for eye disease classiifcation problem using fundus image by having 16gb ram cpu? thank you in advance sir.

[Srivardhan Gadila]

You can refer to the following:

1. Deep learning approach to train new models faster by using pretrained models
2. Get Started with Transfer Learning
3. Transfer Learning Using AlexNet