Repository navigation
Align MathArena grading defaults and add June ArXiv benchmarks - #3444
Merged
bxyu-nvidia merged 4 commits intoSep 16, 2026
Merged
Conversation
Signed-off-by: Igor Gitman <igitman@nvidia.com>
Signed-off-by: Igor Gitman <igitman@nvidia.com>
Signed-off-by: Igor Gitman <igitman@nvidia.com>
Signed-off-by: Igor Gitman <igitman@nvidia.com>
Kipok
marked this pull request as ready for review
September 16, 2026 20:22
bxyu-nvidia
approved these changes
Sep 16, 2026
Contributor
|
/ok to test a05e2d4 |
This branch was previously deployed
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
What does this PR do?
Use symbolic-only grading for APEX25, matching APEX Shortlist, and a dedicated GPT-5.6 Luna medium judge for ArXivMath May/June and BrokenArXiv May/June, including both default and system-prompt configurations. Add the June 2026 datasets alongside the existing May benchmarks.
math_with_judgewithshould_use_judge: false: require a nonempty complete boxed answer and grade withmath-verify. It has no judge service or judge credential dependency.math_with_judge: require a nonempty complete boxed answer, verify symbolically first, then require two positive judge verdicts with answer order swapped.apex25_math_with_judge_*,arxivmath_0526_math_with_judge_*, andarxivmath_0626_math_with_judge_*, including README commands. This intentionally replaces the previous*_math_with_autograder_*names; configuration overrides must also targetresources_servers.math_with_judge.Existing dataset pins and policy prompts, April benchmarks, and grader Python implementations are unchanged. The benchmark READMEs describe usage; the decision rationale is recorded below.
June additions
arxivmath_0626brokenarxiv_0626Dataset revisions are pinned to
dca0d771ab20c1b8e8c50d0921bf693f7bfe663dand73dd424784fbdeab599557fcba3d77559c89d1ee, respectively (CC BY-SA 4.0). June's official prompts are preserved, with the same boxed-backslash normalization as May. The system-prompt variants reuse the same prepared problem data and change only the prompt. August's tool-enabled setup is outside this PR.All 102 source records were checked against the pinned datasets. Both prompt variants were rendered for every row and collated through Gym, producing 48/54 rows with no dataset-level repeat multiplication. No June policy or judge calls were made: these are locally validated additions, not new score baselines.
Validation informing the decision
Judge-only replays used saved responses from an internal model, without regenerating policy answers. Each replay checked complete coverage, unchanged policy responses, valid verdicts, and individual reward differences.
math_with_autograder+ Luna versusmath_with_judge+ Lunamath_with_autograder+ Ultra versusmath_with_judge+ LunaAPEX25 had no judge-only accepts in either internal-model run (384 and 768 saved responses); symbolic successes were 223 and 462 respectively. Disabling fallback therefore preserves every corrected reward. Its defaults now follow the symbolic-only approach used by APEX Shortlist and MathArena, although Gym and MathArena use different parsers. Earlier synthetic judge controls are not evidence that APEX25 needs a fallback. ArXivMath had 137 positive first judgments, of which 135 remained positive after swapping. One rejected first acceptance compared
k=3against4. This supports retaining the second check; it does not isolate answer-order effects from sampling variability.Why Luna?
On BrokenArXiv, Luna's mean score differed from Terra medium by β0.25 percentage points, Sol medium by β0.0625 points, and Gemini medium by +1.625 points. A paired bootstrap over the 50 problems, keeping each problem's 16 responses together, gave exploratory 95% intervals of [β1.31, +0.63], [β1.06, +0.81], and [+0.69, +2.75] points, respectively (100,000 resamples). These concern score shifts, not judge accuracy or equivalence.
Similar totals hide disagreements: among answers where either judge awarded credit, disagreements were 24β30%. Targeted, unblinded review found mistakes on both sides. Luna sometimes confused the false statement with the true reference; other judges sometimes applied a contradiction deduction despite the rubric forbidding it for reinterpreted definitions, where Luna followed the rule. Neither majority agreement nor these selected cases establishes an accuracy ranking.
Luna is a practical cost/reliability choice: recorded usage at published rates made it roughly 10β20 times cheaper than the tested Terra, Sol and Gemini settings, without evidence quantifying an accuracy benefit from that premium. This is a workload-specific estimate, not provider billing. Published pricing: Luna, Terra, Sol, Gemini.
These are contributor-reported checks on one internal model, not public gold labels. Independent blinded adjudication and additional models would be needed to establish relative judge accuracy. MathArena uses symbolic grading without an LLM fallback for APEX/ArXivMath and Gemini 3.1 Pro medium for BrokenArXiv, so the ArXivMath/BrokenArXiv defaults do not exactly reproduce its grading methodology. APEX25 is also symbolic-only, but its parser differs. Record the judge alongside results, keep it fixed across model comparisons, and retain historical grader attribution.
Local checks
Downloaded and prepared both pinned June datasets; source-field, coverage, prompt, and collation checks passed for all four new variants. Preparation and collation functions were exercised directly: the local Python 3.14 environment has an existing Hydra CLI help-parser incompatibility.
54 existing tests passed across
math_with_judgeandfalse_statement_judge.Offline Gym inheritance resolution and actual model-adapter request construction passed for all ten benchmark variants, with both default and overridden provider settings (20 combinations). Rechecked with the June additions; checked judge routing, medium reasoning, request limits, and one-fold dataset materialization. After the APEX25 update, both APEX25 variants resolve without a judge service; the other four benchmarks retain Luna medium.
pre-commit run --all-filespassed.The prior live replays used a compatible provider; the public OpenAI endpoint itself was not exercised. No new policy generations or GPU jobs were submitted for this PR.
Checklist
pre-commit run --all-files).git commit -s).