Semantic Overlays: Mitigating Prompt Injection with Annotations Beyond Tokens and Steering Vectors

작성자

카테고리:

← 피드로
arXiv cs.AI · Joshua Penman · 2026-08-31 AI

[Submitted on 24 Aug 2026 (v1), last revised 28 Aug 2026 (this version, v2)]

View PDF HTML (experimental)

Abstract:Everything a language model sees is tokens. The serving stack knows what each span is — user input, tool output, instructions — but the model must keep track of that itself, and can lose track or be confused: text can be written to read like anything. Prompt injection is a natural exploit of this phenomenon. By scrambling the model’s understanding of span identity, an attacker can induce unwanted and dangerous actions. Adding a non-textual channel to the model’s input — a way to communicate span identity beyond text — mitigates this class of attack. We thus introduce a general steering technique called Semantic Overlays: small learned adapters applied at chosen prefill positions to a frozen model’s residual stream. Laying an overlay over a span creates an out-of-band annotation channel that cannot be replicated by tokens. Unlike steering vectors, Semantic Overlays are trained, adaptable, and selectively applied. An overlay can encode complex semantics that reshape how the model perceives the marked span: asked to copy a code snippet under an overlay asserting a different programming language, the model rewrites the snippet in the asserted language. Overlays compose, allow transparent reading of underlying content, and can carry complex payloads — including imperatives the model will follow. An overlay which marks a span as “non-executable” defends against the broad class of prompt injections that add instructions in untrusted context. We report strong results on five prompt injection benchmarks: SEP separation rises from 24.3% to 99.0% with utility unchanged (our scoring rule; we correct a defect in the published grader), TensorTrust attack success falls from 34.8% to 6.2%, AlpacaFarm from 99.0% to 0%, and the overlay beats every published PIArena defense that leaves the model able to answer — while marked spans stay readable, all at >95% character similarity to the original.

Submission history

From: Joshua Penman [view email]
[v1] Mon, 24 Aug 2026 22:26:58 UTC (114 KB)
[v2] Fri, 28 Aug 2026 09:49:45 UTC (118 KB)

원문에서 계속 ↗

추출 본문 · 출처: arxiv.org · https://arxiv.org/abs/2608.23873