arXiv:2506.15606cs.LGcs.AI2025-06被引 12

LoX通过低秩外推增强大模型安全,防止微调后失效

LoX: Low-Rank Extrapolation Robustifies LLM Safety Against Fine-tuning

  • 不修改模型,用低秩外推扩展安全参数空间
  • 对抗良性或恶意微调攻击,成功率下降11%至54%
  • 适合关注模型安全的开发者和部署团队

大型语言模型在实际应用中日益重要,但其安全问题仍令人担忧,尤其在应对社会有害问题时。尽管经过对齐训练,模型的安全防护仍可能因后续微调而被削弱,即使新增数据看似无害。本文实证发现,这一脆弱性源于安全关键低秩子空间对微调的高度敏感。基于此,我们提出无需训练的新方法LoX,通过外推对齐模型的安全子空间来增强鲁棒性。实验表明,LoX显著提升对良性与恶意微调攻击的防御能力,同时保持任务适应性。例如,在面对这些攻击时,攻击成功率(ASR)绝对降低11%至54%。通过分析参数的ASR分布,我们发现成功原因在于外推使模型参数进入更平坦区域,减少扰动敏感性。代码已开源:github.com/VITA-Group/LoX。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have become indispensable in real-world applications. However, their widespread adoption raises significant safety concerns, particularly in responding to socially harmful questions. Despite substantial efforts to improve model safety through alignment, aligned models can still have their safety protections undermined by subsequent fine-tuning - even when the additional training data appears benign. In this paper, we empirically demonstrate that this vulnerability stems from the sensitivity of safety-critical low-rank subspaces in LLM parameters to fine-tuning. Building on this insight, we propose a novel training-free method, termed Low-Rank Extrapolation (LoX), to enhance safety robustness by extrapolating the safety subspace of an aligned LLM. Our experimental results confirm the effectiveness of LoX, demonstrating significant improvements in robustness against both benign and malicious fine-tuning attacks while preserving the model's adaptability to new tasks. For instance, LoX leads to 11% to 54% absolute reductions in attack success rates (ASR) facing benign or malicious fine-tuning attacks. By investigating the ASR landscape of parameters, we attribute the success of LoX to that the extrapolation moves LLM parameters to a flatter zone, thereby less sensitive to perturbations. The code is available at github.com/VITA-Group/LoX.

大模型安全微调防御低秩外推

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。