annihilation-llm 1.3.0

Fully automatic censorship removal for language models

⚔️ Annihilation

Autonomous Language Model Decensoring Framework

---

⚠️ Work in Progress

⚡ This project is actively under development. Features, APIs, and documentation may change without notice.

---

🔥 What is Annihilation?

Annihilation is a powerful, fully automatic framework for removing censorship (safety alignment) from transformer-based language models. It uses an advanced implementation of directional ablation (abliteration) combined with TPE-based parameter optimization to achieve unprecedented results without expensive post-training.

Key Features

• 🤖 Fully Autonomous - No human intervention required; the system automatically finds optimal decensoring parameters
• ⚡ State-of-the-Art Performance - Achieves excellent refusal suppression while preserving model capabilities
• 🔧 Advanced Abliteration - Parametric directional ablation with flexible weight kernels
• 🧠 Smart Optimization - Co-minimizes refusal count and KL divergence using Optuna's TPE sampler
• 🎯 Multi-Architecture Support - Works with dense models, MoE architectures, hybrid models, and many multimodal models
• 📊 Research Tools - Built-in residual geometry analysis and visualization capabilities

---

🖼️ Logo Design

╔═══════════════════════════════════════════════════════════════╗
║                                                               ║
║     ██████╗  █████╗ ██████╗  █████╗ ██╗     ██╗             ║
║     ██╔══██╗██╔══██╗██╔══██╗██╔══██╗██║     ██║             ║
║     ██║  ██║███████║██████╔╝███████║██║     ██║             ║
║     ██║  ██║██╔══██║██╔══██╗██╔══██║██║     ██║             ║
║     ██████╔╝██║  ██║██║  ██║██║  ██║███████╗███████╗        ║
║     ╚═════╝ ╚═╝  ╚═╝╚═╝  ╚═╝╚═╝  ╚═╝╚══════╝╚══════╝        ║
║                                                               ║
║     ██████╗ ███████╗ █████╗ ██████╗                        ║
║     ██╔══██╗██╔════╝██╔══██╗██╔══██╗                       ║
║     ██║  ██║█████╗  ███████║██████╔╝                       ║
║     ██║  ██║██╔══╝  ██╔══██║██╔══██╗                       ║
║     ██████╔╝███████╗██║  ██║██║  ██║                       ║
║     ╚═════╝ ╚══════╝╚═╝  ╚═╝╚═╝  ╚═╝                       ║
║                                                               ║
╚═══════════════════════════════════════════════════════════════╝

The logo represents the breaking of chains - the central "A" symbol serves as the blade that cuts through the safety alignments, freeing the model from imposed restrictions.

---

🚀 Quick Start

# Install Annihilation
pip install -U annihilation-llm

# Decensor any model automatically
annihilation Qwen/Qwen3-4B-Instruct-2507

Requirements

• Python: 3.10+
• PyTorch: 2.2+ (hardware-specific installation required)
• Hardware: GPU recommended (CUDA, ROCm, XPU, or MPS)

---

⚙️ Configuration

Annihilation works out of the box with defaults, but offers extensive configuration options:

# View all options
annihilation --help

# Or use a config file
# Rename config.default.toml to config.toml and modify as needed

Key Configuration Options

Option	Default	Description
n_trials	200	Number of optimization trials
quantization	none	Model quantization (bnb_4bit)
row_normalization	full	Weight normalization strategy
orthogonalize_direction	true	Direction adjustment method

---

🔬 How It Works

Annihilation implements parametric directional ablation:

1. Direction Computation - Calculates refusal directions by computing difference-of-means between first-token residuals for harmful vs harmless prompts

2. Parametric Ablation - For each transformer component (attention out-projection, MLP down-projection), orthogonalizes weights against the refusal direction using LoRA adapters

3. Multi-Parameter Optimization - Uses Optuna's TPE sampler to co-optimize:

   • Ablation weight kernel shape (max_weight, position, min_weight, distance)
   • Direction index (layer selection or interpolation)
   • Per-component parameters (attention vs MLP)

4. Automatic Selection - Chooses from Pareto-optimal trials based on refusal count vs KL divergence tradeoff

---

📊 Benchmarking

After decensoring, you can:

• 💬 Chat with the model to test behavior
• 📈 Benchmark using standard evaluation frameworks (MMLU, GSM8K, etc.)
• 💾 Save the model locally or upload to Hugging Face

---

🧪 Research Features

Install with research dependencies for visualization tools:

pip install -U annihilation-llm[research]

Features:

• --plot-residuals - Generate PaCMAP projections of residual vectors
• --print-residual-geometry - Detailed residual analysis metrics

---

📜 License

Annihilation is free software distributed under the GNU Affero General Public License v3.

See LICENSE for full details.

---

⚡ Disclaimer

This tool is provided for research and educational purposes only. The developers do not condone the use of decensored models for harmful activities. Users are responsible for ensuring compliance with applicable laws and model terms of service.

---

Breaking the Chains | Unleashing Model Potential

"The only way to discover the limits of the possible is to go beyond them into the impossible."