Speculative Probing: LLM Monitoring at Speculative-Decoding Cost
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
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
- 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,878
- 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,769
- 2022Fast Inference from Transformers via Speculative Decodingcited by 1,615
- 2016Understanding intermediate layers using linear classifier probescited by 1,279
- 2023Accelerating Large Language Model Decoding with Speculative Samplingcited by 975
- 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
- 2024Medusa: Simple LLM Inference Acceleration Framework with Multiple Decoding Headscited by 821
- 2024EAGLE: Speculative Sampling Requires Rethinking Feature Uncertaintycited by 564
- 2024WildGuard: Open One-Stop Moderation Tools for Safety Risks, Jailbreaks, and Refusals of LLMscited by 460
- 2024The Instruction Hierarchy: Training LLMs to Prioritize Privileged Instructionscited by 441
- 2024EAGLE-2: Faster Inference of Language Models with Dynamic Draft Treescited by 362
10 more
- 2024Better & Faster Large Language Models via Multi-token Predictioncited by 344
- 2025Monitoring Reasoning Models for Misbehavior and the Risks of Promoting Obfuscationcited by 338
- 2025EAGLE-3: Scaling up Inference Acceleration of Large Language Models via Training-Time Testcited by 318
- 2025Layer by Layer: Uncovering Hidden Representations in Language Modelscited by 293
- 2024Frontier Models are Capable of In-context Schemingcited by 252
- 2024Semantic Entropy Probes: Robust and Cheap Hallucination Detection in LLMscited by 246
- 2025General-Reasoner: Advancing LLM Reasoning Across All Domainscited by 132
- 2024Hydra: Sequentially-Dependent Draft Heads for Medusa Decodingcited by 113
- 2026Building Production-Ready Probes For Geminicited by 38
- 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
- 2025Speculative Decoding: Performance or Illusion?
- 2025Automatic Task Detection and Heterogeneous LLM Speculative Decoding
- 2026SPEED-Bench: A Unified and Diverse Benchmark for Speculative Decoding
- 2025Scaling Up, Speeding Up: A Benchmark of Speculative Decoding for Efficient LLM Test-Time Scaling
- 2024Closer Look at Efficient Inference Methods: A Survey of Speculative Decoding
- 2024Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decoding
- 2023Accelerating LLM Inference with Staged Speculative Decoding
- 2025AdaSD: Adaptive Speculative Decoding for Efficient Language Model Inference