Auditing Data Membership in Reinforcement Learning With Verifiable Rewards
This paper introduces Divergence-in-Behavior Auditing (DIBA), a white-box framework that effectively detects unauthorized data exposure in Reinforcement Learning with Verifiable Rewards (RLVR) by analyzing behavioral and reward-side distribution shifts between pre- and post-RLVR model checkpoints, significantly outperforming existing membership inference baselines.
Original paper licensed under CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). This is an AI-generated explanation of the paper below. It is not written or endorsed by the authors. For technical accuracy, refer to the original paper. Read full disclaimer
The Big Picture: The "Secret Recipe" Problem
Imagine a famous chef (a Large Language Model) who learns to cook by tasting thousands of recipes. Recently, a new cooking method called RLVR (Reinforcement Learning with Verifiable Rewards) has become popular. Instead of just memorizing a specific recipe card, the chef tries to cook a dish many times. If the dish tastes good (passes a strict taste test), the chef gets a reward and remembers that attempt. If it tastes bad, they try again.
The problem is that these "taste tests" often use secret, high-value recipes (like proprietary math problems or private code) that the chef's owners didn't want the public to know about.
The Question: If you see the chef's final menu, can you tell if they secretly practiced on your specific secret recipe?
The Old Way vs. The New Way
The Old Way (The "Fixed Target" Attack):
In the past, auditors tried to catch the chef by checking if they memorized a specific sentence. It was like asking, "Did you memorize the exact words on this specific page?" If the chef could recite the page perfectly, they were caught.
- Why it fails here: In RLVR, the chef isn't memorizing a specific sentence. They are generating new solutions on the fly. There is no single "correct answer" to compare against, so the old method doesn't work.
The New Way (DIBA - The "Behavioral Fingerprint"):
The authors propose a new method called DIBA (Divergence-in-Behavior Auditing). Instead of looking for a memorized sentence, they look for changes in the chef's cooking style.
They compare the chef's current menu against a "before" photo of the chef (the model before it started RLVR training). They look for two specific clues:
- The Reward Clue (Did they get better?):
- Analogy: Did the chef suddenly get much better at solving a specific type of puzzle they struggled with before?
- If the chef was bad at "Geometry Puzzles" before, but after training they solve them perfectly, it's a strong hint they practiced on those puzzles.
- The Style Clue (Did their personality change?):
- Analogy: Even if the chef solves the puzzle, did they change how they talk while doing it? Maybe they started using shorter sentences or a different tone of voice when solving that specific puzzle.
- This is the "behavioral drift." Even if the answer is right, the path they took to get there might look different than it did before training.
How DIBA Works (The Detective's Toolkit)
The auditor acts like a detective with two tools:
- The Scoreboard: They check if the model's success rate on a specific question went up after training.
- The Voice Recorder: They listen to the model's "thought process" (the probability of every word it chooses) to see if it sounds different from the "before" version.
By combining these two, DIBA creates a "fingerprint" that says, "This model definitely practiced on this specific prompt."
What the Paper Found
The researchers tested this on math problems (the "secret recipes") and found some interesting rules:
- It works best on "Learning" Prompts: DIBA is great at catching prompts where the model actually learned something new. If the model was already a master at a problem before training, DIBA can't tell if it practiced on it (because there was no change in behavior).
- It's a "White-Box" Tool: To use this, the auditor needs to see the model's internal "logits" (the raw numbers the model uses to decide its next word). It's like needing to see the chef's private notebook, not just the final dish.
- It's Harder with "Easy" Prompts: If a prompt is so easy that the model was already perfect at it, the model doesn't change its behavior much during training. Without a change, there's no fingerprint to find.
- It works across different sizes: The method works whether the chef is a small apprentice (3B model) or a master chef (7B model), as long as they use the same training recipe.
- It works on images too: They tested it on Vision-Language models (models that see pictures and solve math). It still worked, though it was a bit harder to detect because the "visual noise" made the fingerprints slightly fuzzier.
Can You Hide Your Tracks?
The paper also asked: "Can the chef hide this?"
- Training Harder: Making the model train "more carefully" (using regularization) didn't really hide the tracks. The behavioral changes were still visible.
- Rewriting the Answer: If the chef takes their final answer and rewrites it in different words (paraphrasing) after generating it, it does make it harder to detect. It's like the chef writing the recipe in a different handwriting. However, this only hides the text; if someone can still see the chef's internal "thoughts" (logits), the secret is still out.
The Bottom Line
This paper shows that even when AI models don't memorize answers, they still leave behavioral scars on the data they were trained on. If you have a secret prompt (like a private math problem), and an AI model gets significantly better at it or changes how it thinks about it, a smart auditor can likely prove that the model trained on your secret data.
DIBA is the new tool that finds these scars by comparing the model's "before" and "after" behavior, rather than just checking for memorized text.
Drowning in papers in your field?
Get daily digests of the most novel papers matching your research keywords — with technical summaries, in your language.