Adversarial Prompts for Acceptance Collapse in Speculative Decoding

· Source: Computation and Language · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Cybersecurity & Data Privacy · Depth: Expert, quick

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

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

Topics

Best for: Research Scientist, AI Architect, MLOps Engineer, AI Scientist, Machine Learning Engineer, AI Security Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.