A Simple Trick for Estimating the Weight Decay Parameter

Thorsteinn S. Rögnvaldsson

Abstract: We present a simple trick to get an approximate estimate of the weight decay parameter λ. The method combines early stopping and weight decay, into the estimate $$ \hat \lambda = ||\nabla E(W_{es} )||/||2W_{es} ||, $$ where W es is the set of weights at the early stopping point, and E(W) is the training data fit error. The estimate is demonstrated and compared to the standard cross-validation procedure for λ selection on one synthetic and four real life data sets. The result is that λ is as good an estimator for the optimal weight decay parameter value as the standard search estimate, but orders of magnitude quicker to compute. The results also show that weight decay can produce solutions that are significantly superior to committees of networks trained with early stopping.