Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Desen Sun, Shuncheng Jie, Sihang Liu• 2025

Related benchmarks

TaskDatasetResultRank
Image RecoveryDiffusionDB
CLIP Similarity0.9
4
Image RecoveryLexica
CLIP Score95
4
Poisoning Success EvaluationDiffusionDB 1M prompts Large
Hit Rate4.93e+3
4
Poisoning Success EvaluationLexica
Hit1.23e+3
4
Prompt RecoveryDiffusionDB
BLEU25
4
Prompt RecoveryLexica
BLEU22
4
Showing 6 of 6 rows

Other info

Follow for update