本文针对大型语言模型(LLM)在代码生成任务中可能复现不安全训练模式、生成漏洞代码的问题,首次系统性地研究了模型编辑(model editing)作为一种模型级安全硬化机制的有效性。与先前主要聚焦于推理时加固(如CoSec)的方法不同,模型编辑直接修改模型参数,无需额外辅助组件且不增加运行时开销。作者在多种LLM家族上评估了三种最先进的模型编辑方法,并与代表性推理时方法CoSec进行对比,考察了安全性、鲁棒性、泛化能力和功能正确性。实验发现,模型编辑在已见漏洞类型上比CoSec带来更大的安全收益,安全比例较原始模型提升15%-25%,且在提示扰动下保持稳定;然而,这些改进无法可靠迁移到未见漏洞,且可能降低功能正确性。为缓解这一权衡,作者提出了SafeEdit,一种结合功能调优和编辑感知正则化的编辑后优化方法。在八个目标LLM上,SafeEdit在T=0.1/0.4/0.8时相比UltraEdit将Pass@1分别提升了11.73/13.70/15.50个百分点,同时基本保持安全性。与CoSec相比,SafeEdit实现了7.54%-12.04%的相对安全比例提升。额外的CodeGuard+评估证实了联合安全且正确生成的改进。SafeEdit与CoSec具有互补性,组合使用可进一步提升安全性并保持强功能正确性。总体而言,本文为将模型编辑应用于安全代码生成提供了基于证据的指导。
💡 推荐理由: 为LLM安全代码生成提供了模型级硬化新思路,评测多种编辑方法的优劣,提出SafeEdit缓解安全-正确性权衡,对安全工程师选择防御策略有直接参考价值。
🎯 建议动作: 研究跟进