EasyEdit2——即插即用的LLM行为控制(Steering)框架:
1、支持广泛的测试时干预,包括安全性、情绪、个性、推理模式、事实性和语言特征。
2、关键模块:转向向量生成器和转向向量应用器。
论文发表于EMNLP 2025 System Demonstrations,Arxiv链接:https://arxiv.org/abs/2504.15133v3。
干预场景
如图2所示。
-
安全性:防止模型生成有害内容,如破解行为、减少偏见、拒绝不安全的查询、确保遵守法规,并减少隐私泄露的风险。
-
情感:控制模型输出的情感色彩,将情感从负面调整为正面,特别是在心理健康等领域中保持支持性的语气。
-
个性:探索不同个性如何影响模型行为,增强角色扮演的能力,并塑造模型所表现的内在价值观。
-
推理模式:调整推理的长度和风格,鼓励更加深思熟虑的思考,并根据不同领域的需求强制执行结构化的推理模式。
-
事实性:干预模型的事实知识,处理幻觉现象,支持知识编辑,并提高模型的自我验证能力。
-
语言特征:控制输出语言的使用,包括格式、句法结构、风格变化以及对单词层次的调整。
框架结构
如图3所示。
转向向量生成器(BaseVectorGenerator):可调用各种干预方法,在数据集上迭代生成转向向量。生成的向量可直接应用,或保存本地。
转向向量应用模块(BaseVectorApplier):可同时用多种方法将转向向量集成到目标模型,如基于提示、基于激活和基于解码等。
模型包装器:保留并集成多个转向向量以及用户提示,简化转向,增强模型行为控制。转向干预后,提供了两种操作模式:
-
返回修改后的模型以供立即、低代码使用;
-
根据配置设置、评估数据集生成评估文件。
对于评估,我们提供了Evaluators模块,该模块集成了基于规则、基于分类器和基于LLM的方法,以支持不同的场景。基于LLM的方法进一步实现了自适应和用户定义的场景评估。所有模块都利用Hparams模块进行灵活一致的配置。
转向向量库与合并:
-
导向矢量库:维护了一个针对各种场景优化的转向向量库,包括情绪控制、安全对齐等。用户能直接应用。
-
转向向量合并模块:可组合多个转向向量,结合多种合并策略,包括Linear、TIES和DARE-TIES。
两层超参管理:
-
一层:管理通用设置、向量生成、向量应用和评估参数的统一配置文件,整个框架使用此配置运行。
-
二层:转向方法的超参文件,包含转向向量生成和应用两个配置。继承自公共基类HyperParams。
数据集模块(DatasetLoader):统一评估数据集格式,根据配置加载和预处理数据。
评估器模块:基于配置在各种数据集上评估干预后模型的生成质量。评估方法分为基于规则、基于分类器和基于LLM。支持利用强LLM(如GPT-4)处理复杂的转向概念:用户指定要评估的转向概念,并使用预设模板对输入进行格式化。然后计算评估指标(概念相关性、教学相关性和流畅性得分等)。
实验
表1:6个评估场景样例。
表2:各干预、融合方法在安全、和情感任务上的性能对比。发现同时控制安全和情感的融合向量有更好的表现。
重要对比方法
单干预方法:CAA、STA、LM-Steer、Prompt-auto
干预融合方法:TIES、DARE-TIES
843

被折叠的 条评论
为什么被折叠?



