mlsanityData to 7 Oct 2026

Speculative Probing: LLM Monitoring at Speculative-Decoding Cost

arXiv 2608.28099 ↗First posted 28 Aug 2026cs.AI cs.CLcited by 0

Not on the map. Why?

Abstract (arXiv)

Real-time classification during language model inference is valuable for safety filtering, behavioral analysis, and model monitoring, but current approaches force a trade-off between accuracy and efficiency. Hidden-state probes are fast but limited: they are either not context-aware: operating on a single vector and cannot model interactions across positions; or they are very costly: having dedicated classifier models (Llama Guard, Qwen Guard, LLM-as-judge) or performing computation on hidden states for all tokens and then pooling the results (MultiMax). This shows an intrinsic trade-off between efficiency and accuracy. However, we find that the speculative-decoding module in recent LLMs can be repurposed for efficient high-quality classification. By appending a trained soft prompt at the end of the target sequence, we can repurpose the speculative-decoding module into a sequence classifier. At inference time in a speculative-decoding pipeline, the KV cache is already in GPU memory, so classification adds negligible overhead. We evaluate on four classification tasks across four models (Qwen3.5-4B, 9B, 27B, MiniCPM4.1-8B). Our small probes consistently outperform zero-shot GPT-5.4-mini and, on multilingual prompt safety, match or beat specialized 8B safety classifiers (Qwen3Guard-Gen-8B, Llama-Guard-3-8B) without running a full LLM.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 22 of its 30 references that are in the map. First posted on arXiv 28 Aug 2026.

Builds on

22 of 30 references.

References · 22

  1. 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
  2. 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,878
  3. 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,769
  4. 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
  5. 2016Understanding intermediate layers using linear classifier probescited by 1,279
  6. 2023Accelerating Large Language Model Decoding with Speculative Samplingcited by 975
  7. 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
  8. 2024Medusa: Simple LLM Inference Acceleration Framework with Multiple Decoding Headscited by 821
  9. 2024EAGLE: Speculative Sampling Requires Rethinking Feature Uncertaintycited by 564
  10. 2024WildGuard: Open One-Stop Moderation Tools for Safety Risks, Jailbreaks, and Refusals of LLMscited by 460
  11. 2024The Instruction Hierarchy: Training LLMs to Prioritize Privileged Instructionscited by 441
  12. 2024EAGLE-2: Faster Inference of Language Models with Dynamic Draft Treescited by 362
10 more
  1. 2024Better & Faster Large Language Models via Multi-token Predictioncited by 344
  2. 2025Monitoring Reasoning Models for Misbehavior and the Risks of Promoting Obfuscationcited by 338
  3. 2025EAGLE-3: Scaling up Inference Acceleration of Large Language Models via Training-Time Testcited by 318
  4. 2025Layer by Layer: Uncovering Hidden Representations in Language Modelscited by 293
  5. 2024Frontier Models are Capable of In-context Schemingcited by 252
  6. 2024Semantic Entropy Probes: Robust and Cheap Hallucination Detection in LLMscited by 246
  7. 2025General-Reasoner: Advancing LLM Reasoning Across All Domainscited by 132
  8. 2024Hydra: Sequentially-Dependent Draft Heads for Medusa Decodingcited by 113
  9. 2026Building Production-Ready Probes For Geminicited by 38
  10. 2025CultureGuard: Towards Culturally-Aware Dataset and Guard Model for Multilingual Safety Applicationscited by 24

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Speculative Decoding: Performance or Illusion?
  2. 2025Automatic Task Detection and Heterogeneous LLM Speculative Decoding
  3. 2026SPEED-Bench: A Unified and Diverse Benchmark for Speculative Decoding
  4. 2025Scaling Up, Speeding Up: A Benchmark of Speculative Decoding for Efficient LLM Test-Time Scaling
  5. 2024Closer Look at Efficient Inference Methods: A Survey of Speculative Decoding
  6. 2024Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decoding
  7. 2023Accelerating LLM Inference with Staged Speculative Decoding
  8. 2025AdaSD: Adaptive Speculative Decoding for Efficient Language Model Inference