arXiv:2604.01025cs.LGcs.AI2026-04

用轻量探针实时预测大模型训练中下游表现,提速近20倍。

Fast and Accurate Probing of In-Training LLMs' Downstream Performances

  • 用内部表征做输入,轻量探针直接预测下游任务成功率。
  • 平均AUROC超0.75,能准确预判模型性能,且早期可预测后期。
  • 将评估耗时从1小时降至3分钟,适合快速迭代训练。

随着大语言模型参数规模和推理时间的持续增长,传统生成式评估方式变得成本过高,导致模型训练期间下游性能评估延迟难以接受。然而,训练损失(困惑度)等简单指标常与实际下游性能脱节,趋势可能偏离任务真实结果。为此,本文提出一种新的训练中评估范式:使用轻量探针,以模型检查点的内部表征为输入,直接预测其在下游任务上的表现(以成功概率pass@1衡量)。在OLMo3-7B的多个检查点上验证了多种探针架构的有效性。结果表明,该方法能准确预测性能(平均AUROC > 0.75),具备良好跨检查点泛化能力(早期预测后期),并将计算延迟从约1小时降低至约3分钟。该工作实现了高效、可靠的训练中下游评估,推动更敏捷、智能的大模型开发流程。

原文摘要 · Abstract (English)

The paradigm of scaling Large Language Models (LLMs) in both parameter size and test time has pushed the boundaries of AI capabilities, but at the cost of making the traditional generative evaluation paradigm prohibitively expensive, therefore making the latency of LLM's in-training downstream performance evaluation unbearable. However, simple metrics like training loss (perplexity) are not always correlated with downstream performance, as sometimes their trends diverge from the actual task outcomes. This dilemma calls for a method that is computationally efficient and sufficiently accurate in measuring model capabilities. To address this challenge, we introduce a new in-training evaluation paradigm that uses a lightweight probe for monitoring downstream performance. The probes take the internal representations of LLM checkpoints (during training) as input and directly predict the checkpoint's performance on downstream tasks measured by success probability (i.e., pass@1). We design several probe architectures, validating their effectiveness using the OLMo3-7B's checkpoints across a diverse set of downstream tasks. The probes can accurately predict a checkpoint's performance (with avg. AUROC$>$0.75), have decent generalizability across checkpoints (earlier predicts later), and reduce the computation latency from $\sim$1 hr (using conventional generative evaluation method) to $\sim$3 min. In sum, this work presents a practical and scalable in-training downstream evaluation paradigm, enabling a more agile, informed, and efficient LLM development process.

大模型评估轻量探针训练监控

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