Twin: Tuning Learning Rate and Weight Decay of Deep Homogeneous Classifiers without Validation
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 57 of its 78 references that are in the map. First posted on arXiv 8 Mar 2024.
Builds on
57 of 78 references.
References · 57
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,498
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,183
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
- 2016Wide Residual Networkscited by 3,070
- 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,884
- 2016Aggregated Residual Transformations for Deep Neural Networkscited by 2,546
- 2019RandAugment: Practical automated data augmentation with a reduced search spacecited by 2,269
- 2017EuroSAT: A Novel Dataset and Deep Learning Benchmark for Land Use and Land Cover Classificationcited by 1,496
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
38 more
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,417
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,352
- 2016Hyperband: A Novel Bandit-Based Approach to Hyperparameter Optimizationcited by 699
- 2021MedMNIST v2 -- A large-scale lightweight benchmark for 2D and 3D biomedical image classificationcited by 690
- 2018On Lazy Training in Differentiable Programmingcited by 675
- 2019Skin Lesion Analysis Toward Melanoma Detection 2018: A Challenge Hosted by the International Skin Imaging Collaboration (ISIC)cited by 669
- 2017Exploring Generalization in Deep Learningcited by 600
- 2021How to train your ViT? Data, Augmentation, and Regularization in Vision Transformerscited by 569
- 2018Do Better ImageNet Models Transfer Better?cited by 565
- 2017Computing Nonvacuous Generalization Bounds for Deep (Stochastic) Neural Networks with Many More Parameters than Training Datacited by 533
- 2019Fantastic Generalization Measures and Where to Find Themcited by 513
- 2018BOHB: Robust and Efficient Hyperparameter Optimization at Scalecited by 470
- 2017Sharp Minima Can Generalize For Deep Netscited by 466
- 2018Bag of Tricks for Image Classification with Convolutional Neural Networkscited by 431
- 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 320
- 2019Optimizing Millions of Hyperparameters by Implicit Differentiationcited by 304
- 2021ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networkscited by 293
- 2021Escaping the Big Data Paradigm with Compact Transformerscited by 264
- 2018A System for Massively Parallel Hyperparameter Tuningcited by 202
- 2022Omnigrok: Grokking Beyond Algorithmic Datacited by 146
- 2017A Bayesian Perspective on Generalization and Stochastic Gradient Descentcited by 135
- 2019An Exponential Learning Rate Schedule for Deep Learningcited by 121
- 2020Hyper-Parameter Optimization: A Review of Algorithms and Applicationscited by 114
- 2018Predicting the Generalization Gap in Deep Networks with Margin Distributionscited by 113
- 2023Grokking as the Transition from Lazy to Rich Training Dynamicscited by 108
- 2023A Modern Look at the Relationship between Sharpness and Generalizationcited by 92
- 2021Characterizing signal propagation to close the performance gap in unnormalized ResNetscited by 88
- 2020In Search of Robust Measures of Generalizationcited by 79
- 2023Depthwise Hyperparameter Transfer in Residual Networks: Dynamics and Scaling Limitcited by 66
- 2019Do We Train on Test Data? Purging CIFAR of Near-Duplicatescited by 62
- 2020Computing the Testing Error without a Testing Setcited by 54
- 2022Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximationscited by 42
- 2017Early Stopping without a Validation Setcited by 39
- 2021How I failed machine learning in medical imaging -- shortcomings and recommendationscited by 37
- 2018The Goldilocks zone: Towards better understanding of neural network loss landscapescited by 32
- 2022Robust Training of Neural Networks Using Scale Invariant Architecturescited by 29
- 2021Measuring Generalization with Optimal Transportcited by 25
- 2021Last Layer Marginal Likelihood for Invariance Learningcited by 21
50 most cited of 57 All 57 (JSON)
Built on by
Not cited yet.