Search papers, labs, and topics across Lattice.
This paper introduces a novel backdoor attack targeting pipeline-parallel LLM post-training, where a malicious participant controlling an intermediate pipeline stage injects a trigger to misalign the model. The attack bypasses existing defenses designed for data poisoning and federated learning, demonstrating vulnerability even with limited adversary control. Experiments show a significant reduction in alignment (80% to 6%) upon trigger insertion, and the attack remains effective in 60% of cases even after safety alignment training.
Even a single compromised pipeline stage can inject backdoors that drastically misalign LLMs, bypassing standard safety alignment.
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.
