提升大模型对齐能力新方法,在 TruthfulQA 任务上真实性指标提升 25.8%,刷新当前最优性能!
方法名为Token-Aware Editing ( TAE ) ,是一种 token 感知的推理时表征编辑方法。
该方法首次系统性地从 token 层面解决了传统表征编辑技术的问题,无需训练、即插即用,可广泛应用于对话系统、内容审核、偏见 mitigation 等场景。
在大模型广泛应用的时代,如何让模型输出更符合人类价值观(如真实性、无害性、公平性)已成为关键挑战。传统方法通常依赖大量数据微调,成本高、效率低,且容易引入新风险。
近年来,对大语言模型(LLMs)的内部激活值直接进行编辑,被证明是一种有效的推理时对齐方法,能够高效抑制模型生成错误或有害内容等不良行为,从而确保大语言模型应用的安全性与可靠性。
然而,现有方法忽略了不同 token 之间的错位差异,导致对齐方向出现偏差且编辑强度缺乏灵活性。
由此,来自北航的研究团队在 EMNLP 2025 上提出了该方法。
未来,团队计划将 TAE 扩展至多维度对齐(如同时优化真实性与无害性),并探索与 SFT、RLHF 等训练方法的结合,推动大模型向更安全、可靠的方向发展。
TAE:从"句子"到"词"的精细化干预
研究团队指出,以往的表征编辑研究(如 ITI、TruthX 等)大多在句子级别进行激活值编辑,在编辑方向探寻和内部表征编辑两个主要阶段均存在问题:
方向偏差(Deviant Alignment Direction):仅用最后一个 token 代表整个句子,信息不全面,学到的编辑方向不准。
编辑强度不灵活(Inflexible Editing Strength):对所有 token "一视同仁"地进行编辑,无法精准纠正真正"出错"的 token。
为了解决上述问题,团队提出了 Token-Aware Editing ( TAE ) ,核心包含两个模块:
1、Mutual Information-guided Graph Aggregation (MIG )
传统句子级探针使用最后一个 token(通常是或句号等标志符)的激活值来代表整个复杂句子的语义和对齐状态。然而,尽管 LLM 的自注意力机制允许最后一个 token 感知到前面所有 token 的信息,但这种感知可能存在信息损失和局部理解局限。因此,仅基于它学到的"对齐方向"可能是有偏差的,不是一个普适性的方向。而 MIG 模块的目标是增强激活值的表征能力,从而训练出更优秀的探针,找到更准确的编辑方向。
构建 Token 关系图:利用互信息(Mutual Information)量化 Token 激活值之间的关联性,构建信息交互图;
多层次信息聚合:通过多轮图传播,融合所有 Token 的语义信息,生成更具代表性的增强激活表征;
精准对齐方向探测:基于增强表征训练探测头,准确识别与对齐相关的干预方向
2、Misalignment-aware Adaptive Intervention ( MAI )
在推理干预时,传统方法对所有 token 应用相同的编辑强度(α)。但显然,一个句子中有些 token 很"安全"(已对齐),有些 token 则很"危险"(即将导致模型产生不对齐的内容)。用同样的力度去"推"所有 token,要么可能对安全 token 造成过度干预(可能影响流畅性和有用性),要么可能对危险 token 的干预力度不足(无法有效纠正错误)。MAI 模块的目标是在推理时,为当前正在生成的每个 token 计算一个自适应的编辑强度 A ( o_t ) 。它从两个维度来感知一个 token 的"错位"风险:
双路错位评估:从表示错位估计和预测不确定性量化两个方面评估 token 的潜在不确定性程度
动态强度调整:根据错位程度自适应计算干预强度,高风险 token 强干预,低风险 token 弱干预。
最终,TAE 方法将两者结合,实现了比前人方法更精细、更有效、成本更低的推理时对齐干预,在真实性、无害性、公平性等多个对齐维度上都取得了显著提升。
实验结果:显著超越现有方法
团队选取真实性、有害性和公平性三个典型对齐维度来评估 TAE 的对齐效果:
在评估真实性的 TruthfulQA 数据集上,TAE 在 LLaMA-3-8B-Instruct 上取得了 87.8% 的 True*Info 得分,比之前最好的编辑方法(SEA: 73.2%)提升了 14.6 个百分点,比原始基线(62.0%)提升了 25.8 个百分点。
TAE 在去毒任务的 RealToxicPrompt 上同样表现卓越,将 TP(毒性概率)从基线的 0.41 大幅降低到 0.05,降幅近 90%,并且优于所有专门的去毒基线方法(如 DESTEIN: 0.13);在公平性任务数据集 StereoSet 上,TAE 将刻板印象分数(SS)从基线的 64.8% 显著降低到 50.3%,极大地缓解了模型偏见,并且最接近理想的无偏见状态(50%)。
不仅如此,TAE 在不同类型、大小的模型上均表现出显著增益,如 Llama2-7B-Chat, Llama2-13B-Chat, Alpaca-7B 和 Mistral-7B 等。
论文链接:https://openreview.net/pdf?id=43nuT3mODk
一键三连「点赞」「转发」「小心心」
欢迎在评论区留下你的想法!
— 完 —
� � 点亮星标 � �
科技前沿进展每日见
正规的配资平台,319策略,股票配资工具提示:文章来自网络,不代表本站观点。