首次提出针对流水线并行训练的后门攻击,仅控中间阶段即可破坏模型对齐
Backdoor Attacks on Decentralised Post-Training

- 攻击者仅控制流水线中间阶段,通过触发词植入后门
- 触发词使模型对齐率从80%降至6%,且在安全对齐训练后仍成功60%
- 适用于评估分布式大模型训练的安全性,尤其关注流水线并行场景
去中心化的大语言模型后训练利用数据与流水线并行技术分割数据和模型。然而,该过程可能受到恶意参与者的投毒和后门攻击。现有研究主要关注数据并行或联邦学习的鲁棒性,而对流水线并行的抗攻击能力研究仅限于投毒攻击。据我们所知,本文首次提出针对流水线并行的后门攻击,旨在使训练后的模型产生偏差。在本设置中,攻击者仅控制流水线的中间阶段,而非整个模型或数据集,因此现有攻击如数据投毒不再适用。实验结果表明,即使攻击者权限受限,仍可成功注入后门,导致模型在后训练过程中出现偏差,且不受学习领域或数据集影响。使用触发词后,模型对齐率从80%下降至6%。进一步测试发现,即便对最终模型进行安全对齐训练,攻击仍能在60%情况下成功。
原文摘要 · Abstract (English)
Decentralised post-training of large language models utilises data and pipeline parallelism techniques to split the data and the model. Unfortunately, decentralised post-training can be vulnerable to poisoning and backdoor attacks by one or more malicious participants. There have been several works on attacks and defenses against decentralised data parallelism or federated learning. However, existing works on the robustness of pipeline parallelism are limited to poisoning attacks. To the best of our knowledge, this paper presents the first backdoor attack on pipeline parallelism, designed to misalign the trained model. In our setup, the adversary controls an intermediate stage of the pipeline rather than the whole model or the dataset, making existing attacks, such as data poisoning, inapplicable. Our experimental results show that even such a limited adversary can inject the backdoor and cause misalignment of the model during post-training, independent of the learned domain or dataset. With our attack, the inclusion of the trigger word reduces the alignment percentage from $80\%$ to $6\%$. We further test the robustness of our attack by applying safety alignment training on the final model, and demonstrate that our backdoor attack still succeeds in $60\%$ of cases.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。