Title
Identifying Generalization Properties in Neural Networks.
Abstract
While it has not yet been proven, empirical evidence suggests that model generalization is related to local properties of the optima which can be described via the Hessian. We connect model generalization with the local property of a solution under the PAC-Bayes paradigm. In particular, we prove that model generalization ability is related to the Hessian, the higher-order smoothness terms characterized by the Lipschitz constant of the Hessian, and the scales of the parameters. Guided by the proof, we propose a metric to score the generalization capability of the model, as well as an algorithm that optimizes the perturbed model accordingly.
Year
Venue
Field
2018
arXiv: Learning
Mathematical optimization,Hessian matrix,Lipschitz continuity,Local property,Smoothness,Artificial neural network,Mathematics
DocType
Volume
Citations 
Journal
abs/1809.07402
5
PageRank 
References 
Authors
0.41
11
4
Name
Order
Citations
PageRank
Huan Wang145719.67
nitish shirish keskar232516.71
Caiming Xiong396969.56
Richard Socher46770230.61