arXiv · 2401.17790
RADIN: Souping on a Budget
Abstract
Model Soups, extending Stochastic Weights Averaging (SWA), combine models fine-tuned with different hyperparameters. Yet, their adoption is hindered by computational challenges due to subset selection issues. In this paper, we propose to speed up model soups by approximating soups performance using averaged ensemble logits performances. Theoretical insights validate the congruence between ensemble logits and weight averaging soups across any mixing ratios. Our Resource ADjusted soups craftINg (RADIN) procedure stands out by allowing flexible evaluation budgets, enabling users to adjust his budget of exploration adapted to his resources while increasing performance at lower budget compared to previous greedy approach (up to 4% on ImageNet).
Explore related subjects
Keep this discovery
Thibaut Menes, Olivier Risser-Maroix. 2024-01-31. RADIN: Souping on a Budget. https://arxiv.org/abs/2401.17790
Cite the original work for its findings. Save a collection to share your selection of sources.