Many Optimizers But Only One Training Path: Repeated Resampling for Adaptive Optimizer Selection
Organizations: insureAI · ETH Zürich
Abstract
An optimizer is usually chosen before training a deep neural network and then kept fixed. Treating optimizer choice as a hyperparameter could boost performance, but it requires several complete training runs and discards all but the winner. Repeated Optimizer Resampling (ROR) instead searches during one evolving run. Every epochs, each candidate optimizer scouts from the current model weights for epochs. The best scout continues for the remaining epochs, and that completed segment becomes the new incumbent if it improves the validation objective. This design allows the preferred optimizer to change as training progresses. We compare two variants of ROR on MNIST, Fashion-MNIST, and two motor insurance claim-count models. Nine fixed optimizers and both ROR variants are evaluated with the same ten seeds. One-epoch ROR uses 24% to 35% of the aggregate training needed to identify the best fixed optimizer exhaustively and remains close to that optimizer on all four tasks. These results support short scouting as a practical way to search over optimizers without completing every candidate run.
Figures & tables
| Training rule | MNIST | Fashion-MNIST | ||||
| Accuracy | Rank | Epoch-eq. | Accuracy | Rank | Epoch-eq. | |
| Adam | 98.003 0.244 | 7 | 16.5 | 88.287 0.472 | 13 | 16.1 |
| AdamW | 97.928 0.213 | 12 | 15.0 | 88.262 0.454 | 14 | 15.8 |
| Lion | 97.552 0.189 | 16 | 13.4 | 88.578 0.299 | 9 | 12.9 |
| Muon | 98.095 0.139 | 4 | 15.2 | 88.431 0.516 | 10 | 13.3 |
| SGD-N | 97.787 0.213 | 14 | 18.1 | 88.084 0.356 | 15 | 17.6 |
| Architecture | Training rule | Test deviance | Rank | Calibration | Epoch-eq. |
| MLP | Adam | 0.238463 0.000263 | 7 | 0.999 | 44.3 |
| AdamW | 0.238463 0.000263 | 8 | 0.999 | 44.2 | |
| Lion | 0.238365 0.000353 | 2 | 0.995 | 31.3 | |
| Muon | 0.238652 0.000178 | 15 | 0.999 | 22.1 | |
| SGD-N | 0.249816 0.007473 | 16 | 1.003 | 58.1 | |
| Nadam | 0.238449 0.000240 | 6 | 1.000 | 47.4 |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Task | Policy | Comparison | Metric | Difference (95% CI) | Short eq. | eq. |
|---|---|---|---|---|---|---|
| MNIST | CS-ROR | minus | Accuracy (pp) | -0.184 [-0.355, -0.013] | 37.4 | 116.1 |
| CS-ROR | minus | Accuracy (pp) | -0.095 [-0.256, 0.066] | 58.9 | 116.1 | |
| SP-ROR | minus | Accuracy (pp) | -0.143 [-0.307, 0.021] | 39.6 | 110.7 | |
| SP-ROR | minus | Accuracy (pp) | -0.151 [-0.285, -0.017] | 68.4 | 110.7 | |
| Fashion-MNIST | CS-ROR | minus | Accuracy (pp) | 0.019 [-0.258, 0.296] | 53.9 | 110.7 |
| CS-ROR | minus | Accuracy (pp) | -0.091 [-0.377, 0.195] | 81.7 | 110.7 |
Deep Learning Optimization
Deep learning optimization develops algorithms and training strategies that help neural networks converge, generalize, and use compute efficiently.
See all 205 papers