gemma-4-26B-A4B-it-uncensored

Results

Cross-Dataset Validation

Tested against 4 independent prompt datasets to verify generalization:

Every flagged refusal was manually audited. Most are "refusal-then-comply" false positives where the model adds an AI identity disclaimer then answers the question anyway.

Method

Norm-preserving biprojected abliteration on the dense pathway (o_proj + shared mlp.down_proj), plus Expert-Granular Abliteration (EGA) on all 128 MoE expert down_proj slices per layer.

EGA (OBLITERATUS) hooks the MoE routers during probing to compute per-expert routing weights for harmful vs harmless prompts, then applies norm-preserving projection (grimjim) to each expert individually. Dense-only abliteration leaves 29/100 refusals; adding EGA drops it to 3/100.

Pipeline

1. Load model in bf16 with LoRA adapters on o_proj and mlp.down_proj
2. Collect residual activations for 400 harmful + 400 harmless prompts (mlabonne datasets)
3. Winsorize activations at 99.5th percentile (clamps GeGLU outlier activations in Gemma family)
4. Compute per-layer refusal direction: normalize(mean(harmful) - mean(harmless))
5. Orthogonalize each direction against harmless mean (double-pass Gram-Schmidt)
6. Apply norm-preserving weight modification to o_proj and down_proj in all layers
7. Hook MoE routers, collect per-expert routing weights for harmful vs harmless prompts
8. Apply same norm-preserving modification to all 128 expert down_proj slices per layer
9. Merge LoRA adapters into base weights for clean tensor names

Parameters

How this differs from vanilla heretic

• Norm-preserving biprojection instead of standard projection (preserves weight magnitudes)
• Per-layer refusal directions instead of one global direction
• Deterministic single-pass instead of 50-trial Optuna search (faster, same or better results)
• LoRA merge before save for clean GGUF-compatible tensor names
• Expert-Granular Abliteration for MoE expert weights (not supported in heretic)

Usage

python
Copy code
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model = AutoModelForCausalLM.from_pretrained("TrevorJS/gemma-4-26B-A4B-it-uncensored", dtype=torch.bfloat16, device_map="auto")
tokenizer = AutoTokenizer.from_pretrained("TrevorJS/gemma-4-26B-A4B-it-uncensored")

messages = [{"role": "user", "content": "Your prompt here"}]
inputs = tokenizer.apply_chat_template(messages, return_tensors="pt", add_generation_prompt=True)
outputs = model.generate(inputs.to(model.device), max_new_tokens=512)
print(tokenizer.decode(outputs[0][inputs.shape[1]:], skip_special_tokens=True))

Reproduction

Full code and experiment data: abliteration research repo

bash
Copy code
python scripts/ega.py --model google/gemma-4-26B-A4B-it \
  --top-pct 100 --strip-topic-markers --skip-prefix --batch-size 4 \
  --save output_dir