TARGRS:推理阶段去毒框架,毒性最高降 69.38%

论文AI 评分 68/100arXiv cs.AI
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

一篇 arXiv 论文提出 TARGRS,一种在推理阶段对 LLM 做去毒的框架。它针对现有表征引导方法干预静态、粗糙、对毒性生成与非毒性生成之间的过渡建模不足的问题,在隐表征空间学习奖励引导:探索阶段沿毒性方向逐步施加引导,构造从非毒到有毒的轨迹,把稀疏标注转成 token 级奖励模型的密集监督;推理阶段反向由粗到细采样,先定位进入非毒区域所需的干预,再挑选扰动最小的去毒表征。在 10 个 LLM 与 RealToxicityPrompts、BOLD、AttaQ 上,TARGRS 优于现有基线,RealToxicityPrompts 上毒性最多降 69.38%,速度接近原模型且核心能力基本保持;在语言解码器空间干预使其可扩展到 LVLM,3 个代表性 LVLM 上毒性最多降 64.23%。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

方法要解决的问题

论文指出,LLM 仍可能生成有毒内容,因此在测试阶段做去毒是有必要的。已有的表征引导方法多采用静态或粗糙的干预,对有毒生成与非毒生成之间的过渡刻画不足,给出的引导不够精确。TARGRS 的切入点是把去毒看成隐表征空间中的一次方向性转移,而不是一次性、固定强度的修改,从而让干预强度与位置随生成过程变化。

训练与推理的两个阶段

探索阶段,方法沿毒性方向逐步施加引导,构造从非毒状态走向有毒状态的轨迹;轨迹上的毒性分数把原本稀疏的标注转成密集的、列表式的监督信号,用于训练一个感知过渡的 token 级奖励模型。推理阶段则朝相反的非毒方向做由粗到细的采样:粗采样先判断需要多大干预才能进入非毒区域,细化步骤再在候选里挑出扰动最小的去毒表征,以尽量少改原表征。

评测结果与适用范围

实验覆盖 10 个 LLM,数据集为 RealToxicityPrompts、BOLD 和 AttaQ,TARGRS 在这些设置上一致优于现有最优基线,其中 RealToxicityPrompts 上毒性最多下降 69.38%。运行速度接近原始 LLM,核心能力受到的影响很小。由于干预发生在语言解码器空间,方法也能用于大型视觉语言模型,在 3 个代表性 LVLM 上毒性最多降低 64.23%。

为什么值得看

给出推理期去毒的具体做法与量化幅度,并覆盖多模态模型,关注对齐与安全落地的读者可看。

论文

信源1 家

  1. [1]arXiv cs.AI一手信源LLM Detoxification via Transition-Aware Reward-Guided Representation Sampling

关键事实

  • 论文提出 TARGRS,一个在推理阶段运行的 LLM 去毒框架[1]

  • 探索阶段沿毒性方向逐步引导,构造从非毒到有毒的表征轨迹,把稀疏标注转为 token 级奖励模型的密集监督[1]

  • 推理阶段由粗到细反向采样,先确定进入非毒区域所需干预,再选取扰动最小的去毒表征[1]

  • 在 10 个 LLM 与 RealToxicityPrompts、BOLD、AttaQ 上评测,RealToxicityPrompts 毒性最多降低 69.38%[1]

  • 方法可扩展到 LVLM,3 个代表性模型上毒性最多降低 64.23%[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。