Fine-tuning with Very Large Dropout
Not on the map. Why?
Abstract (arXiv)
It is impossible today to pretend that the practice of machine learning is always compatible with the idea that training and testing data follow the same distribution. Several authors have recently used ensemble techniques to show how scenarios involving multiple data distributions are best served by representations that are both richer than those obtained by regularizing for the best in-distribution performance, and richer than those obtained under the influence of the implicit sparsity bias of common stochastic gradient procedures. This contribution investigates the use of very high dropout rates instead of ensembles to obtain such rich representations. Although training a deep network from scratch using such dropout rates is virtually impossible, fine-tuning a large pre-trained model under such conditions is not only possible but also achieves out-of-distribution performances that exceed those of both ensembles and weight averaging methods such as model soups. This result has practical significance because the importance of the fine-tuning scenario has considerably grown in recent years. This result also provides interesting insights on the nature of rich representations and on the intrinsically linear nature of fine-tuning a large network using a comparatively small dataset.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 56 of its 76 references that are in the map. First posted on arXiv 1 Mar 2024.
Builds on
56 of 76 references.
References · 56
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2021On the Opportunities and Risks of Foundation Modelscited by 3,961
- 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,875
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,845
- 2020The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution Generalizationcited by 1,842
- 2019Invariant Risk Minimizationcited by 1,832
- 2019Do ImageNet Classifiers Generalize to ImageNet?cited by 1,507
- 2019Natural Adversarial Examplescited by 1,401
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
38 more
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
- 2018Moment Matching for Multi-Source Domain Adaptationcited by 1,327
- 2017Random Erasing Data Augmentationcited by 1,235
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,232
- 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,194
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
- 2017Deep Hashing Network for Unsupervised Domain Adaptationcited by 1,181
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
- 2020In Search of Lost Domain Generalizationcited by 1,042
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,013
- 2019Learning Robust Global Representations by Penalizing Local Predictive Powercited by 1,010
- 2017Deeper, Broader and Artier Domain Generalizationcited by 983
- 2016Deep Networks with Stochastic Depthcited by 973
- 2021Robust fine-tuning of zero-shot modelscited by 785
- 2017Learning to Generalize: Meta-Learning for Domain Generalizationcited by 762
- 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 737
- 2022Fine-Tuning can Distort Pretrained Features and Underperform Out-of-Distributioncited by 722
- 2018On Lazy Training in Differentiable Programmingcited by 671
- 2018Recognition in Terra Incognitacited by 655
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
- 2021SWAD: Domain Generalization by Seeking Flat Minimacited by 431
- 2019Rapid Learning or Feature Reuse? Towards Understanding the Effectiveness of MAMLcited by 359
- 2014CNN Features off-the-shelf: an Astounding Baseline for Recognitioncited by 316
- 2020Gradient Starvation: A Learning Proclivity in Neural Networkscited by 262
- 2021TrivialAugment: Tuning-free Yet State-of-the-Art Data Augmentationcited by 225
- 2020Pitfalls of In-Domain Uncertainty Estimation and Ensembling in Deep Learningcited by 208
- 2022Diverse Weight Averaging for Out-of-Distribution Generalizationcited by 148
- 2018Domain Generalization via Conditional Invariant Representationcited by 146
- 2021Ensemble of Averages: Improving Model Selection and Boosting Performance in Domain Generalizationcited by 144
- 2019Implicit regularization for deep neural networks driven by an Ornstein-Uhlenbeck like processcited by 123
- 2015Path-SGD: Path-Normalized Optimization in Deep Neural Networkscited by 117
- 2022Head2Toe: Utilizing Intermediate Representations for Better Transfer Learningcited by 83
- 2022SGD with Large Step Sizes Learns Sparse Featurescited by 66
- 2020Empirical or Invariant Risk Minimization? A Sample Complexity Perspectivecited by 62
- 2021No One Representation to Rule Them All: Overlapping Features of Training Methodscited by 57
- 2021Universal Representation Learning from Multiple Domains for Few-shot Classificationcited by 56
- 2017Universal representations:The missing link between faces, text, planktons, and cat breedscited by 45
- 2022Rich Feature Construction for the Optimization-Generalization Dilemmacited by 34
50 most cited of 56 All 56 (JSON)
Built on by
Citing papers per year: 2025: 3.
3 papers.