AWS Certified Machine Learning – Specialty — Question 8

A Machine Learning Specialist built an image classification deep learning model. However, the Specialist ran into an overfitting problem in which the training and testing accuracies were 99% and 75%, respectively.
How should the Specialist address this issue and what is the reason behind it?

Answer options

• A. The learning rate should be increased because the optimization process was trapped at a local minimum.
• B. The dropout rate at the flatten layer should be increased because the model is not generalized enough.
• C. The dimensionality of dense layer next to the flatten layer should be increased because the model is not complex enough.
• D. The epoch number should be increased because the optimization process was terminated before it reached the global minimum.

Correct answer: B

Explanation

The correct answer is B because increasing the dropout rate helps to prevent overfitting by randomly dropping units during training, which encourages the model to generalize better. Option A is incorrect because simply increasing the learning rate does not solve overfitting, and option C is wrong as increasing the complexity of the model could exacerbate overfitting. Option D is also incorrect, as more epochs would likely lead to further overfitting rather than resolving the issue.