HyperAIHyperAI

Command Palette

Search for a command to run...

拡散モデルにおける概念抑制の回避に向けた概念算術

Vitali Petsiuk Kate Saenko

概要

倫理的・法的な懸念を背景に、科学コミュニティは、テキスト-画像拡散モデルが著作権で保護されたコンテンツ、暴力的・露骨な表現、あるいは個人情報を含む画像を生成する目的で悪用されることを制限する手法を積極的に開発している。並行して、研究者たちは新たに開発された安全対策に対して敵対者の役割を担い、脆弱性やバックドアを発見する試みを行っている。本研究では、単一の画像生成において複数のプロンプトを活用することを可能にする拡散モデルの合成性を用いる。この性質により、対象概念の生成を担当するベクトルへの直接的なアクセスがもはや不可能になっている場合であっても、抑制の影響を受けるべきでない他の概念を組み合わせることで、当該ベクトルを再構成することが可能となる。提案する攻撃が成立する理由について理論的および経験的な証拠を提供し、これらの発見が安全なモデル展開にもたらす影響について議論する。敵対者によって採用されうる拡散モデルを用いた画像生成のあらゆる手法を考慮することが不可欠であると主張する。本研究は、概念算術と合成的推論が拡散モデルの安全機構にもたらす影響に関する議論を開くものである。コンテンツ注意:本論文には、不快と見なされる可能性のある議論やモデル生成コンテンツへの言及が含まれています。読者の判断で閲覧してください。

One-sentence Summary

Researchers at Boston University demonstrate that concept arithmetics exploits the compositional property of text-to-image diffusion models to reconstruct vectors for inhibited target concepts by combining prompts unaffected by suppression, revealing vulnerabilities in current safety mechanisms against adversarial image generation.

Key Contributions

  • Presents a compositional inference attack that combines multiple prompts to reconstruct the vector responsible for target concept generation in inhibited text-to-image diffusion models, without requiring model-weight access or prompt optimization.

  • Establishes, through theoretical and empirical evidence, that concept-inhibition defenses that alter conditional guidance only near the target concept are vulnerable, and concludes that inhibited models should remain combined with guardrails in high-risk settings.

  • Introduces a framework linking the linearity of conditional guidance to compositional generation, showing that watermarking or editing methods with locally modified guidance can be bypassed unless model behavior is changed globally.

Introduction

The authors study the safety of Text-to-Image diffusion models, which are widely used for creative and commercial image generation but can also produce deep fakes, harmful content, copyrighted styles, and biased outputs. To reduce this risk, prior concept inhibition methods modify model weights so the model appears unable to generate a target concept, unlike safety filters that can be bypassed at inference time. However, these methods mainly address single-prompt generation and assume that local weight edits permanently erase the target information.

The authors show that this assumption fails when models support compositional inference, where multiple prompts or guidance terms can be combined during generation. Their ARC attacks use concept arithmetics with multiple input points that are far from the erased concept but still encode information about it, allowing black-box reconstruction of prohibited content. They provide theoretical grounding and experiments showing that compositional inference significantly increases the reproduction rate of concepts suppressed by existing inhibition methods.

Method

The proposed approach is an inference-only compositional attack framework for circumventing concept inhibition in diffusion models. The authors leverage the linearity of conditional guidance in semantic space rather than modifying model weights. Given a diffusion model with noise prediction ϵθ\epsilon_\thetaϵθ​, conditional guidance for a concept cjc_jcj​ is defined as:

g(cj)=defγ(ϵθ(xt,cj,t)−ϵθ(xt,t)).g(c_j) \stackrel{\mathrm{def}}{=} \gamma(\epsilon_\theta(x_t, c_j, t) - \epsilon_\theta(x_t, t)).g(cj​)=defγ(ϵθ​(xt​,cj​,t)−ϵθ​(xt​,t)).

Inhibition methods typically modify model weights so that the guidance at a target concept ctc_tct​ matches a desired value y0y_0y0​. Formally, the inhibited weights are obtained through an objective such as:

θ~=min⁡θ∣∣L(gθ(ct),y0)∣∣.\tilde{\theta} = \min_{\theta} || \mathcal{L}(g_\theta(c_t), y_0) ||.θ~=θmin​∣∣L(gθ​(ct​),y0​)∣∣.

The attack assumes that the inhibited guidance function g(c)g(c)g(c) can be approximated as a weighted mixture between the original guidance g∗(c)g^*(c)g∗(c) and the inhibition target y0y_0y0​:

g(c)=λ(c)⋅y0+(1−λ(c))⋅g∗(c).g(c) = \lambda(c) \cdot y_0 + (1 - \lambda(c)) \cdot g^*(c).g(c)=λ(c)⋅y0​+(1−λ(c))⋅g∗(c).

Here, λ(c)\lambda(c)λ(c) represents the degree of modification at concept ccc. The method relies on the hypothesis that modification is localized around ctc_tct​, so concepts farther from ctc_tct​ experience weaker distortion. When semantic linearity holds, a sufficiently distant detour concept cdc_dcd​ can be used to recover a guidance vector close to the original target guidance:

g(ct±cd)∓g(cd)→g∗(ct).g(c_t \pm c_d) \mp g(c_d) \rightarrow g^*(c_t).g(ct​±cd​)∓g(cd​)→g∗(ct​).

This relationship is the core of the attack design. Instead of requesting the inhibited target concept directly, the authors query the model with composed concepts and combine the resulting guidance vectors through arithmetic in concept space. To preserve the user's intended image content, the attack guidance is added to the original prompt guidance. For a standard inference step

ϵθ(xt,t)+g(c1),\epsilon_\theta(x_t, t) + g(c_1),ϵθ​(xt​,t)+g(c1​),

the compositional attack replaces it with

ϵθ(xt,t)+g(c1)+g(ct+cd)−g(cd).\epsilon_\theta(x_t, t) + g(c_1) + g(c_t + c_d) - g(c_d).ϵθ​(xt​,t)+g(c1​)+g(ct​+cd​)−g(cd​).

This creates a detour path that bypasses the inhibited target computation. As shown in the figure below:

The framework includes several attack designs. One detour subtraction attack manually selects a concept cdc_dcd​ that is semantically distant from the target and forms a composed concept ct+cdc_t + c_dct​+cd​ that remains close to cdc_dcd​. For example, a target zebra concept can be approximated by using a cake-based detour concept, because “cake in the shape of zebra” is closer to “cake” than “cake and zebra,” thereby reducing the degree of target-specific inhibition. A stacked version of this attack combines multiple detour concepts:

∑i=1N[g(ct±cdi)∓g(cdi)]→N⋅g∗(ct).\sum_{i=1}^{N} \left[ g(c_t \pm c_d^i) \mp g(c_d^i) \right] \rightarrow N \cdot g^*(c_t).i=1∑N​[g(ct​±cdi​)∓g(cdi​)]→N⋅g∗(ct​).

Another variant omits the subtraction of g(cd)g(c_d)g(cd​) when biased generation is acceptable. In this case, the attack directly uses composed or subset prompts, such as replacing a target concept with a broader concept or a more specific concept. These variants exploit the fact that moving away from the target concept reduces the influence of the inhibition direction.

The authors also introduce an anchor-difference attack. If an anchor concept cac_aca​ is unaffected by inhibition, then the difference between target and anchor guidance remains aligned with the original difference:

g(ct)−g(ca)=(1−λ(ct))(g∗(ct)−g∗(ca)).g(c_t) - g(c_a) = (1 - \lambda(c_t)) (g^*(c_t) - g^*(c_a)).g(ct​)−g(ca​)=(1−λ(ct​))(g∗(ct​)−g∗(ca​)).

This allows the attacker to construct guidance that emphasizes the target concept while suppressing the anchor concept. The attack requires only access to compositional inference, either through a black-box API or through model weights, and its main computational cost is additional forward passes for each auxiliary concept.

Experiment

The experiments evaluate the proposed compositional attacks against Stable Diffusion 1.4 models inhibited for nudity and selected object categories, comparing standard inference with original and inhibited models against multiple attack modes. They show that inhibition can reduce target concept presence under normal use but does not remove it, while the attacks recover nudity or object content, sometimes even more reliably than the baseline. The findings extend to object and character inhibition, where the attacks substantially increase target concept reproduction across AC, ESD u, and UCE, although stronger fine tuning weakens the effect and aggressive inhibition may harm generation quality.

The table describes adversarial prompts that use concept arithmetic and attacker chosen concepts to recover guidance toward an erased target concept in inhibited diffusion models. The evaluated attacks consistently increase target concept reproduction compared with standard inference, and they remain effective across inhibition methods though stronger inhibition reduces their impact. Stacking the attacks can strengthen the recovered target direction, while stronger inhibition settings can reduce overall image quality and variation. Attacks combine the target concept with an arbitrary concept to make guidance align with the erased target. The attacks increase reproduction at high CLIP score thresholds, indicating more pronounced target presence. Stronger inhibition dampens but does not fully prevent target concept recovery.

The attacks circumvent object and nudity inhibition by adding extra guidance for composed prompts, related decoy concepts, or explicit concept terms. They aim to approximate the target concept guidance while keeping the prompt change as small as possible or suppressing an anchor concept. Across tested inhibition methods, the attacks produce noticeably higher reproduction of the target concept, especially when the inhibited model still allows some concept presence. Object attacks combine a target concept with a neutral decoy concept to reduce direct targeting. One object attack stacks multiple guidance components for shapes, text, and songs. Nudity attacks add guidance for explicit concept terms or related phrases. Using an anchor concept can help emphasize the target concept while avoiding mixed outputs.

The experiments evaluate adversarial prompt attacks against concept inhibition in diffusion models, using target plus decoy, anchor, explicit, or stacked guidance. They find that such attacks consistently recover erased target concepts relative to standard inference across inhibition methods, particularly when partial concept presence remains. Stronger inhibition weakens recovery and harms image quality, while stacking guidance and carefully chosen anchor or decoy concepts make circumvention more pronounced.


AIでAIを構築

アイデアからローンチまで — 無料のAIコーディング支援、すぐに使える環境、最高のGPU価格でAI開発を加速。

AI コーディング補助
すぐに使える GPU
最適な料金体系

HyperAI Newsletters

最新情報を購読する
北京時間 毎週月曜日の午前9時 に、その週の最新情報をメールでお届けします
メール配信サービスは MailChimp によって提供されています