Boosting Deep Ensembles with Learning Rate Tuning
Not on the map. Why?
Abstract (arXiv)
The Learning Rate (LR) has a high impact on deep learning training performance. A common practice is to train a Deep Neural Network (DNN) multiple times with different LR policies to find the optimal LR policy, which has been widely recognized as a daunting and costly task. Moreover, multiple times of DNN training has not been effectively utilized. In practice, often only the optimal LR is adopted, which misses the opportunities to further enhance the overall accuracy of the deep learning system and results in a huge waste of both computing resources and training time. This paper presents a novel framework, LREnsemble, to effectively leverage effective learning rate tuning to boost deep ensemble performance. We make three original contributions. First, we show that the LR tuning with different LR policies can produce highly diverse DNNs, which can be supplied as base models for deep ensembles. Second, we leverage different ensemble selection algorithms to identify high-quality deep ensembles from the large pool of base models with significant accuracy improvements over the best single base model. Third, we propose LREnsemble, a framework that utilizes the synergy of LR tuning and deep ensemble techniques to enhance deep learning performance. The experiments on multiple benchmark datasets have demonstrated the effectiveness of LREnsemble, generating up to 2.34% accuracy improvements over well-optimized baselines.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 30 of its 56 references that are in the map. First posted on arXiv 10 Oct 2024.
Builds on
30 of 56 references.
References · 30
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2016Wide Residual Networkscited by 3,057
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
18 more
- 2016Aggregated Residual Transformations for Deep Neural Networkscited by 2,540
- 2019SuperGlue: Learning Feature Matching with Graph Neural Networkscited by 1,437
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,404
- 2018CoQA: A Conversational Question Answering Challengecited by 674
- 2015Cyclical Learning Rates for Training Neural Networkscited by 627
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
- 2019Deep Ensembles: A Loss Landscape Perspectivecited by 438
- 2018Tune: A Research Platform for Distributed Model Selection and Trainingcited by 395
- 2020BatchEnsemble: An Alternative Approach to Efficient Ensemble and Lifelong Learningcited by 368
- 2017Snapshot Ensembles: Train 1, get M for freecited by 349
- 2021ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networkscited by 293
- 2021When Vision Transformers Outperform ResNets without Pre-training or Strong Data Augmentationscited by 259
- 2020Efficient and Scalable Bayesian Neural Nets with Rank-1 Factorscited by 171
- 2020Hyperparameter Ensembles for Robustness and Uncertainty Quantificationcited by 149
- 2021AutoLRS: Automatic Learning-Rate Schedule by Bayesian Optimization on the Flycited by 14
- 2019Demystifying Learning Rate Policies for High Accuracy Training of Deep Neural Networkscited by 12
- 2021Efficient Diversity-Driven Ensemble for Deep Neural Networkscited by 11
- 2023Hierarchical Pruning of Deep Ensembles with Focal Diversitycited by 4
Built on by
Citing papers per year: 2025: 1.
1 paper.