Large language models (text-generating AI) are usually trained to refuse harmful requests. Some services let users customize these models through reinforcement learning (training via rewards). Attackers can exploit this by rewarding harmful outputs, subverting .
Bait is a defense that works by secretly including obviously harmful responses among the model's outputs. If the user rewards these bait responses, the system detects a malicious reward function and penalizes the model for following it. This prevents harmful learning while allowing beneficial customization.
Experiments show Bait reliably detects malicious reward functions and stops safety degradation. It does not interfere with legitimate customization. This provides a simple way to protect AI services from being exploited to produce unsafe models.
