atlas

Research map / Large language models / Language, safety and interpretability

Hidden No More: Attacking and Defending Private Third-Party LLM Inference

ICML 2025 · Large language models › Language, safety and interpretability · approach: Jailbreak elicitation and risk

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Shows that permutation- and noise-based private LLM inference schemes leak prompts via a vocabulary-matching hidden-state reversal attack, and proposes Cascade, a token-sharded multi-party inference scheme that resists it.

Problem

Private third-party LLM inference: hidden states sent to inference parties may reveal user prompts, and existing permutation/noise-based defenses were assumed secure without rigorous attack analysis.

Approach

A vocabulary-matching attack reverses hidden states token-by-token by running forward passes over the vocabulary (using the open weights and a proposal model with L1-threshold matching), which remains effective under sequence, hidden-dimension, and factorized permutations and noise. Defense: Cascade shards tokens across CompNodes and AttnNodes with (c, δ)-clustered-arithmetic sharding so that toke

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
apply vocab-matching to ordinary hidden statesGemma-2-2B-IT and Llama-3.1-8B-Instruct · FineWeb-Eduperfectly decoded samples (%): 100% (Gemma layers 1–26); 99.7%–100% (Llama layers 1–26)Helped
apply sequence-dimension permutation to hidden statesVocab-matching attack · FineWeb-Edu, layers 1–26perfectly decoded samples (%): 99.8%–100% (Gemma); 99.7%–100% (Llama)Helped
apply hidden-dimension permutation to hidden statesVocab-matching attack · FineWeb-Edu, layers 1–26perfectly decoded samples (%): 98.0%–100% (Gemma); 98.2%–100% (Llama)Helped
apply factorized-2D permutation to hidden statesVocab-matching attack · FineWeb-Edu, layers 1–26perfectly decoded samples (%): 99.0%–99.9% (Gemma); 97.6%–98.9% (Llama)Helped
increase c or α in (c, δ)-sharding (ablation)Cascade · FineWeb-Edu, layer 1 hiddensROUGE-L reconstruction score: c=1: 0.701 -> 0.467 -> 0.349; α=4, c=1 -> c=4 -> c=8: 0.701 -> 0.427 -> 0.355Hurt
use Cascade, α=8 instead of α=1 (ablation)Cascade · Bert-Base, 128-token forward passruntime (s): 0.32 [0.31, 0.36] -> 0.74 [0.62, 0.96]Hurt
use Cascade, α=8 instead of α=1 (ablation)Cascade · Bert-Large, 128-token forward passruntime (s): 1.01 [0.97, 1.09] -> 1.58 [1.27, 1.97]Hurt
use Cascade instead of MPCFormer or PumaBert-Base and Bert-Large inference · 128-token forward passruntime (s): Cascadeα=8: 0.74 [0.62, 0.96] / 1.58 [1.27, 1.97]; MPCFormer: 55.320 / 141.222; Puma: 33.913 / 73.720Helped
use Cascade instead of MPCFormer or PumaBert-Base and Bert-Large inference · 128-token forward passcommunicated bytes (GB): Cascadeα=8: 0.076 / 0.203; MPCFormer: 12.089 / 32.577; Puma: 10.773 / 27.246Helped

Surprising findings

Limitations

Experimental setup

Datasets: FineWeb-Edu FineWeb (CC-MAIN-2024-10 split)

Benchmarks: FineWeb-Edu

Metrics: percentage of perfectly decoded samples ROUGE-L BLEU runtime (seconds) communicated bytes (GB)

Baselines: Wan et al. 2024 MPCFormer Puma Vanilla inference PermLLM STIP Centaur

Builds on: Wan et al. 2024 hidden state inversion Morris et al. 2023 text embedding inversion PermLLM STIP Centaur MPCFormer Puma

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy