Privacy from Symmetry: Orthogonally Equivariant Transformers for LLM Inference

Alexander Yukhimchuk, Andrey Shulga, Mladen Kolar, Martin Takáč
6/15/2026
cs.LG

No confident code implementation found for this paper.

Abstract

Running large language models locally is often impractical, pushing inference on sensitive text to third-party providers. Split inference partially mitigates this by keeping tokens on the client and sending only hidden representations, but these representations can still be recovered via nearest-neighbor search against the public embedding table. We propose an orthogonal obfuscation procedure in which the client multiplies embeddings by a secret orthogonal matrix before transmission. To enable correct inference under arbitrary rotations, we introduce ConjFormer, a transformer variant that is exactly $\mathrm{O}(d)$-equivariant via a lightweight normalization change (scalar RMSNorm) together with blockwise orthogonal conjugation of all linear weights. As a result, the server performs the full forward pass entirely in the rotated basis and never observes unrotated hidden states. Experiments on GPT-2 and Llama 3.2 1B models fine-tuned on PubMed show that orthogonal obfuscation eliminates direct cosine nearest-neighbor inversion and reduces token recovery from over 35% top-10 to at most 1.3%, while increasing perplexity by only 0.4% after fine-tuning. These results indicate that enforcing symmetry at the architectural level can provide a practical defense for privacy-preserving LLM inference without noise injection or heavy cryptographic machinery.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Cite this paper

@article{yukhimchuk2026privacy,
  title  = {Privacy from Symmetry: Orthogonally Equivariant Transformers for LLM Inference},
  author = {Alexander Yukhimchuk and Andrey Shulga and Mladen Kolar and Martin Takáč},
  year   = {2026},
  eprint = {2606.16461},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.16461v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2606.16461.svg)](https://codeofpaper.com/paper/2606.16461)

Discussion