arXiv:2505.14903cs.LG2025-05ICML被引 10

提出基于不确定性的模型重训练决策方法,解决真实场景中何时重训的难题。

When to retrain a machine learning model

  • 通过持续预测模型性能变化趋势做重训决策
  • 在7个数据集上均优于现有基线方法
  • 适合数据稀疏、分布漂移难测的工业场景

维护真实世界机器学习模型的一大挑战是应对数据持续且不可预测的演化。从业者常面临关键问题:何时应重训或更新模型?这一看似简单的问题因三方面困难而复杂:1)决策依赖极有限信息——通常仅有少量样本;2)分布偏移的性质、程度和影响未知;3)需设定重训与性能下降之间的成本比,而该比例难以刻画。现有工作仅解决部分问题,但无一提供全面方案。分布偏移检测无法考虑成本权衡;数据稀缺且结构异常,使现有离线强化学习方法不适用;在线学习框架则忽略关键实际因素。为此,我们提出重训问题的合理建模,并设计一种基于不确定性的方法,通过持续预测以有界指标评估的模型性能演变来做出决策。实验显示,该方法在分类任务中于7个数据集上均持续优于现有基线。

原文摘要 · Abstract (English)

A significant challenge in maintaining real-world machine learning models is responding to the continuous and unpredictable evolution of data. Most practitioners are faced with the difficult question: when should I retrain or update my machine learning model? This seemingly straightforward problem is particularly challenging for three reasons: 1) decisions must be made based on very limited information - we usually have access to only a few examples, 2) the nature, extent, and impact of the distribution shift are unknown, and 3) it involves specifying a cost ratio between retraining and poor performance, which can be hard to characterize. Existing works address certain aspects of this problem, but none offer a comprehensive solution. Distribution shift detection falls short as it cannot account for the cost trade-off; the scarcity of the data, paired with its unusual structure, makes it a poor fit for existing offline reinforcement learning methods, and the online learning formulation overlooks key practical considerations. To address this, we present a principled formulation of the retraining problem and propose an uncertainty-based method that makes decisions by continually forecasting the evolution of model performance evaluated with a bounded metric. Our experiments addressing classification tasks show that the method consistently outperforms existing baselines on 7 datasets.

模型重训分布偏移不确定性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。