Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Predicting Neural Network Accuracy from Weights

About

We show experimentally that the accuracy of a trained neural network can be predicted surprisingly well by looking only at its weights, without evaluating it on input data. We motivate this task and introduce a formal setting for it. Even when using simple statistics of the weights, the predictors are able to rank neural networks by their performance with very high accuracy (R2 score more than 0.98). Furthermore, the predictors are able to rank networks trained on different, unobserved datasets and with different architectures. We release a collection of 120k convolutional neural networks trained on four different datasets to encourage further research in this area, with the goal of understanding network training and performance better.

Thomas Unterthiner, Daniel Keysers, Sylvain Gelly, Olivier Bousquet, Ilya Tolstikhin• 2020

Related benchmarks

TaskDatasetResultRank
Adapter RetrievalMixed Adapter Pool (ARC-C, BoolQ, GSM8K, MBPP) LoRA (test)
Average Score27.82
6
Attribute ClassificationCelebA-LoRA SD1.5
Macro F111.51
6
Generalization predictionSmallCNN Zoo CIFAR-10-GS ReLU (test)
Kendall's Tau0.914
6
Generalization predictionSmallCNN Zoo SVHN-GS ReLU (test)
Kendall's Tau0.8463
6
Generalization predictionSmallCNN Zoo CIFAR-10-GS Tanh (test)
Kendall's Tau0.914
6
Generalization predictionSmallCNN Zoo SVHN-GS Tanh (test)
Kendall's Tau0.844
6
Generalization predictionSmallCNN Zoo CIFAR-10-GS both activations ReLU Tanh (test)
Kendall's Tau91.5
6
Attribute ClassificationGoEmotions LoRA
Macro F1 Score2.87
5
Attribute ClassificationCUB-LoRA
Macro-F17.44
5
Attribute ClassificationCelebA-LoRA
Macro F17.73
5
Showing 10 of 25 rows

Other info

Follow for update