We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

stat.ML

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Statistics > Machine Learning

Title: A generalization gap estimation for overparameterized models via the Langevin functional variance

Abstract: This paper discusses the estimation of the generalization gap, the difference between generalization performance and training performance, for overparameterized models including neural networks. We first show that a functional variance, a key concept in defining a widely-applicable information criterion, characterizes the generalization gap even in overparameterized settings where a conventional theory cannot be applied. As the computational cost of the functional variance is expensive for the overparameterized models, we propose an efficient approximation of the function variance, the Langevin approximation of the functional variance (Langevin FV). This method leverages only the $1$st-order gradient of the squared loss function, without referencing the $2$nd-order gradient; this ensures that the computation is efficient and the implementation is consistent with gradient-based optimization algorithms. We demonstrate the Langevin FV numerically by estimating the generalization gaps of overparameterized linear regression and non-linear neural network models, containing more than a thousand of parameters therein.
Comments: 40 pages, no figure, accepted to Journal of Computational and Graphical Statistics
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG); Methodology (stat.ME)
Cite as: arXiv:2112.03660 [stat.ML]
  (or arXiv:2112.03660v3 [stat.ML] for this version)

Submission history

From: Akifumi Okuno [view email]
[v1] Tue, 7 Dec 2021 12:43:05 GMT (33kb)
[v2] Sun, 26 Dec 2021 09:48:21 GMT (35kb)
[v3] Mon, 20 Mar 2023 00:22:11 GMT (39kb)

Link back to: arXiv, form interface, contact.