An accurate flatness measure to estimate the generalization performance of CNN models

Rahman Taleghani, Maryam Mohammadi, Francesco Marchetti

Flatness measures based on the spectrum or the trace of the Hessian of the loss are widely used as proxies for the generalization ability of deep networks. However, most existing d…