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

Twin: Tuning Learning Rate and Weight Decay of Deep Homogeneous Classifiers without Validation

About

We introduce Tune without Validation (Twin), a simple and effective pipeline for tuning learning rate and weight decay of homogeneous classifiers without validation sets, eliminating the need to hold out data and avoiding the two-step process. Twin leverages the margin-maximization dynamics of homogeneous networks and an empirical scaling law that links training and test losses across hyper-parameter configurations. This mathematical modeling yields a regime-dependent, validation-free selection rule: in the non-separable regime, training loss is monotonic in test loss and therefore predictive of generalization, whereas in the separable regime, the parameters' norm becomes a reliable indicator of generalization due to margin maximization. Across 37 dataset-architecture configurations for image classification, we demonstrate that Twin achieves a mean absolute error of 1.28% compared to an Oracle baseline that selects HPs using test accuracy. We demonstrate Twin's benefits in scenarios where validation data is scarce, such as small-data regimes, or difficult and costly to collect, as in medical imaging. Code available at https://github.com/lorenzobrigato/twin.

Lorenzo Brigato, Stavroula Mougiakakou• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationTinyImageNet (test)
Accuracy60.8
562
Image ClassificationEuroSAT (test)--
195
Image ClassificationCUB (test)--
41
Medical Image ClassificationMedMNIST v2 (test)
Average Accuracy83.1
16
Image ClassificationISIC 2018 (test)
Balanced Test Accuracy68.8
9
Image ClassificationCLaMM (test)
Balanced Accuracy75.2
9
Image ClassificationCIFAR-10 (test)
Test Accuracy91.8
6
Image ClassificationCIFAR-100 (test)
Test Accuracy67.7
6
Showing 8 of 8 rows

Other info

Follow for update