Passer à la navigation principale Passer à la recherche Passer au contenu principal

Safe Grid Search with Optimal Complexity

  • Riken
  • Université Paris-Saclay
  • Université de Montpellier 2
  • Nagoya Institute of Technology

Résultats de recherche: Contribution à un journalArticle de conférenceRevue par des pairs

Résumé

Popular machine learning estimators involve regularization parameters that can be challenging to tune, and standard strategies rely on grid search for this task. In this paper, we revisit the techniques of approximating the regularization path up to predefined tolerance ɛ in a unified framework and show that its complexity is O(1/d√ɛ) for uniformly convex loss of order d ≥ 2 and O(1/ɛ) for Generalized Self-Concordant functions. This framework encompasses least-squares but also logistic regression, a case that as far as we know was not handled as precisely in previous works. We leverage our technique to provide refined bounds on the validation error as well as a practical algorithm for hyperparameter tuning. The latter has global convergence guarantee when targeting a prescribed accuracy on the validation set. Last but not least, our approach helps relieving the practitioner from the (often neglected) task of selecting a stopping criterion when optimizing over the training set: our method automatically calibrates this criterion based on the targeted accuracy on the validation set.

langue originaleAnglais
Pages (de - à)4771-4780
Nombre de pages10
journalProceedings of Machine Learning Research
Volume97
étatPublié - 1 janv. 2019
Modification externeOui
Evénement36th International Conference on Machine Learning, ICML 2019 - Long Beach, États-Unis
Durée: 9 juin 201915 juin 2019

Empreinte digitale

Examiner les sujets de recherche de « Safe Grid Search with Optimal Complexity ». Ensemble, ils forment une empreinte digitale unique.

Contient cette citation