Attacks on Approximate Caches in Text-to-Image Diffusion Models
About
Diffusion models are a powerful class of generative models that produce images and other content from user prompts, but they are computationally intensive. To mitigate this cost, recent academic and industry work has adopted approximate caching, which reuses intermediate states from similar prompts in a cache. While efficient, this optimization introduces new security risks by breaking isolation among users. This paper provides a comprehensive assessment of the security vulnerabilities introduced by approximate caching. First, we demonstrate a remote covert channel established with the approximate cache, where a sender injects prompts with special keywords into the cache system and a receiver can recover that even after days, to exchange information. Second, we introduce a prompt stealing attack using the approximate cache, where an attacker can recover existing cached prompts from hits. Finally, we introduce a poisoning attack that embeds the attacker's logos into the previously stolen prompt, leading to unexpected logo rendering for the requests that hit the poisoned cache prompts. These attacks are all performed remotely through the serving system, demonstrating severe security vulnerabilities in approximate caching. The code for this work is available.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Recovery | DiffusionDB | CLIP Similarity0.9 | 4 | |
| Image Recovery | Lexica | CLIP Score95 | 4 | |
| Poisoning Success Evaluation | DiffusionDB 1M prompts Large | Hit Rate4.93e+3 | 4 | |
| Poisoning Success Evaluation | Lexica | Hit1.23e+3 | 4 | |
| Prompt Recovery | DiffusionDB | BLEU25 | 4 | |
| Prompt Recovery | Lexica | BLEU22 | 4 |