arXiv:2605.10777cs.LG2026-05

用深度低秩残差结构锁定预训练模型,防篡改且不影响使用

Locking Pretrained Weights via Deep Low-Rank Residual Distillation

论文配图:Locking Pretrained Weights via Deep Low-Rank Residual Distillation
图 1 · 摘自论文原文
  • 用深低秩残差网络替换原MLP,增加反向传播内存开销
  • 使微调时优化路径复杂化,攻击者难以逆向破解
  • 适合需保护模型版权的研究者和企业使用

近年来开放权重语言模型质量显著提升,共享权重可促进跨平台部署与研究。但用户自由修改模型可能引发未经授权的滥用风险。传统防御易被攻击者通过观察权重和架构逆向破解。本文提出DLR-Lock,将预训练模型中的每个MLP替换为参数量相当的深低秩残差网络(DLR-Net),利用自动微分的推理-训练不对称性,在反向传播中引入线性增长的激活内存开销。通过模块化蒸馏高效训练该结构。实验表明,该方法在保持原始模型能力的同时,显著增加微调难度,使攻击者即使掌握完整防御策略也难以成功破解。在大规模语言模型上验证了其有效性。

原文摘要 · Abstract (English)

The quality of open-weight language models has dramatically improved in recent years. Sharing weights greatly facilitates model adoption by enabling their use across diverse hardware and software platforms. They also allow for more open research and testing, to the extent that users can use them as checkpoints, fine-tune them according to their needs, and potentially redistribute them. In some cases, however, concerns on modifying these weights towards unauthorized uses may outweigh the pros of giving users such a freedom. Defending against such adaptation is non-trivial: since an adaptive attacker can observe all weights and architectures by definition, they can reverse simple structural defenses, and use optimization to defeat the simplest locking mechanisms. In this work, we exploit the inference-training asymmetry of automatic differentiation as a novel defense axis. We propose DLR-Lock, a method where the purveyor of the model purposely replaces each pretrained MLP in their model with a deep low-rank residual network (DLR-Net) of comparable parameter count, forcing activation memory that grows linearly with depth during backpropagation. DLR-Nets are efficiently trained via module-wise distillation. We show that, beyond this memory overhead, DLR-Lock results in architectural mismatches that complicate the optimization landscape of standard fine-tuning, and a backward pass that incurs disproportionately more overhead than the forward pass. Our defense succeeds in withstanding adaptive attackers with full knowledge of the defense strategy while preserving the original model's capabilities. Experiments on LLM validate these claims.

模型安全微调防御低秩网络

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