提出新指标评估大模型感知能力与预训练的匹配度。
Next Token Perception Score: Analytical Assessment of your LLM Perception Skills
- 用线性设定构建感知特征与预训练特征的重叠度评分
- 该评分与12个数据集上线性探测准确率强相关,可预测微调增益
- 适用于快速筛选适合LoRA微调的模型,尤其对大模型有效
自回归预训练已成为大语言模型学习通用表征的主流范式,但其在下游感知任务上的线性探测性能存在显著差异,表明为下一词预测优化的特征未必能有效迁移到感知任务。我们发现自回归学习的表征可能包含偏离感知关键子空间的成分。为此,提出“下一词感知得分”(NTPS)——一种在线性设置下衡量自回归与感知特征子空间重叠度的指标。该得分可从预训练表征和标注数据中闭式计算,并被证明同时上下界超额损失。实验表明,NTPS在12个多样化NLP数据集、8种预训练模型(270M至8B参数)上与线性探测准确率高度相关,验证其作为对齐度量的有效性。此外,我们发现低秩适应(LoRA)微调后NTPS上升,尤其在大模型中显著,说明LoRA使表征更贴近感知任务。更重要的是,NTPS能可靠预测LoRA带来的额外精度提升,为LoRA适配提供轻量级预筛选工具。研究提供理论洞见与实用分析工具,用于解析评估大模型感知能力。
原文摘要 · Abstract (English)
Autoregressive pretraining has become the de facto paradigm for learning general-purpose representations in large language models (LLMs). However, linear probe performance across downstream perception tasks shows substantial variability, suggesting that features optimized for next-token prediction do not consistently transfer well to downstream perception tasks. We demonstrate that representations learned via autoregression capture features that may lie outside the subspaces most informative for perception. To quantify the (mis)alignment between autoregressive pretraining and downstream perception, we introduce the Next Token Perception Score (NTPS)-a score derived under a linear setting that measures the overlap between autoregressive and perception feature subspaces. This metric can be easily computed in closed form from pretrained representations and labeled data, and is proven to both upper- and lower-bound the excess loss. Empirically, we show that NTPS correlates strongly with linear probe accuracy across 12 diverse NLP datasets and eight pretrained models ranging from 270M to 8B parameters, confirming its utility as a measure of alignment. Furthermore, we show that NTPS increases following low-rank adaptation (LoRA) fine-tuning, especially in large models, suggesting that LoRA aligning representations to perception tasks enhances subspace overlap and thus improves downstream performance. More importantly, we find that NTPS reliably predicts the additional accuracy gains attained by LoRA finetuning thereby providing a lightweight prescreening tool for LoRA adaptation. Our results offer both theoretical insights and practical tools for analytically assessing LLM perception skills.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。