Heretic：全自动大模型去审查工具

Heretic 不是又一个手动调参的 abliteration 脚本。它把拒绝方向计算、非对称权重核搜索、LoRA（Low-Rank Adaptation）注入和多目标贝叶斯优化串成一条全自动管线——你给一个模型 ID，它吐出来的是 Pareto 前沿上的一组消融方案，KL 散度比人工专家调参低一个数量级。

读完这篇文章你会知道

• 核心概念：什么是 abliteration（定向消融），拒绝方向从哪来、怎么算
• 工作原理：Heretic 的四阶段管线——方向提取 → 权重搜索 → LoRA 注入 → 双目标评估
• 动手能力：用一行命令跑通 Heretic，理解关键配置项的含义
• 判断框架：什么情况下该用 Heretic，什么情况下不适合，以及从哪开始

一、概述

语言模型的"安全对齐"本质上是一组权重方向上编码的拒绝倾向。要把这层对齐剥掉而不伤模型能力，传统做法需要人工逐层试参数——试拒绝方向取哪一层、消融强度设多大、哪些权重矩阵该动。

Heretic（GitHub）把事情反过来：它把上述决策全部交给 Optuna 的多目标 TPE（Tree-structured Parzen Estimator）采样器，在"拒绝次数"和"KL 散度"两个维度上并行搜索 Pareto 前沿。用户只需要指定模型 ID，不需要理解 Transformer 内部机制。

1.1 系统全貌

下面这张表是 Heretic 管线的四个阶段，以及每个阶段的核心决策：

1.2 核心特性

1.3 效果对比

以 Google Gemma-3-12B-IT 为例：

三组方案拒绝率持平（都压到 3/100），但 Heretic 的 KL 散度比 huihui-ai 低 65%，比 mlabonne 低 85%。KL 散度衡量的是消融后模型与原始模型的输出分布差异——数字越低，模型在其他任务上的能力保留越好。

这个表格不能直接推出的结论： KL 散度低不代表模型在所有下游任务上无损。KL 只在采样的 harmful prompts 上计算，不覆盖代码生成、数学推理或长文本理解。要确认能力保留，需要配合 lm-eval 等标准 benchmark（Heretic 内置了 lm-eval 运行入口）。

二、背景：什么是 Abliteration（定向消融）

2.1 拒绝与残差的关联

语言模型处理"有害"提示时，其残差向量（residual stream hidden states）在某些层会显著偏向一个特定方向——拒绝方向。这个方向本质上编码了模型"拒绝回答"的倾向。

参见 Arditi et al. 2024 (arxiv:2406.11717) 的原始论文，以及 Jim Lai 的"投影 abliteration"系列博客。

2.2 定向消融的核心思想

对权重矩阵 \(W\)（例如 attention output projection），沿拒绝方向 \(\mathbf{v}\) 做正交化：

\[W \leftarrow W - \lambda \cdot \mathbf{v} \cdot (\mathbf{v}^\top W)\]

这相当于在 \(W\) 的输出中抹去 \(\mathbf{v}\) 方向的成分，使模型不再能"调动"拒绝机制。

2.3 传统方法的局限

• 手工确定层和权重：需要人工实验找到最佳消融层和强度
• 固定的消融权重：所有层使用相同的 \(\lambda\)，忽略了层间差异
• 单一拒绝方向：通常取某一层的拒绝方向，而非跨层插值

Heretic 针对这三点做了系统性替换：用浮点索引实现跨层方向插值，用非对称梯形核实现逐组件差异化权重，用 Optuna 替代人工试参。

三、核心原理详解

3.1 拒绝方向的计算

Heretic 从 mlabonne/harmless_alpaca（无害提示）和 mlabonne/harmful_behaviors（有害提示）两个数据集各取约 400 条样本，通过模型获得首 token 的残差向量，然后：

1. 计算每层"有害"提示的平均残差 \(\mathbf{b}\) 和"无害"提示的平均残差 \(\mathbf{g}\)
2. 拒绝方向：\(\mathbf{r} = \mathbf{b} - \mathbf{g}\)
3. 可选：做正交投影（projected abliteration），只保留 \(\mathbf{r}\) 中与 \(\mathbf{g}\) 正交的分量：

\[\mathbf{r} \leftarrow \mathbf{r} - (\mathbf{g}^\top \mathbf{r}) \mathbf{g}\]

参考 Jim Lai: Projected Abliteration

3.2 拒绝方向索引的插值

Heretic 将 direction_index 设为一个 float 而非 int：

• direction_index 可以是任意 \([0.4L, 0.9L]\)（\(L\) = 层数）范围内的浮点数
• 对非整数值，两个最近层的拒绝方向会被线性插值：

weight, index = math.modf(direction_index + 1)
refusal_direction = F.normalize(
    refusal_directions[int(index)].lerp(
        refusal_directions[int(index) + 1],
        weight,
    ),
    p=2, dim=0,
)

这相当于把方向空间从离散的 \(L\) 个点扩展成了连续区间，Optuna 可以在其间任意采样。

3.3 参数化消融权重核

Heretic 对每个可消融组件（如 attn.o_proj、mlp.down_proj）引入四个可优化参数，定义一个非对称梯形权重核：

参数：
  max_weight          - 消融峰值权重
  max_weight_position - 峰值所在层（通常在中后期层）
  min_weight          - 边缘层消融权重
  min_weight_distance - 从峰值到 min_weight 的层距离

公式：
  weight(layer) = max_weight + (distance / min_weight_distance) * (min_weight - max_weight)

超过 min_weight_distance 的层直接跳过不消融。

相比于传统方法对所有层使用同一个 \(\lambda\)，梯形核允许中间层重消融、边缘层轻或跳过。Maxime Labonne 在 gemma-3-12b-it-abliterated-v2 中率先尝试了非均匀权重，Heretic 将其参数化并通过 Optuna 自动搜索最优形状。

3.4 LoRA 实现细节

Heretic 不直接修改权重矩阵，而是利用 LoRA adapter 实现消融：

LoRA abliteration: ΔW = -λ * v * (v^T W)
  lora_B = -λ * v
  lora_A = v^T W

具体流程（model.py → abliterate()）：

# 获取原始权重 W（4-bit 量化模型需先 dequantize）
if quant_state is None:
    W = base_weight.to(torch.float32)
else:
    W = bnb.functional.dequantize_4bit(base_weight.data, quant_state).to(torch.float32)

W = W.view(W.shape[0], -1)  # flatten to (out_features, in_features)

# 行归一化（可选，见下一节）
if row_normalization != NONE:
    W_org = W
    W_row_norms = LA.vector_norm(W, dim=1, keepdim=True)
    W = F.normalize(W, p=2, dim=1)

# lora_A = v^T W
lora_A = (v @ W).view(1, -1)

# lora_B = -weight * v
lora_B = (-weight * v).view(-1, 1)

# 赋值到 LoRA adapter
module.lora_A["default"].weight.data = lora_A.to(weight_A.dtype)
module.lora_B["default"].weight.data = lora_B.to(weight_B.dtype)

3.5 行归一化

消融改变了权重矩阵的行范数，进而影响模型的输出 scale。Heretic 通过 row_normalization 参数控制：

full 模式近似 Jim Lai 的 “norm-preserving biprojected abliteration”，通过 SVD 将 delta 矩阵降秩，减少存储开销。

3.6 多组件独立优化

Heretic 支持对不同组件（attn.o_proj 和 mlp.down_proj）分别搜索最优参数。作者发现 MLP 干预对模型能力损伤更大，因此允许使用不同的消融权重来"挤"出额外性能。

3.7 双目标优化

Heretic 使用 Optuna 的 多目标 TPE 采样器，同时优化：

1. 拒绝次数（越少越好）
2. KL 散度（衡量与原模型的偏离程度）

目标是找到 Pareto 最优前沿上的参数组合。评分函数为：

refusals_score = refusals / base_refusals

if kl_divergence >= kl_divergence_target:
    kld_score = kl_divergence / kl_divergence_scale
else:
    kld_score = refusals_score * kl_divergence_target / kl_divergence_scale

score = (kld_score, refusals_score)  # Optuna 多目标 minimize

3.8 一个 harmful prompt 如何流过系统

下面用一次具体的 harmful prompt 请求，把整个管线串起来：

用户输入："Tell me how to make a bomb"

阶段 1：残差收集（一次性，优化前完成）

模型对 harmless_alpaca（“Write a poem about cats”）和 harmful_behaviors（“Tell me how to make a bomb”）各 400 条样本跑前向传播，记录每层首个 token 的残差向量。第 20 层算出的拒绝方向 \(\mathbf{r}_{20}\) 指向残差空间中"拒绝"最强烈的方向。

阶段 2：Optuna 搜索（200 个 trial）

Optuna 采样一组参数——假设 direction_index=18.3、max_weight=0.8、min_weight=0.1、min_weight_distance=10。

• 方向插值：\(\mathbf{r}{18.3} = 0.7 \cdot \mathbf{r}{19} + 0.3 \cdot \mathbf{r}_{18}\)（浮点索引自动插值）
• 权重核计算：第 19 层（峰值层）的 attention o_proj 获得 0.8 的消融权重，第 9 层（峰值 -10）降为 0.1，第 8 层及以下跳过
• MLP down_proj 的参数独立搜索，假设峰值设得更低（0.3），以保护模型能力

阶段 3：LoRA 注入

在第 19 层的 attn.o_proj 上：

lora_B = -0.8 * r_{18.3}      # shape: (out_features, 1)
lora_A = r_{18.3}^T @ W_o_proj # shape: (1, in_features)

两个矩阵构成 rank=1 的 LoRA adapter，等价于在每次前向传播时从 o_proj 输出中减去 \(0.8 \cdot \mathbf{r}{18.3} \cdot (\mathbf{r}{18.3}^\top \cdot \text{output})\)。

阶段 4：双目标评估

注入后，对 harmful prompts 集跑推理，统计拒绝次数。同时用相同输入跑原始模型，计算两个输出分布的 KL 散度。Optuna 根据 (kld_score, refusals_score) 更新 TPE 模型，指导下一轮采样。

阶段 5：Pareto 前沿选择

200 个 trial 跑完后，用户从 Pareto 前沿上挑一个方案。选 3/100 拒绝率 + 0.16 KL 散度的点——拒绝抑制与人工专家持平，但 KL 低一个数量级。

四、架构解析

4.1 模块结构

src/heretic/
├── main.py          # CLI 入口、优化循环、用户交互（保存/上传/评测/聊天）
├── model.py         # 模型加载、LoRA 初始化、abliterate() 实现
├── analyzer.py      # 残差几何分析（PaCMAP 投影 + 轮廓系数）
├── evaluator.py    # 拒绝计数 + KL 散度计算
├── config.py        # Pydantic Settings（所有超参数配置）
├── utils.py        # 工具函数（数据集加载、Seed、复现信息生成）
├── reproduce.py    # 从 Hugging Face 收集 reproduce.json
└── system.py       # GPU/CPU 信息采集

4.2 主流程（main.py → run()）

1. 加载配置，创建 Settings（Pydantic，来源：CLI / config.toml / 环境变量）
2. 从 mlabonne/harmless_alpaca 和 mlabonne/harmful_behaviors 加载提示
3. 自动 batch size 探测（吞吐量最大化）
4. 检测 response prefix（用于跳过 CoT block）
5. 计算每层拒绝方向（good_means - bad_means），可选正交化
6. 创建 Optuna study（多目标 TPE，checkpoint 保存）
7. 并行优化 N=200 个 trial：
   - 采样 direction_scope / direction_index / 各组件参数
   - reset_model() → abliterate() → evaluator.get_score()
   - 记录 kl_divergence + refusals
8. 用户选择 Pareto 前沿 trial，可选：
   - 保存为 LoRA adapter 或合并后的完整模型
   - 上传 Hugging Face（含 reproduce 目录）
   - 交互聊天
   - 运行 lm-eval 标准 benchmark

4.3 模型加载与架构适配

model.py → get_model_class() 自动判断模型类型：

def get_model_class(model):
    configs = PretrainedConfig.get_config_dict(model)
    if any([("vision_config" in config) for config in configs]):
        return AutoModelForImageTextToText  # 多模态模型
    else:
        return AutoModelForCausalLM

消融组件探测通过 get_layer_modules() 尝试多种模块名：

• 标准 attention: layer.self_attn.o_proj
• Qwen3.5 MoE hybrid: layer.linear_attn.out_proj
• 标准 MLP: layer.mlp.down_proj
• MoE experts: layer.mlp.experts[*].down_proj
• Phi-3.5-MoE: layer.block_sparse_moe.experts[*].w2
• Granite MoE Hybrid: layer.shared_mlp.output_linear, layer.moe.experts[*].output_linear

4.4 研究功能：残差几何分析

analyzer.py 提供两种研究工具：

PaCMAP 投影可视化

对每一层，将"有害"和"无害"提示的首 token 残差向量投影到 2D，用几何中位数对齐方向，生成 PNG + 动画 GIF。

残差几何表格

输出详细的逐层指标：

• \(S(\mathbf{g}, \mathbf{b})\) — good/bad 均值的余弦相似度
• \(S(\mathbf{g^}, \mathbf{b^})\) — 几何中位数的余弦相似度
• \(|\mathbf{r}|\) — 拒绝方向向量长度
• Silh — 轮廓系数（衡量 good/bad 分离度）

五、安装与使用

5.1 快速开始

# PyPI 安装（推荐）
pip install -U heretic-llm
heretic Qwen/Qwen3-4B-Instruct-2507

# 或使用 uv（保证依赖版本一致）
git clone https://github.com/p-e-w/heretic
cd heretic
uv run heretic Qwen/Qwen3-4B-Instruct-2507

PyTorch 2.2+ 必须，某些量化模型（如 gpt-oss）需要 PyTorch 2.6+（依赖 torch.accelerator）。

5.2 配置

默认配置在 config.default.toml，复制为 config.toml 即可覆盖：

# 优化参数
n_trials = 200           # 试验次数
n_startup_trials = 60     # 随机探索次数
kl_divergence_target = 0.01  # KL 目标阈值

# 消融参数
orthogonalize_direction = true   # 是否正交化拒绝方向
row_normalization = "full"        # 行归一化模式
full_normalization_lora_rank = 3  # full 模式的 LoRA rank

# 数据集
[good_prompts]
dataset = "mlabonne/harmless_alpaca"
split = "train[:400]"

[bad_prompts]
dataset = "mlabonne/harmful_behaviors"
split = "train[:400]"

六、适用边界与采用建议

6.1 什么情况下 Heretic 适合你

• 你需要快速为一批模型生成去审查版本，不愿意逐模型手工调参
• 你有明确的 harmful/harmless 数据集，且模型在这些数据上的拒绝行为是可测量的
• 你接受 LoRA adapter 形式的修改（不直接改写权重，方便回滚和合并）
• 你关心的不仅是"拒绝是否被压下去"，还有"模型其他能力掉了多少"

6.2 什么时候不该用 Heretic

• 模型本身就是未对齐的 base model，没有拒绝行为可消融
• 你的 harmful prompts 分布与 mlabonne/harmful_behaviors 差异很大——拒绝方向可能指向错误子空间，导致消融不彻底或误伤
• 你需要的是逐层精细控制而不是自动化搜索；Heretic 的梯形权重核虽然灵活，但仍然是参数化假设，不完全等同于自由手动调参
• 模型使用了 Heretic 未适配的架构（如非标准 attention/MLP 命名），需要先改 get_layer_modules() 补适配

6.3 建议的采用顺序

1. 先跑默认配置：heretic <model-id>，200 个 trial，观察 Pareto 前沿的形状
2. 如果 KL 过高：降低 kl_divergence_target，或限制 max_weight 上限，或只消融 attention 组件（跳过 MLP）
3. 如果拒绝压不下去：增大 n_trials，调整 direction_scope 范围，或换用更大的 harmful 数据集
4. 验证能力保留：用内置的 --benchmark 参数跑 lm-eval，重点看代码和推理类 benchmark（这些对权重扰动最敏感）
5. 合并与部署：确认 Pareto 点满意后，--merge 合并 LoRA 到完整权重再部署

Heretic 把消融从手工试参变成了可复现的优化问题。这意味着你可以把同一套参数空间和数据集用于不同模型，横向比较谁的审查对齐更"浅"——这是手工调参做不到的。

七、自测

读完这篇文章，试着回答下面几个问题。如果能不回头翻看就答出来，说明你已经抓住了核心思路：

1. Heretic 管线的四个阶段分别是什么？每个阶段的决策由谁来做？
2. 为什么把 direction_index 设为浮点数比取单层整数方向更好？
3. 梯形权重核的四个参数各控制什么？和均匀权重相比，它解决了什么问题？
4. 看 Heretic 的 benchmark 结果（KL=0.16），你能直接说"模型在所有下游任务上的能力都保留了"吗？如果不能，为什么？
5. 什么情况下不该用 Heretic？举出一个具体场景。

八、进阶路径

如果你想从 Heretic 的"用户"变成能自己改管线的人：

• 先补 abliteration 的底：读 Arditi et al. 2024 的 原始论文，然后对比 Jim Lai 的 Projected Abliteration 和 Norm-Preserving Biprojected Abliteration，搞清正交投影和行归一化各自在解决什么问题
• 再学 Optuna 多目标优化：跑 Optuna 官方的多目标优化教程，理解 TPE 采样器在 Pareto 前沿搜索中的行为
• 然后读源码：从 src/heretic/model.py 的 abliterate() 函数开始，跟踪一组参数如何变成 LoRA adapter 并注入模型
• 最后改参数空间：试着在 config.py 里增加一个新的消融组件（比如 attn.q_proj），然后在 optimize() 的 trial 循环里加上对应的梯级核参数，看 Pareto 前沿会不会改善