LEMON 无需针对模型训练即可监控任何 LLM 的幻觉与越狱
等变 GNN 在生成时读取注意力图,在 16 个 LLM 上把 ROC-AUC 最多提高 14 分

来自以色列理工学院、特拉维夫大学和 Google Research 的研究者发布了 LEMON,一个读取语言模型内部注意力模式来检测不当行为 —— 包括幻觉和越狱驱动的拒绝绕过 —— 的安全监控框架,无需任何来自被监控模型的标注样本。论文出现在今天 arXiv 的 cs.AI 列表中,瞄准的是部署团队目前为新开放权重模型实现安全覆盖方式中的一个根本缺口。
核心进展是可移植性。多数 LLM 安全探针通过在特定模型的内部激活上训练分类器来工作,这意味着部署新模型的团队必须收集带事故标注的数据、训练新探针、验证它,然后才能获得覆盖。按当前的发布节奏,这条流水线可能让每一个进入生产的新模型暴露数周而无人监控。LEMON 消除了这一要求:在一个或几个源 LLM 上训练的探针可以零样本迁移到监控器从未见过的架构。
为什么标准探针跨不过架构边界
跨模型迁移的障碍是结构性的。标准激活探针作用于从模型残差流中取出的隐状态,而隐状态的维度和语义对齐在不同架构间各不相同。注意力图共享一种更一致的语义 —— 每个条目描述在给定层和头上一个 token 对另一个 token 的关注程度 —— 但朴素地把跨模型的注意力图池化仍然会失败,因为不同架构使用不同数量的头和层,没有把一个模型的第 3 个头等同于另一个模型的第 3 个头的规范映射。
LEMON 把这表述为一个代数对称性问题。一个 N token 序列跨 L 层和 H 个头产生的注意力图构成一个张量 A ∈ [0,1]^(N×N×L×H)。置换头的索引 —— 把第 3 个头重新标为第 7 个 —— 得到的是同一底层行为的等价表示。一个在此类置换下等变的监控器把所有有效的头排序同等对待,消除了对任何特定架构内部约定的依赖。论文通过一个带 W 等变消息传递的图神经网络实现这一点:token 成为图节点,注意力分数成为节点和边的属性,GNN 被约束为在头和层两个维度的置换下不变。[LINK: LEMON arXiv 2610.22723 equivariant GNN architecture]
监控器在 token 生成期间在线运行,对照近期注意力状态的一个窗口给每个新 token 打分。论文报告这种实时模式相对标准推理增加的计算开销不到 8% —— 对延迟预算敏感的生产部署来说是关键属性。[LINK: LEMON arXiv 2610.22723 inference overhead table]
在 16 个模型和六个基准上的一致迁移
团队在跨模型零样本设定下评估 LEMON:在源模型上训练的探针直接应用于监控器训练期间从未遇到过的留出 LLM。评估横跨来自六个架构家族的 16 个 LLM —— 参数从 1.35 亿到 130 亿,发布时间从 2023 年到 2025 年 —— 对照覆盖幻觉检测和越狱监控任务的六个基准数据集。LEMON 相对最强基线(作者称为「Prob」的基于概率的检测器)把 ROC-AUC 最多提高 14 分,并在 18 个模型-基准组合中的 16 个取得最佳结果。[LINK: LEMON arXiv 2610.22723 experimental results tables]
Lookback Lens(Chuang 等,EMNLP 2024)证明了注意力模式在同一模型家族内支持幻觉检测 —— 在 7B 模型上训练的探针可以泛化到其 13B 同胞 —— 但没有实现向无关架构的零样本迁移。[LINK: Lookback Lens EMNLP 2024] ACT-ViT 是以色列理工学院部分相同作者更早在 NeurIPS 2025 发表的论文,它用视觉 transformer 处理激活张量,迁移到新 LLM 时需要微调。[LINK: ACT-ViT NeurIPS 2025 Bar-Shalom Frasca Maron] LEMON 把零样本的覆盖范围扩展到架构上不相似的模型,同时把开销保持在 8% 以下。
白盒访问、规模限制与下一步
该方法需要对被监控模型注意力层的白盒访问,这意味着它不能应用于 GPT-4o 或 Claude 这类不暴露注意力图的闭源 API 系统。对运营开放权重模型的组织 —— 可移植性问题最尖锐的部署场景 —— 这一约束通常不构成阻碍,但它划定了当前的适用边界。
评估覆盖的模型最大到 130 亿参数。在注意力图上做等变消息传递能否在 700 亿以上的规模、或在注意力分布可能结构上不同的推理导向架构中继续有效,是一个开放的实证问题。该论文是等待同行评审的预印本,列表时注明代码即将发布。
实际含义很直接:为新的开放权重模型提供安全覆盖,不再需要等待带标注的事故数据。一个在有代表性的源架构集合上训练的等变探针,可以从新版本可用的那一刻起充当生产监控器,压缩当前部署与覆盖之间数周的空档 —— 对一个随每一次新的开放权重前沿发布而扩大的威胁面来说,这是一次有意义的收缩。