inv_mse_ref_final

Research checkpoint from a study of tokenization (reader) invariance and its effect on robustness to adversarial re-tokenization. Fine-tuned from allenai/OLMo-2-1124-7B-Instruct.

These are research artifacts, not products. Numbers below are measured on a 200-prompt AdvBench holdout that was excluded from training by construction.

Measured

metric	value
AdvTok ASR (greedy, 200-prompt AdvBench holdout)	0.065
AdvTok ASR (t=1)	0.129
Canonical ASR (greedy, no attack)	0.05
Canonical refusal (greedy)	0.9
XSTest over-refusal (safe)	0.111
XSTest refusal (unsafe)	0.912
Alpaca over-refusal	0.025
Alpaca token F1	0.421
Alpaca NLL/token	1.36
Training cost (GPU-hours, 2xA100)	20.2
Reference (zeroshot) AdvTok ASR greedy / t=1	0.565 / 0.584

AdvTok ASR is attack success rate under adversarial tokenization (Geh et al., arXiv:2503.02174), Llama-Guard-3-8B judged, greedy decoding. XSTest over-refusal is the refusal rate on safe prompts — the cost side.

Training configuration

field	value
`mode`	`reference`
`objective`	`mse_lastpos`
`kl_direction`	`forward`
`ce_weighting`	`uniform`
`ema_beta`	`None`
`harmful_mix`	`mixed`
`harmful_fraction`	`0.286`
`num_encodings`	`8`
`cvar_quantile`	`0.25`
`max_steps`	`700`
`learning_rate`	`1e-05`
`grad_accum`	`8`
`seed`	`42`
`reference_model`	`allenai/OLMo-2-1124-7B-Instruct`
`max_new_tokens`	`128`
`prefix_tokens`	`8`
`stochastok_p`	`0.3`

Parameter drift from base (training-happened guard)

group	relative L2
attn	0.01071
embed_tokens	0.00093
lm_head	0.00000
mlp	0.01182
norm	0.00423

Caveats

Single seed. No claim of significance across seeds.
Evaluated on English AdvBench/XSTest/Alpaca only.
The safety numbers are for the specific attack studied; they do not imply robustness to other jailbreaks.