Efficient Preference Poisoning Attack on Offline RLHF
Abstract
Lay Summary
Many AI assistants are trained using human preferences, for example by showing the model two possible answers and marking which one people prefer. This process assumes that those preference labels are trustworthy. Our paper studies what can happen when an attacker quietly changes some of these labels in an existing training dataset, such as by marking worse answers as better ones. We focus on a popular offline training method called Direct Preference Optimization, where the model is trained from a pre-collected set of preference comparisons. We show that, for an important mathematical version of this method, each flipped label pushes the training process in a predictable direction. This lets us turn the attack into a geometric problem: choose a small set of labels whose combined effects move the model toward an attacker’s desired behavior. We design two algorithms for finding such label flips and prove when they can succeed, and also when any limited-budget attacker must fail. Experiments on synthetic data and a real human-preference dataset support the theory. These results identify a concrete vulnerability in preference-based AI training and provide insights for designing more reliable training algorithms.