mlsanityData to 14 Oct 2026

Next Token Perception Score: Analytical Assessment of your LLM Perception Skills

arXiv 2505.17169 ↗First posted 22 May 2025cs.CL cs.AIcited by 0

On the map: Mechanistic interpretability → SAE features and their applications

Abstract (arXiv)

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.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in SAE features and their applications, one of 13 directions in Mechanistic interpretability. Builds on 21 of its 35 references that are in the map. First posted on arXiv 22 May 2025.

Builds on

21 of 35 references.

References · 21

  1. 2020Language Models are Few-Shot Learnerscited by 32,419
  2. 2023GPT-4 Technical Reportcited by 18,396
  3. 2024The Llama 3 Herd of Modelscited by 14,715
  4. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
  5. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  6. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
  7. 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
  8. 2022Fine-Tuning can Distort Pretrained Features and Underperform Out-of-Distributioncited by 739
  9. 2019What do you learn from context? Probing for sentence structure in contextualized word representationscited by 438
  10. 2020Efficient Intent Detection with Dual Sentence Encoderscited by 433
  11. 2020Evaluation of Neural Architectures Trained with Square Loss vs Cross-Entropy in Classification Taskscited by 138
  12. 2024Language models scale reliably with over-training and on downstream taskscited by 113
9 more
  1. 2022Same Pre-training Loss, Better Downstream: Implicit Bias Matters for Language Modelscited by 84
  2. 2024OpenELM: An Efficient Language Model Family with Open Training and Inference Frameworkcited by 75
  3. 2024From Words to Numbers: Your Large Language Model Is Secretly A Capable Regressor When Given In-Context Examplescited by 57
  4. 2024Learning by Reconstruction Produces Uninformative Features For Perceptioncited by 55
  5. 2020Effective Transfer Learning for Identifying Similar Questions: Matching User Questions to COVID-19 FAQscited by 48
  6. 2024Scaling Laws for Predicting Downstream Performance in LLMscited by 48
  7. 2019Eigenvalue and Generalized Eigenvalue Problems: Tutorialcited by 46
  8. 2016Stop Clickbait: Detecting and Preventing Clickbaits in Online News Mediacited by 39
  9. 2024Understanding LLM Embeddings for Regressioncited by 19

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025I Predict Therefore I Am: Is Next Token Prediction Enough to Learn Human-Interpretable Concepts from Data?
  2. 2026LLMs as High-Dimensional Nonlinear Autoregressive Models with Attention: Training, Alignment and Inference
  3. 2025Where MLLMs Attend and What They Rely On: Explaining Autoregressive Token Generation
  4. 2023Measuring Progress in Fine-grained Vision-and-Language Understanding
  5. 2026Representation Collapse in Sequential Post-Training of Large Language Models
  6. 2025Learning to See Before Seeing: Demystifying LLM Visual Priors from Language Pre-training
  7. 2025QuickMerge++: Fast Token Merging with Autoregressive Prior
  8. 2026VIVAS: Vitalizing Visual Perception in VLM Pre-training via Vision-language Unified Autoregressive Supervision