mlsanityData to 14 Oct 2026

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

arXiv 2403.05532 ↗First posted 8 Mar 2024cs.LG cs.CVcited by 0

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

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

50 most cited of 57 All 57 (JSON)

Built on by

Not cited yet.