Mar 23, 2026

TaintGuard

Revathi Prasad

TaintGuard is a system that augments existing LLM-based monitors with

dynamic data-flow taint tracking to detect sabotage in AI agent trajectories. Taint tracking provides a structurally different signal from LLM monitoring — it detects data-flow violations (exfiltration, integrity writes, code injection) that are provable and immune to prompt injection, while LLM monitors detect semantic intent.

Reviewer's Comments

Reviewer's Comments

Arrow
Arrow
Arrow

Interesting project!

- I like layer 1. It's undeniable that good control protocols should take advantage of such cheap detectors. However, it's important to test, or discuss when time is limited, the strategies a potential Red Team (RT) could come up with to avoid triggering your monitor.

- Layer 2 seems more interesting in practice.

Most important in the future: show results capping the audit budget (Honest FPR) to a threshhold. It's easy have great attack catch rate, at the cost of catching every non-attack. It's important to give an idea of how FPR trades off against recall, and here it's hard to tell.

I'm quite interested in using more traditional security approaches to aid in monitoring/blue team protocols and this was an interesting exploration of that IMO.

I would like to have seen more work to make the FPR a bit lower, and would have liked to see ROC curves to properly compare the techniques.

I think that implementing the taint-finding-aware validation idea would have been interesting. Also I would maybe use larger numbers of trajs, though it makes sense that you were resource/time constrained there since this was a weekend project.

Overall, I thought that this was quite good work.

Cite this work

@misc {

title={

(HckPrj) TaintGuard

},

author={

Revathi Prasad

},

date={

3/23/26

},

organization={Apart Research},

note={Research submission to the research sprint hosted by Apart.},

howpublished={https://apartresearch.com}

}

Recent Projects

Feb 2, 2026

Markov Chain Lock Watermarking: Provably Secure Authentication for LLM Outputs

We present Markov Chain Lock (MCL) watermarking, a cryptographically secure framework for authenticating LLM outputs. MCL constrains token generation to follow a secret Markov chain over SHA-256 vocabulary partitions. Using doubly stochastic transition matrices, we prove four theoretical guarantees: (1) exponentially decaying false positive rates via Hoeffding bounds, (2) graceful degradation under adversarial modification with closed-form expected scores, (3) information-theoretic security without key access, and (4) bounded quality loss via KL divergence. Experiments on 173 Wikipedia prompts using Llama-3.2-3B demonstrate that the optimal 7-state soft cycle configuration achieves 100\% detection, 0\% FPR, and perplexity 4.20. Robustness testing confirms detection above 96\% even with 30\% word replacement. The framework enables $O(n)$ model-free detection, addressing EU AI Act Article 50 requirements. Code available at \url{https://github.com/ChenghengLi/MCLW}

Read More

Feb 2, 2026

Prototyping an Embedded Off-Switch for AI Compute

This project prototypes an embedded off-switch for AI accelerators. The security block requires periodic cryptographic authorization to operate: the chip generates a nonce, an external authority signs it, and the chip verifies the signature before granting time-limited permission. Without valid authorization, outputs are gated to zero. The design was implemented in HardCaml and validated in simulation.

Read More

Feb 2, 2026

Fingerprinting All AI Cluster I/O Without Mutually Trusted Processors

We design and simulate a "border patrol" device for generating cryptographic evidence of data traffic entering and leaving an AI cluster, while eliminating the specific analog and steganographic side-channels that post-hoc verification can not close. The device eliminates the need for any mutually trusted logic, while still meeting the security needs of the prover and verifier.

Read More

Feb 2, 2026

Markov Chain Lock Watermarking: Provably Secure Authentication for LLM Outputs

We present Markov Chain Lock (MCL) watermarking, a cryptographically secure framework for authenticating LLM outputs. MCL constrains token generation to follow a secret Markov chain over SHA-256 vocabulary partitions. Using doubly stochastic transition matrices, we prove four theoretical guarantees: (1) exponentially decaying false positive rates via Hoeffding bounds, (2) graceful degradation under adversarial modification with closed-form expected scores, (3) information-theoretic security without key access, and (4) bounded quality loss via KL divergence. Experiments on 173 Wikipedia prompts using Llama-3.2-3B demonstrate that the optimal 7-state soft cycle configuration achieves 100\% detection, 0\% FPR, and perplexity 4.20. Robustness testing confirms detection above 96\% even with 30\% word replacement. The framework enables $O(n)$ model-free detection, addressing EU AI Act Article 50 requirements. Code available at \url{https://github.com/ChenghengLi/MCLW}

Read More

Feb 2, 2026

Prototyping an Embedded Off-Switch for AI Compute

This project prototypes an embedded off-switch for AI accelerators. The security block requires periodic cryptographic authorization to operate: the chip generates a nonce, an external authority signs it, and the chip verifies the signature before granting time-limited permission. Without valid authorization, outputs are gated to zero. The design was implemented in HardCaml and validated in simulation.

Read More

This work was done during one weekend by research workshop participants and does not represent the work of Apart Research.
This work was done during one weekend by research workshop participants and does not represent the work of Apart Research.