Ranking Deep Learning Generalization using Label Variation in Latent Geometry Graphs
Abstract
Measuring the generalization performance of a Deep Neural Network (DNN) without relying on a validation set is a difficult task. In this work, we propose exploiting Latent Geometry Graphs (LGGs) to represent the latent spaces of trained DNN architectures. Such graphs are obtained by connecting samples that yield similar latent representations at a given layer of the considered DNN. We then obtain a generalization score by looking at how strongly connected are samples of distinct classes in LGGs. This score allowed us to rank 3rd on the NeurIPS 2020 Predicting Generalization in Deep Learning (PGDL) competition.
Cite
@article{arxiv.2011.12737,
title = {Ranking Deep Learning Generalization using Label Variation in Latent Geometry Graphs},
author = {Carlos Lassance and Louis Béthune and Myriam Bontonou and Mounia Hamidouche and Vincent Gripon},
journal= {arXiv preprint arXiv:2011.12737},
year = {2020}
}
Comments
Short paper describing submission that got the 3rd place on the NeurIPS 2020 Predicting Generalization in Deep Learning (PGDL) competition. We hope to update this with more analysis when the full data is made available