arXiv:2606.17660cs.LGcs.AI2026-06中稿 · ICML

用短探针预判大模型微调效果,避免无效训练

TuneAhead: Predicting Fine-tuning Performance Before Full Training Begins

论文配图:TuneAhead: Predicting Fine-tuning Performance Before Full Training Begins
图 1 · 摘自论文原文
  • 用静态数据特征+短探针动态特征生成元特征向量
  • 测试集上预测误差仅1.47个百分点,95%预测在±3%内
  • 可辅助决策是否继续训练,适合资源有限的团队

微调大型语言模型计算成本高且易出错:性能对数据质量和超参数敏感,盲目训练甚至可能降低性能。我们提出TUNEAHEAD,一个轻量级框架,可在完整训练前预判微调表现。该框架将每个候选训练任务编码为包含静态数据描述符与短标准化探针的动态特征向量,通过预测器映射到性能估计值,并利用SHAP方法提供可解释诊断,揭示关键驱动因素。在Qwen2.5-7B-Instruct上的1,300+次微调实验中,TUNEAHEAD持续优于早期停止外推和ProxyLM等强基线。在370次保留测试集上,均方根误差为1.47百分点,95.1%的预测值位于真实得分±3百分点范围内。准确的连续预测支持实用的是否继续筛选策略,可减少不必要的完整微调,同时保留大部分有潜力的训练任务。

原文摘要 · Abstract (English)

Fine-tuning large language models (LLMs) is compute-intensive and error-prone: model performance depends sensitively on data quality and hyperparameter choices, and naïve runs can even degrade model performance. This raises a practical question:can we predict fine-tuning performance before committing to a full training run? We present TUNEAHEAD, a lightweight framework for pre-hoc prediction of fine-tuning performance. TUNEAHEAD encodes each candidate run as a meta-feature vector that combines static dataset descriptors with dynamic probe features from a short standardized probe. A predictor maps these features to performance estimates, while SHAP-based attributions provide interpretable diagnostics that reveal which specific features drive the prediction. Across 1,300+ fine-tuning runs on Qwen2.5-7B-Instruct, TUNEAHEAD consistently outperforms strong baselines such as Early-Stop Extrapolation and ProxyLM. On a held-out test set of 370 runs, TUNEAHEAD achieves an RMSE of 1.47 percentage points and places 95.1% of predictions within +3/-3 percentage points of the true score. These accurate continuous predictions support practical go/no-go screening policies that can reduce unnecessary full fine-tuning while retaining most promising runs.

微调预测轻量推理模型评估高效训练

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