Adversarial Prompts for Acceptance Collapse in Speculative Decoding
Summary
A novel adversarial prompt-suffix attack, ADSD, has been introduced to exploit a vulnerability in speculative decoding, a lossless acceleration scheme for large language model inference. ADSD systematically attacks the dynamic token-level alignment between a draft and a target model, causing "acceptance collapse" by pushing draft probability mass towards tokens the target model is unlikely to accept. The attack employs "Soft-Collapse," a verifier-aligned surrogate derived from the asymmetric speculative acceptance rule, alongside a target-preservation objective to prevent obvious task corruption. Evaluated on the GSM8K dataset, ADSD increased the mean sample time by 62.3% while preserving the original task quality. This vulnerability is demonstrated to exist across various domains, speculative decoding strategies, and model architectures.
Key takeaway
For Machine Learning Engineers deploying LLMs with speculative decoding, you must account for a new adversarial vulnerability. The ADSD attack can significantly increase inference latency by 62.3% without degrading output quality. This impacts operational costs and user experience. Implement robust monitoring for unexpected latency spikes. Consider adversarial training or prompt filtering to mitigate this inference-time attack. Your deployment strategy should factor in this potential for performance degradation.
Key insights
Speculative decoding, while efficient, is vulnerable to adversarial prompts that collapse verifier acceptance without degrading task quality.
Principles
- Speculative decoding relies on dynamic token alignment.
- Draft-target alignment is a systemic vulnerability.
- Attacks can increase inference time without quality loss.
Method
ADSD generates prompt suffixes using Soft-Collapse, a verifier-aligned surrogate, combined with a target-preservation objective. This pushes draft probability mass toward tokens the target model is unlikely to accept, collapsing acceptance.
In practice
- Increase inference time by 62.3% on GSM8K.
- Vulnerability spans domains and architectures.
Topics
- Speculative Decoding
- Adversarial Prompts
- LLM Security
- Inference Optimization
- Acceptance Collapse
- GSM8K Dataset
Best for: Research Scientist, AI Architect, MLOps Engineer, AI Scientist, Machine Learning Engineer, AI Security Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.