arXiv:2607.10694cs.LGcs.AI2026-07中稿 · ICML

在算力有限时,智能决定何时微调大模型以保持最佳性能。

Learning to Fine-tune Foundation Models under Resource Limitations

论文配图:Learning to Fine-tune Foundation Models under Resource Limitations
图 1 · 摘自论文原文
  • 用强化学习建模动态决策问题,结合性能、预算与数据相关性。
  • 相比同预算方法提升超4%准确率,仅用25%微调步数达97%全参数效果。
  • 适合边缘设备部署的持续学习场景,尤其关注资源受限下的高效更新。

本文研究资源受限设备上预训练基础模型的最优持续微调问题。每个时间片有新数据到达,控制器需决定是否使用数据进行微调(产生计算开销)或直接丢弃。微调后模型性能以特定任务指标(如分类准确率)衡量。目标是学习一个最优策略,决定在有限计算预算下何时对单任务(如情感分析)进行微调。将该在线决策问题建模为带约束的马尔可夫决策过程,系统状态包含三要素:模型性能、计算预算、当前数据与历史数据分布的相关性。状态转移具有随机性,因此提出基于演员-评论家的强化学习方法求解。此外,若可预先预测微调效果,则问题退化为动态规划。在大型预训练模型和常用文本分类数据集上的实验表明,本方法在相同计算预算下准确率超越传统微调方法超过4%,且仅需25%的微调步骤即可达到全参数微调97%的性能。

原文摘要 · Abstract (English)

We study the problem of optimal continual fine-tuning for a pre-trained Foundation Model deployed at a resource-limited device. At each time slot, a new batch of training data arrives, and the controller is faced with two options: either use the data to fine-tune the model and incur a compute cost, or do not fine-tune the model and discard the data. After the decision, the performance of the current model is measured in terms of an application-specific performance metric such as classification accuracy. Our objective is to learn an optimal policy that determines \emph{when to fine-tune the model} on a single task (e.g., sentiment analysis), under a finite compute budget. We formulate this online decision-making problem as a constrained Markov Decision Process, where the system state captures three essential aspects: (\textit{i}) model's performance, (\textit{ii}) computational budget, and (\textit{iii}) data distribution relevance to historic data encountered up to that point. The transition to the next state is stochastic and therefore, we propose a reinforcement learning-based method to solve this problem, namely the \emph{actor-critic} algorithm. We also consider the special case where the performance of fine-tuning for a given model can be predicted or estimated prior to decision; in this case the problem becomes a Dynamic Programming one. Experiments with a large pre-trained model on a widely-used text classification dataset demonstrate that our method consistently outperforms fine-tuning approaches with the same compute budget by more than $4\%$ in terms of accuracy and achieves $97\%$ of full-parameter fine-tuning accuracy while requiring only $25\%$ of the fine-tuning steps.

模型微调强化学习资源优化持续学习

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