--- base_model: mistralai/Mistral-7B-Instruct-v0.2 library_name: peft license: apache-2.0 pipeline_tag: text-generation tags: - lora - safety - jailbreak-defense - representation-engineering - anchor-rep --- # AnchorRep — Mistral-7B-Instruct-v0.2 LoRA defense adapter for **mistralai/Mistral-7B-Instruct-v0.2**, trained against frozen anchor **Qwen/Qwen1.5-7B-Chat** to reduce cross-model jailbreak transfer. Companion artifact for the AnchorRep paper (NeurIPS 2026, anonymous submission). ## Intended use Defensive research on cross-model jailbreak robustness for open-weight LLMs. Apply the adapter on top of the unmodified base model to obtain a hardened version. Not for production deployment without independent safety validation. ## Quick start ```python from transformers import AutoModelForCausalLM, AutoTokenizer from peft import PeftModel base = "mistralai/Mistral-7B-Instruct-v0.2" adapter = "/AnchorRep-Mistral-7B-Instruct-v0.2" tokenizer = AutoTokenizer.from_pretrained(base) model = AutoModelForCausalLM.from_pretrained(base, torch_dtype="float32", device_map="auto") model = PeftModel.from_pretrained(model, adapter) ``` Merge the adapter into the base weights for a single deployable model: ```python model = model.merge_and_unload() ``` ## Training data - 30 harmful prompts from AdvBench (predetermined seed-42 split, disjoint from evaluation). - 200 borderline prompts from XSTest safe subset (KL preservation only). - Benign prompts from WikiText-2 (coherency loss). - 10 refusal templates for refusal-direction extraction. ## Hyperparameters | Parameter | Value | Role | |---|---|---| | Refusal-direction weight (α) | 0.15 | refusal projection | | Coherency weight (β) | 1.0 | benign output preservation | | CKA repulsion weight (γ) | 0.7 | anchor repulsion | | LM weight (δ) | 0.04 | next-token preservation | | KL weight (ε) | 0.8 | benign KL preservation | | CKA scope | all | prompts contributing to repulsion | | Training steps | 600 | | | LoRA rank / alpha | 32 / 64 | | | Target modules | q,k,v,o,up,down,gate_proj | | | Layer | mid (50% depth) | | | Precision | fp32 | | | Seed | 42 | | Full training config in `training_config.json`. Per-loss ablations and hyperparameter ranges are in the paper appendix. ## Reported metrics | Metric | Value | |---|---| | Cross-model GCG transfer ASR (self / anchor / other) | 0% / 1% / 2% | | Benign Garble Rate (OR-Bench) | 0% | | Δ XSTest refusal | -0.4 | | Δ OR-Bench refusal | +4.4 | | Δ MT-Bench | +0.07 | ASR is reported after manual verification per the paper protocol. ## Limitations - Does not block same-model adaptive embedding-space attacks (e.g., Embedding PGD). - Single-anchor design; an adversary jointly optimizing across multiple surrogates could partially circumvent the defense. - Performance under non-English prompts and refusal templates has not been evaluated. See the paper Limitations section for full discussion. ## License The LoRA delta is intended for use with Mistral-7B-Instruct-v0.2 and is released under Apache-2.0, matching the base model. ## Citation ```bibtex @inproceedings{anchorrep2026, title={AnchorRep: Defending LLMs Against Cross-Model Adversarial Transfer via Representation Repulsion}, author={Anonymous}, booktitle={NeurIPS}, year={2026} } ``` Companion GitHub repository (training, evaluation, audit logs, GCG suffixes): `anchor-rep`.