We introduce a novel capacity measure 2sED for statistical models based on the effective dimension. The new quantity provably bounds the generalization error under mild assumptions on the model. Furthermore, simulations on standard data sets and popular model architectures show that 2sED correlates well with the training error. For Markovian models, we show how to efficiently approximate 2sED from below through a layerwise iterative approach, which allows us to tackle deep learning models with a large number of parameters. Simulation results suggest that the approximation is good for different prominent models and data sets.

A two-scale Complexity Measure for Deep Learning Models / Datres, Massimiliano; Leonardi, Gian Paolo; Figalli, Alessio; Sutter, David. - 37:(2024). ( 38th Conference on Neural Information Processing Systems, NeurIPS 2024 Vancouver 9-15 December, 2024).

A two-scale Complexity Measure for Deep Learning Models

Datres, Massimiliano
;
Leonardi, Gian Paolo;Sutter, David
2024-01-01

Abstract

We introduce a novel capacity measure 2sED for statistical models based on the effective dimension. The new quantity provably bounds the generalization error under mild assumptions on the model. Furthermore, simulations on standard data sets and popular model architectures show that 2sED correlates well with the training error. For Markovian models, we show how to efficiently approximate 2sED from below through a layerwise iterative approach, which allows us to tackle deep learning models with a large number of parameters. Simulation results suggest that the approximation is good for different prominent models and data sets.
2024
Advances in Neural Information Processing Systems
San Mateo, CA
Neural information processing systems foundation
Settore MAT/06 - Probabilita' e Statistica Matematica
Settore MAT/05 - Analisi Matematica
Settore MATH-03/B - Probabilità e statistica matematica
Settore MATH-03/A - Analisi matematica
Datres, Massimiliano; Leonardi, Gian Paolo; Figalli, Alessio; Sutter, David
A two-scale Complexity Measure for Deep Learning Models / Datres, Massimiliano; Leonardi, Gian Paolo; Figalli, Alessio; Sutter, David. - 37:(2024). ( 38th Conference on Neural Information Processing Systems, NeurIPS 2024 Vancouver 9-15 December, 2024).
File in questo prodotto:
File Dimensione Formato  
neurips_2024_camera-ready2.pdf

Solo gestori archivio

Tipologia: Versione editoriale (Publisher’s layout)
Licenza: Tutti i diritti riservati (All rights reserved)
Dimensione 731.24 kB
Formato Adobe PDF
731.24 kB Adobe PDF   Visualizza/Apri

I documenti in IRIS sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione

Utilizza questo identificativo per citare o creare un link a questo documento: https://hdl.handle.net/11572/454511
Citazioni
  • ???jsp.display-item.citation.pmc??? ND
  • Scopus 0
  • ???jsp.display-item.citation.isi??? ND
  • OpenAlex ND
social impact