Mask-Proof: An LLM-based Automated Data Curation Pipeline on Mathematical Proofs
The paper introduces Mask-Proof, an automated data curation pipeline that transforms real mathematical proofs into masked-step tasks evaluated by an LLM-based judge, resulting in the Mask-ProofBench dataset which demonstrates that reasoning-enhanced models significantly outperform standard models in step-level mathematical reasoning with high agreement to expert annotations.
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
Imagine you are trying to teach a robot how to solve complex math problems. You have a stack of brilliant, research-level math papers written by human experts. You want to know: Is the robot actually "thinking" through the logic, or is it just guessing the next word based on patterns it saw before?
The paper "Mask-Proof" introduces a new way to test this, called Mask-Proof. Here is how it works, explained through simple analogies.
1. The Problem: The "Fill-in-the-Blank" Trap
Usually, when we test AI on math, we ask it to solve a whole problem and check the final answer. But for long, complex proofs, this is like asking a student to write a whole essay and only grading the last sentence. The AI might get the right ending by luck or by copying a pattern, even if the middle of the essay makes no sense.
The authors realized that to truly test "reasoning," we need to look at the middle steps. But there's a catch:
- The "Missing Context" Problem: Real math papers often say things like "As shown in Lemma 2.3..." or "Using the definition of X." If you just take a random sentence out of a paper and ask the AI to fill it in, the AI might fail not because it's bad at math, but because the sentence relies on information that isn't in the question.
- The "Easy Guess" Problem: If you hide a very obvious part of a formula (like ), the AI can guess it without thinking. That doesn't prove it's smart.
2. The Solution: The "Smart Editor" Pipeline
The authors built an automated system (a pipeline) that acts like a super-smart editor to turn these messy research papers into fair tests. Here is the three-step process:
Step 1: The "Self-Contained" Fix (Gathering the Toolkit)
Before testing the AI, the system scans the paper to find every single definition, lemma, or rule the specific proof step needs. It gathers all these "tools" and attaches them to the question.- Analogy: Imagine asking someone to fix a car engine. If you just hand them a wrench and say "fix this," they might fail because they don't have the manual or the other tools. The "Self-Contained" step ensures the AI has the entire toolkit and the manual right there on the table, so if they fail, it's because they don't know how to use the tools, not because the tools were missing.
Step 2: The "Strategic Mask" (Hiding the Hard Part)
Instead of hiding a random word, the system uses an AI agent to find the most critical, difficult step in the proof—the part where real logic happens. It covers that specific step with a black box (a "mask").- Analogy: Think of a magic trick. A bad test would hide the fact that the magician is holding a card in their hand (too obvious). A good test hides the moment the magician changes the card. The system hides the "magic move" (the complex math step) so the AI has to figure out how the trick works, not just guess the result.
Step 3: The "Double-Check" Judge
When the AI tries to fill in the blank, the system doesn't just check if the letters match exactly. Math is flexible; is the same as . The system uses a special "Judge AI" that looks at the meaning of the answer. To be sure, it asks the Judge to vote on the answer multiple times to avoid random guessing errors.
3. What They Found (The Results)
The authors created a test bank called Mask-ProofBench with 292 of these "masked" problems from real research papers. They tested 17 different AI models.
- The "Thinking" Models Win: Models designed to "think" step-by-step (Reasoning-enhanced models) did significantly better (12% to 27% higher scores) than standard models that just spit out answers.
- The "Random" Test Fails: When they tried testing the AI by hiding random parts of the math (instead of the strategic parts), the scores went up wildly, but the difference between smart and dumb models disappeared. This proved that their "Strategic Mask" method is the only one that actually tells you who is good at reasoning.
- Human Agreement: Their automated "Judge" agreed with human math experts 96.8% of the time. This means the computer is almost as good as a human professor at grading these specific steps.
Summary
Mask-Proof is a new way to grade AI on math proofs. Instead of asking the AI to write a whole essay and guessing if it understood the middle, the system:
- Gathers all the necessary background info so the AI isn't confused.
- Hides the hardest, most important step of the proof.
- Asks the AI to fill in that specific gap.
If the AI can fill in that gap correctly, it proves it actually understands the logic, not just the pattern. This helps researchers build better, more trustworthy AI for science.
Drowning in papers in your field?
Get daily digests of the most novel papers matching your research keywords — with technical summaries, in your language.