atlas

Research map / Large language models / Instruction and fine-tuning

SentinelLMs: Encrypted Input Adaptation and Fine-Tuning of Language Models for Private and Secure Inference

AAAI 2024 · Large language models › Instruction and fine-tuning · approach: Safety alignment after fine-tuning · 7 citations (OpenAlex)

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.

Adapts pre-trained BERT/RoBERTa/mBERT to passkey-encrypted tokens (Blake2 tokenizer encryption, glide-reflection embedding transforms, vocabulary shuffling), then fine-tunes on encrypted data, reporting near-parity with original models on GLUE, CoNLL2003 NER, and XNLI.

Problem

Text sent to cloud-hosted language models can be intercepted or logged by servers, but naive encryption of inputs makes pre-trained models unable to use the resulting out-of-vocabulary or information-destroying tokens.

Approach

A one-time post-hoc adaptation (no further pre-training) that keyed-encrypts the tokenizer vocabulary with Blake2, applies repeated glide-reflections to token embeddings (distance-preserving affine transforms), and shuffles vocabulary and embedding indices, followed by fine-tuning on encrypted training data.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Adapt and fine-tune BERT/RoBERTa on passkey-encrypted inputs instead of using the original modelBERT and RoBERTa · GLUE taskstask metrics: 55.24 vs 67.8 (CoLA); 56.34 vs 56.34 (WNLI)Mixed
Adapt and fine-tune mBERT on passkey-encrypted inputsmBERT · XNLI in-language and zero-shotaccuracy: 77.3 -> 78.7; 75.2 -> 76.8; 74.2 -> 73.9; 74.3 -> 75.5; 70.5 -> 71.1; 63.8 -> 65.3Helped
Adapt and fine-tune RoBERTa on encrypted inputsRoBERTa · CoNLL2003 NERF1: 95.66 -> 95.37Hurt
Use passkey nlp2023 instead of llm123 (ablation)Encrypted BERT and RoBERTa · GLUE taskstask metrics: CoLA: 55.3 vs 55.24 (EBERT); 59.05 vs 60.57 (ERoBERTa)Mixed
Increase glide-reflection iterations from 3 to 10 (ablation)Encrypted models · GLUE taskstask metrics: CoLA: 55.3 -> 59.05 (llm123 RoBERTa); MRPC: 89.76 -> 92.04 (llm123 RoBERTa)Mixed
Fine-tune the encrypted WNLI model with learning rate 1e-4 instead of the default 2e-5 (ablation)Encrypted WNLI models · GLUE WNLIaccuracyHelped
Fine-tune original and adapted models on 10 examples for one epoch before testing token-recovery rankingsBERT and RoBERTa · GLUE taskstoken-ranking match: 0.01%Hurt

Surprising findings

Limitations

Experimental setup

Datasets: GLUE CoNLL2003 XNLI

Benchmarks: GLUE CoNLL2003 XNLI

Metrics: Matthews correlation accuracy F1 Pearson correlation precision recall

Baselines: BERT original RoBERTa original mBERT original

Builds on: BERT RoBERTa Blake2

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