English

Agile Online Model Selection: Resolving Adaptation Lag via Safeguarded Large Learning Rates

Machine Learning 2026-05-27 v1 Machine Learning

Abstract

Maintaining predictive accuracy in non-stationary environments requires online model selection to adapt autonomously to unknown distribution shifts. However, existing tuning-free algorithms face a fundamental trade-off between robustness and agility. Specifically, to ensure dynamic regret bounds, they must restrict learning rates to small constants (e.g., O(1)O(1)). This restriction inevitably causes significant adaptation lag during abrupt changes. To resolve this, we propose a novel optimistic online mirror descent that utilizes safeguarded large learning rates up to Θ(T)\Theta(T), where TT is the number of rounds. Our key technical contribution is a post-hoc penalty mechanism that dynamically monitors unstable updates and excludes learning rates incurring excessive regret, eliminating the need for restrictive a priori constraints. We show that the cumulative penalty remains O(logT)O(\log T), allowing our algorithm to match near-optimal worst-case guarantees while achieving superior rates in benign cases. Empirical evaluations on synthetic and eleven diverse real-world datasets demonstrate that our approach reduces the adaptation lag from hundreds of rounds to a few rounds, consistently outperforming tuning-free baselines.

Keywords

Cite

@article{arxiv.2605.26919,
  title  = {Agile Online Model Selection: Resolving Adaptation Lag via Safeguarded Large Learning Rates},
  author = {Kei Takemura and Ryuta Matsuno and Keita Sakuma},
  journal= {arXiv preprint arXiv:2605.26919},
  year   = {2026}
}

Comments

Accepted to KDD 2026