Hints, Critics, and Teachers: Prior Injection for Sparse-Reward RL in Vision-Language Math Reasoning
Abstract
Reinforcement learning for vision-language math reasoning starves under sparse reward: on a pool of 20,830 visual-math problems where Qwen2-VL-2B answers 3.6% of rollouts correctly, 85-97% of GRPO rollout groups are entirely wrong and contribute zero gradient. We train eleven methods under identical conditions in this regime, each injecting a different prior: text (reference-solution hints), distribution (on-policy distillation from a 7B teacher), and value (a value-pretrained critic with an MSE or HL-Gauss categorical loss). A prior helps exactly when it is delivered: the six arms whose prior effectively reaches the policy separate with no overlap from the remaining five -- the no-prior baseline and four arms whose prior is teacher-capped, gated away, or lost to a mis-parameterized critic -- both on the pooled in-domain metric and on cross-domain transfer (DynaMath). The central finding, however, concerns evaluation: one slice of the in-domain pool -- long used as this project's general-distribution check -- anti-correlates with genuine cross-domain transfer (Spearman rho = -0.74, n = 11 arms, permutation p = 0.011), while the hardest in-domain slice predicts it closely (rho = +0.89, p < 0.001). We attribute the inversion to a near-chance multiple-choice subset that rewards models for not having changed; read through it, the best cross-domain method looked mediocre and the worst looked like the champion. Among the methods, hint-guided exploration -- not UFT's auxiliary loss -- drives hint gains, and replacing the critic's MSE loss with HL-Gauss cross-entropy is worth +14.4 points in-domain. All accuracies are blind-judged, with paired exact tests.
Figures & tables
| Arm | Prior | Mechanism |
|---|---|---|
| grpo | none | plain GRPO, groups of (the baseline) |
| hintonly | text | reference-solution prefix, cosine-annealed , then off |
| uft001 | text | hintonly + NLL on hint tokens, coef. 0.001 (full UFT) |
| uft002 | text | hintonly + NLL on hint tokens, coef. 0.002 |
| opd | distribution | pure on-policy distillation: advantage reverse KL, no reward |
| reopd | distribution | opd + teacher-trajectory prefix replay |
| Domain | Set | Nature | Base (%) | |
|---|---|---|---|---|
| in-domain | MathV (pooled) | 1119 | sparse-800 old-319, size-weighted 71.5/28.5 | |
| sparse-800 | 800 | training-distribution hard tail, never trained on | ||
| old-319 | 319 | earlier MathV360K slice; 69% Geometry3K 4-option MC | — | |
| cross-domain | DynaMath | 477 | programmatically rendered; zero overlap with MathV360K | 25.8 |
| easy | 373 | 30.3 | ||
| hard | 104 | 11.5 |
| Arm | Prior | sparse-800 | old-319 | MathV (pooled) | DynaMath |
|---|---|---|---|---|---|
| uft002 | text | 52.4 | 19.1 | 42.9 | 32.9 |
| vapohl | value (HL-Gauss) | 53.6 | 19.7 | 44.0 | 31.9 |
| hintonly | text | 51.4 | 21.3 | 42.8 | 31.2 |
| hintsaf | text + distribution | 49.4 | 26.0 | 42.7 | 30.8 |
| uft001 | text | 52.0 | 15.7 | 41.6 | 30.8 |
| saf | distribution (annealed) | 49.6 | 30.4 | 44.1 | 30.6 |
| X | Y | |
|---|---|---|
| sparse-800 | DynaMath | |
| old-319 | DynaMath | |
| MathV pooled (71.5/28.5) | DynaMath | |
| MathV pooled (50/50) | DynaMath | |
| sparse-800 | old-319 | |
| sparse-800 | DynaMath-easy |
| Earlier reading | Corrected |
|---|---|
| All arms lie on one specialization-versus-generalization trade-off curve; none advances both ends. | The apparent curve is capacity reallocation within MathV360K. Across domains there is no trade-off: stronger in-domain predicts stronger out-of-domain. |
| vapohl is the specialization extreme, paying a generalization tax. | vapohl is second-best cross-domain (31.9). Its low old-319 score reflects only that it stopped guessing option letters. |
| vapo is the robustness champion—best on the “general-distribution” set. | vapo is the weakest trained arm cross-domain (27.0, below plain grpo ). old-319 had the ranking inverted. |
| Hypothesis | Verdict |
|---|---|
| Answer-space guessing | Rejected: 48.9% accuracy against an 8.7% aggregate guessing ceiling, with a diverse answer distribution. |
| Format-reward hacking | Rejected: correct rows carry genuine multi-step reasoning (median 270 characters). |
| Template leakage from training | Partial: roughly 3 pp of attributable inflation; rows without a template duplicate in the training pool still score 45.9% against a 3.5% base. |
| Composition artifacts on old-319 | Confirmed—and followed to the evaluation-validity finding of Section 4 . |
Reinforcement Learning
Reinforcement learning studies how agents learn sequential decisions from rewards and interaction, balancing exploration, performance, safety, and adaptation.
See all 1,727 papers