The unreasonable effectiveness of early discarding after one epoch in neural network hyperparameter optimization

被引:2
作者
Egele, Romain [1 ,2 ]
Mohr, Felix [3 ]
Viering, Tom [4 ]
Balaprakash, Prasanna [5 ]
机构
[1] Argonne Natl Lab, Lemont, IL USA
[2] Univ Paris Saclay, Saclay, France
[3] Univ La Sabana, Chia, Colombia
[4] Delft Univ Technol, Delft, Netherlands
[5] Oak Ridge Natl Lab, Oak Ridge, TN USA
关键词
Hyperparameter optimization; Multi-fidelity optimization; Deep neural network; Learning curve;
D O I
10.1016/j.neucom.2024.127964
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
To reach high performance with deep learning, hyperparameter optimization (HPO) is essential. This process is usually time-consuming due to costly evaluations of neural networks. Early discarding techniques limit the resources granted to unpromising candidates by observing the empirical learning curves and canceling neural network training as soon as the lack of competitiveness of a candidate becomes evident. Despite two decades of research, little is understood about the trade-off between the aggressiveness of discarding and the loss of predictive performance. Our paper studies this trade-off for several commonly used discarding techniques such as successive halving and learning curve extrapolation. Our surprising finding is that these commonly used techniques offer minimal to no added value compared to the simple strategy of discarding after a constant number of epochs of training. The chosen number of epochs mostly depends on the available compute budget. We call this approach i-Epoch ( i being the constant number of epochs with which neural networks are trained) and suggest to assess the quality of early discarding techniques by comparing how their Pareto-Front (in consumed training epochs and predictive performance) complement the Pareto-Front of i-Epoch.
引用
收藏
页数:13
相关论文
共 31 条
  • [1] Adriaensen S, 2023, Arxiv, DOI arXiv:2310.20447
  • [2] Performance indicators in multiobjective optimization
    Audet, Charles
    Bigeon, Jean
    Cartier, Dominique
    Le Digabel, Sebastien
    Salomon, Ludovic
    [J]. EUROPEAN JOURNAL OF OPERATIONAL RESEARCH, 2021, 292 (02) : 397 - 422
  • [3] Awad N, 2021, PROCEEDINGS OF THE THIRTIETH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, IJCAI 2021, P2147
  • [4] Baker B, 2017, Arxiv, DOI [arXiv:1705.10823, 10.48550/arXiv.1705.10823]
  • [5] Bansal A., 2022, 36 C NEUR INF PROC S
  • [6] Domhan T, 2015, PROCEEDINGS OF THE TWENTY-FOURTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE (IJCAI), P3460
  • [7] Egelé R, 2023, Arxiv, DOI arXiv:2307.15422
  • [8] Eggensperger K, 2021, Arxiv, DOI arXiv:2109.06716
  • [9] El-Yaniv R., 2017, arXiv
  • [10] Falkner S., 2023, 11 INT C LEARN REPR