A machine learning engineer is training a logistic regression model and notices that the loss is decreasing very slowly. The learning rate is set to 0.001. What is the MOST likely cause and appropriate fix?
Trap 1: The learning rate is too high; decrease it to 0.0001
A high learning rate would cause the loss to oscillate or increase, not slowly decrease.
Trap 2: The model is overfitting; add L2 regularisation
Overfitting does not cause slow decrease of loss; regularisation may reduce overfitting but not address the learning rate.
Trap 3: The batch size is too large; reduce it
Batch size affects noise but not the rate of decrease as directly as learning rate.
- A
The learning rate is too low; increase it to 0.01
A learning rate of 0.001 is very small; increasing it to 0.01 will speed up convergence without causing divergence.
- B
The learning rate is too high; decrease it to 0.0001
Why wrong: A high learning rate would cause the loss to oscillate or increase, not slowly decrease.
- C
The model is overfitting; add L2 regularisation
Why wrong: Overfitting does not cause slow decrease of loss; regularisation may reduce overfitting but not address the learning rate.
- D
The batch size is too large; reduce it
Why wrong: Batch size affects noise but not the rate of decrease as directly as learning rate.