Source-linked AI summary

Easy-to-Hard Generalization: Scalable Alignment Beyond Human Supervision

Zhiqing Sun, Longhui Yu, Yikang Shen, Weiyang Liu, Yiming Yang, Sean Welleck, Chuang Gan

arXiv:2403.09472v2cs.LGcs.AIcs.CL

TL;DR

Human supervision may not extend to hard reasoning tasks beyond human expertise. This paper trains reward models on easier problems to guide solutions to harder ones, finding improved hard-task performance and 34.0 versus 33.0 for easy-task PRM-reward training and the previous state of the art, respectively.

  • Problem

    Human demonstrations and judgments may be unavailable or unhelpful for hard tasks beyond human expertise, leaving open how systems can generalize beyond that supervision.

    Source: §1 · pp. 1–2
  • Method

    The approach trains reward models on annotated easy problems, then uses evaluators to guide policy models on hard problems through re-ranking or reinforcement learning.

    Source: §1 · pp. 2–3
  • Results

    Reward models improved hard-task performance, and easy-task PRM-reward training reached 34.0 versus 33.0 for the previous state-of-the-art model trained across all task levels.

    Source: §1 · p. 3, §4.2.2 · p. 9
  • Takeaways & Limitations

    The findings support easy-to-hard evaluation as a direction for improving difficult-task performance without direct human supervision on those harder tasks.

    Source: §5 · p. 10

    Evidence is scoped to specific model sizes, datasets, and reasoning tasks, while robustness and long-term implications require further research.

    Source: §5 · p. 10

Abstract

from arXiv · show

Current AI alignment methodologies rely on human-provided demonstrations or judgments, and the learned capabilities of AI systems would be upper-bounded by human capabilities as a result. This raises a challenging research question: How can we keep improving the systems when their capabilities have surpassed the levels of humans? This paper answers this question in the context of tackling hard reasoning tasks (e.g., level 4-5 MATH problems) via learning from human annotations on easier tasks (e.g., level 1-3 MATH problems), which we term as easy-to-hard generalization. Our key insight is that an evaluator (reward model) trained on supervisions for easier tasks can be effectively used for scoring candidate solutions of harder tasks and hence facilitating easy-to-hard generalization over different levels of tasks. Based on this insight, we propose a novel approach to scalable alignment, which firstly trains the (process-supervised) reward models on easy problems (e.g., level 1-3), and then uses them to evaluate the performance of policy models on hard problems. We show that such easy-to-hard generalization from evaluators can enable easy-to-hard generalizations in generators either through re-ranking or reinforcement learning (RL). Notably, our process-supervised 7b RL model and 34b model (reranking@1024) achieves an accuracy of 34.0% and 52.5% on MATH500, respectively, despite only using human supervision on easy problems. Our approach suggests a promising path toward AI systems that advance beyond the frontier of human supervision.

1 Introduction

The introduction asks how AI systems can improve on hard tasks when human supervision is limited to easier ones. It proposes training evaluators on easy tasks and using them to guide generators on hard tasks.

  • Current alignment methods rely on human demonstrations or judgments, limiting supervision when tasks exceed human expertise; easy-to-hard generalization asks whether easy-task supervision can enable hard-task success.The setting requires no human supervision on the harder tasks.Source: §1 · pp. 1–2
  • Because SFT or in-context learning on easy tasks often fails to generalize to hard tasks, the authors propose using evaluators, based on the intuition that evaluation is easier than generation.An evaluator trained on easy tasks can supervise a generator on hard tasks.Source: §1 · p. 2
  • The proposed approach trains outcome- and process-supervised reward models on easy tasks, then uses their evaluations to guide hard-task generation through re-ranking or reinforcement learning.The training setup pairs labeled easy questions and high-quality solutions with unlabeled hard questions.Source: §1 · p. 2
  • Training with human supervision only on Level 1–3 problems and answers can outperform SFT and Final-Answer RL trained on the full Level 1–5 dataset.The paper uses this finding to argue that easy-to-hard evaluation can improve generators across varying task complexity.Source: §1 · p. 3

2 Related Work

The related work distinguishes this setting from approaches that extend human oversight or study compositional generalization, and situates its evaluation approach among prior easy-to-hard generation studies.

2.1 Scalable Oversight

  • Unlike approaches that use AI to evaluate other AIs [1] [38] [60] [9], this framework enables models to excel at hard tasks where human supervision may be unavailable, unlike weak-to-strong generalization [11], where supervision may be unreliable.It aligns with the “sandwiching” concept, in which domain experts evaluate outputs of AI-assisted non-experts [9] [57].Source: §2.1 · p. 3

2.2 Compositional Generalization

  • Least-to-most prompting [95] [20] achieves perfect easy-to-hard generalization on SCAN [37] and CFQ [35], and has also succeeded on mathematical reasoning in GSM8K [16] and DROP [21].The cited work connects this success to language models’ capacity to break complex tasks into simpler sub-tasks (Perez et al. [51]).Source: §2.2 · p. 4

2.3 Easy-to-Hard Generalization

  • Prior work trains easy-to-hard generators on easy tasks using SFT or ICL, while this paper evaluates on challenging MATH [32] and reports gains beyond those approaches.Prior findings vary: BERT performed poorly on common-sense reasoning trained only on easy data [71], and ICL showed similar limits on GSM8K [24]; concurrent work found ICL or SFT strong on multiple-choice ARC [15] and MMLU [30]. Iterative machine teaching [43] theoretically supports training classifiers from easy to hard examples.Source: §2.3 · p. 4

3 Methodology

The methodology studies easy-to-hard generalization on MATH by training and evaluating with levels 1–3 as easy tasks and levels 4–5 as hard tasks. It compares generator and evaluator generalization, then uses evaluators to optimize generators on hard problems.

3 Methodology

  • The study treats MATH levels 1–3, with demonstrations and answers, as easy tasks and levels 4–5, represented only by questions, as hard tasks.MATH has 12,500 problems; MATH500 is the test set, 500 problems are sampled for validation, and the remaining 11,500 problems form the training set. SFT uses filtered PRM800K and MetaMATH data, with process labels from human annotations or Math-Shepherd [75].Source: §3 · p. 4, §3.1 · p. 5

3.2 Generators

  • Generator baselines include full- and hard-level ICL, easy-to-hard ICL and SFT, and full SFT, evaluated with greedy decoding or self-consistency [76].Full SFT is treated as a task-specific ceiling [11] [29], while easy-to-hard SFT may generalize to hard tasks with performance degeneration [71].Source: §3.2 · p. 5

3.3 Evaluators

  • OPRM combines process-level judgments on intermediate steps with outcome-level assessment of the final answer, alongside ORM and PRM evaluator baselines.The setup considers evaluators trained on easy tasks or the full dataset; unlike final-answer rewards, reward models trained on easy tasks can score hard-task solutions.Source: §3.3 · pp. 5–6

3.4 Optimizing Generators Against Evaluators

  • Generators are optimized against evaluators by best-of-n reranking, reward-weighted voting, or ReST, DPO, and PPO reinforcement learning.Best-of-n selects the highest-scoring sampled solution; weighted voting weights solutions by reward-model scores.Source: §3.4 · p. 6

3.5 Evaluation Metrics

  • Evaluation focuses on generators’ absolute and relative MATH500 performance, rather than weak-to-strong or easy-to-hard supervision-gap recovery metrics [11, 29, 40].The authors note that models trained only on simpler tasks, particularly with RL, can sometimes outperform models trained across all difficulty levels.Source: §3.5 · p. 6

3.6 Implementation Details

  • Figure 4 reports that easy-trained PRMs improve weighted-voting reranking on hard tasks; experiments use 7b and 34b Llemma models.Models are fully fine-tuned with input-output embeddings and normalization layers frozen; reward models add a scalar head, and PPO’s value model is initialized from the reward model.Source: §3.6 · pp. 6–7, Figure 4 · p. 7

4 Main Results

Easy-trained evaluators improve hard-task performance through re-ranking and reinforcement learning, while generator-only transfer depends on SFT data quality; the evaluator approach also extends to coding.

4.1 Easy-to-Hard Generalization of Generators

  • 35.4 for full MetaMath SFT exceeds 32.2 for its easy-to-hard counterpart by 3.2; only MetaMath-trained models show clear easy-to-hard gaps, unlike PRM800K.SFT also outperforms ICL; the authors hypothesize that lower-quality SFT data may teach task format, while high-quality imitation data can teach solving principles.Source: §4.1 · p. 7

4.2 Easy-to-Hard Generalization of Evaluators

  • A 34b evaluator significantly improves a 7b generator, while a 7b evaluator also improves a 34b generator.Both generators and evaluators are trained on easy data in this comparison.Source: Figure 5 · p. 8, §4.2.1 · p. 8

4.2.1 Re-ranking

  • OPRMs outperform PRMs and ORMs, and weighted voting beats Best-of-n and majority voting, with its advantage growing on harder tasks.On MetaMath/Math-Shepherd, adding ORM training examples yields no significant improvement; Best-of-n remains competitive with majority voting when producing one full solution.Source: §4.2.1 · p. 8

4.2.2 Reinforcement Learning (RL)

  • Top performance of 34.0 with easy-task human PRM rewards exceeds the 33.0 prior state of the art; PPO leads on PRM800K, while PPO and DPO lead on MetaMATH.DPO is on par with PPO when initialized from high-quality SFT, whereas PPO is less dependent on SFT quality.Source: Table 2 · p. 9, §4.2.2 · p. 9

4.3 Easy-to-Hard Generalization on the Coding Domain

  • An ORM trained on outcome supervision from easy Introductory APPS data significantly improves model performance on hard Interview and Competition problems, including Best-of-N reranking performance.The APPS evaluation subsampled 500 test questions: 100 Introductory, 300 Interview, and 100 Competition.Source: §4.3 · p. 10, Table 3 · p. 10

5 Conclusion … C.3 Reinforcement Learning

The paper argues that process-supervised reward models trained on easier tasks can guide policies toward harder problems through reranking or reinforcement learning, without direct human supervision on those harder tasks. The authors note limitations in model and dataset coverage, domain specificity, and evidence about long-term robustness.

5 Conclusion

  • Easy-to-hard evaluators enabled policy models to generalize to difficult tasks through reranking and reinforcement learning, offering a scalable alignment direction.The authors frame the findings as a promising path for AI systems to advance in complex domains without direct human supervision on harder tasks.Source: §5 · p. 10
  • The study is limited by its focus on specific model sizes, datasets, and reasoning domains, with long-term implications and robustness left for future research.Source: §5 · p. 10

A.1 Rationale-Augmented (Mathematical) Reasoning

  • The paper evaluates easy-to-hard generalization in mathematical reasoning, a task with clear difficulty levels, asking whether models can generalize to more challenging problems when trained only on easy mathematical data.Source: §A.1 · p. 18

A.2 Outcome Reward Models & Process Reward Models

  • Unlike prior PRM studies of IID settings, this work studies PRMs as easy-to-hard evaluators and reports that they can enable policy models to generalize to harder tasks.Prior work reported PRM advantages over ORMs in reranking, with similar performance in RL via expert iteration in one study [74] and significant reranking gains in another [40].Source: §A.2 · p. 18

B Reinforcement Learning Algorithms

  • ReST alternates policy sampling with reward-model-weighted supervised fine-tuning, while DPO trains on paired positive and negative samples and PPO samples from the policy during fine-tuning.The passage identifies expert iteration [4] and rejection sampling fine-tuning [72] [87] as ReST variants, and cites PPO use in RLHF [7] [50] and RLAIF [69].Source: B Reinforcement Learning Algorithms · p. 18

C.1 Supervised Fine-Tuning & Reward Modeling

  • SFT uses correct-labeled PRM800K steps, while MetaMath training filters to at most three solutions per question per epoch to reduce overfitting.The PRMs are trained on their corresponding released datasets; ORM solution generation samples 32 solutions per question with top-K 20 and temperature 0.7, balancing positive-to-negative samples between 1:3 and 3:1.Source: §C.1 · p. 19

C.2 Re-Ranking

  • For reranking, the ORM scores a solution at its final token, whereas the PRM scores intermediate reasoning steps and the final token.Sampling uses top-K 20 and temperature 0.7 for majority voting, weighted voting, and best-of-n.Source: §C.2 · p. 19

C.3 Reinforcement Learning

  • RL implementation details include two ReST-EM epochs, with performance degeneration observed from epoch three, and PPO training for 500 steps on non-MATH500 training and validation questions.For iterative DPO, preference pairs require a score difference greater than 1.0; the PPO setup initializes the value model from the reward model and clips gradient norm at 1.Source: §C.3 · pp. 19–20

D Re-ranking Results on MetaMath · E Qualitative Comparison · F Performance of DPO Iterations

The appendix reports that PRMs trained on MetaMath/Math-Shepherd are more effective on harder tasks, alongside qualitative comparisons and analyses of DPO iterations and PPO-related distribution shift.

D Re-ranking Results on MetaMath

  • PRMs are more effective on harder tasks when trained on MetaMath/Math-Shepherd [86] [75].Figure 6 evaluates easy-to-hard generalization with both SFTs and reward models trained on easy data.Source: D Re-ranking Results on MetaMath · p. 20, Figure 6 · p. 21
  • Table 6 reports Pass@N scores, the upper bound of Best-of-N, on APPS coding problems.Source: Table 6 · p. 21
  • Figure 6 presents easy-to-hard evaluator generalization with both SFTs and reward models trained on easy data.Source: Figure 6 · p. 21

E Qualitative Comparison

  • Table 7 compares MetaMath-7b-SFT with MetaMath-7b-PPO and shows the first error Math-Shepherd detected in SFT greedy decoding.The examples are cherry-picked for illustration.Source: Table 7 · p. 22

F Performance of DPO Iterations

  • Figure 7 plots test performance as a function of DPO iterations.Source: Figure 7 · p. 22
  • The authors suggest that a large pre/post-PPO distribution shift may explain PPO’s exceptional performance on PRM800K from a weak SFT initialization.Table 8 reports the shift for a Llemma-7B model trained on PRM800K.Source: Table 8 · p. 23

G Re-ranking Performance Analysis of PRMs, ORMs & OPRMs

In the Llemma + MATH setting, OPRMs trained on mixed process and outcome data outperform PRMs and ORMs in re-ranking, while PRMs and ORMs perform similarly.

G Re-ranking Performance Analysis of PRMs, ORMs & OPRMs

H Re-ranking Results on MetaMath · I More Comparisons · J Hard-to-Easy Generalization

The appendices report reward-model comparisons across MetaMath and PRM800K settings, and find that language models struggle to generalize between easy and hard tasks; training on all data performs best overall.

H Re-ranking Results on MetaMath

  • PRMs trained on MetaMath/Math-Shepherd are more effective on harder tasks, extending the reported easy-to-hard reward-model result to this dataset.The comparison includes PRMs, ORMs, and OPRMs for 7b models trained on MetaMath, with both SFTs and reward models trained on easy data.Source: H Re-ranking Results on MetaMath · p. 26, Figure 10 · p. 26

I More Comparisons

  • Figures 11–14 compare weighted voting and Best-of-N across 7b and 34b PRM800K settings and 7b MetaMath settings, with SFTs and reward models trained on easy data.The comparisons also include a 34b SFT paired with a 7b reward model on PRM800K.Source: Figure 11 · p. 27, Figure 12 · p. 27, Figure 13 · p. 28

J Hard-to-Easy Generalization

  • Full-data SFT consistently performs best overall, while performance deteriorates when training transfers between easy and hard tasks.On hard test questions, Full SFT trails Easy-to-Hard and Hard-to-Easy SFT, but it outperforms both across all test questions.Source: Table 9 · p. 29, J Hard-to-Easy Generalization · p. 29

I More Comparisons

  • Figure 15 compares agreement between a Llemma-7b reward model trained on ORM or PRM data and agreement with ground-truth final-answer labels.Source: Figure 15 · p. 29

J Hard-to-Easy Generalization

  • Hard-task-only reinforcement learning underperforms training on easy tasks or all tasks, especially for PPO on PRM800K.This motivates asking whether training only on hard tasks generalizes to easy tasks.Source: J Hard-to-Easy Generalization · p. 29
  • Training on all data consistently beats easy-only or hard-only Llemma-7b SFT, while transferring from either task difficulty to the other reduces generator performance.Source: J Hard-to-Easy Generalization · p. 29, Table 9 · p. 29
  • Full SFT trails Easy-to-Hard and Hard-to-Easy SFT on hard test questions but surpasses both across all test questions.The authors attribute the full-distribution advantage to exposure to a wider variety of unique questions and difficulties.Source: J Hard-to-Easy Generalization · p. 29, Table 9 · p. 29

K On ORM’s Approximation of PRM Labels

ORMs trained to predict final-answer correctness show higher agreement with PRMs on PRM800K and MetaMath, suggesting they approximate step-wise correctness. This supports easy-to-hard generalization without explicit step-wise annotations, and ORM-based re-ranking performs similarly to PRM-based re-ranking in the reported DPO setting.

K On ORM’s Approximation of PRM Labels

  • ORMs had higher agreement with PRMs on PRM800K and MetaMath despite being trained on final-answer rewards.The evaluation addresses whether ORM’s approximation of PRM labels also holds on the more challenging MATH dataset variants.Source: K On ORM’s Approximation of PRM Labels · p. 29
  • The results indicate that ORMs tend to predict whether the full trace is correct, rather than only whether its final answer is correct.The paper attributes this tendency to ORMs approximating PRM labels [74]; Uesato et al. [74] suggest recognizing correct steps can be simpler than internally computing the final answer.Source: K On ORM’s Approximation of PRM Labels · pp. 29–30
  • Easy-to-hard generalization also applies to value-estimation ORMs, not only reward models trained on explicit step-wise annotations.The paper describes these ORMs as practically evaluating each solution step.Source: K On ORM’s Approximation of PRM Labels · p. 30
  • In DPO training on a MetaMath-initialized Llemma-7b model, ORM re-ranking gave similar performance to PRM re-ranking (29.2 vs. 30.4; 31.2 vs. 34.0).The passage does not specify the metric labels for these paired values.Source: K On ORM’s Approximation of PRM Labels · p. 30

L Analysis of Aggregation Functions in PRMs & OPRMs · M Societal Impact

Across the evaluated datasets and decoding methods, product aggregation works well for weighted voting, while minimum works well for best-of-n; societal impacts could be beneficial but also raise risks requiring safeguards.

L Analysis of Aggregation Functions in PRMs & OPRMs

M Societal Impact

  • Easy-to-hard generalization could enable AI systems to tackle increasingly complex problems in scientific discovery, healthcare, and education, potentially benefiting society.Source: M Societal Impact · p. 33
  • Systems operating beyond human supervision raise concerns about transparency, accountability, and potential misuse.Source: M Societal Impact · p. 33
  • The paper calls for robust safeguards, ethical consideration, ongoing research, and public discourse to mitigate risks and support responsible development and deployment.Source: M Societal Impact · p. 33

N Fine-Grained Analysis of OPRMs’ Re-ranking strategies

OPRM-enhanced re-ranking performs well on harder, unseen level 4–5 problems despite training on level 1–3 data. Its generalization varies by problem type, performing best on Algebra, Intermediate Algebra, and Precalculus and worst on Counting & Probability.

N Fine-Grained Analysis of OPRMs’ Re-ranking strategies

  • Best-of-N and Weighted Voting perform strongly across difficulty levels, including unseen level 4–5 problems, although both SFTs and OPRMs were trained on level 1–3 data.The results indicate that OPRM-based re-ranking can generalize from easier to harder tasks.Source: Figure 19 · p. 34, N Fine-Grained Analysis of OPRMs’… · p. 34
  • OPRMs generalize best on Algebra, Intermediate Algebra, and Precalculus, and worst on Counting & Probability problems.The comparison covers Algebra, Counting & Probability, Geometry, Intermediate Algebra, Number Theory, Prealgebra, and Precalculus.Source: N Fine-Grained Analysis of OPRMs’… · p. 35, Figure 20 · p. 35

O Pass@N Analysis for Different Math Questions

Pass@N generally improves as the number of samples grows, but many problems do not reach 100%; results vary with problem difficulty, model size, and math-question category.

O Pass@N Analysis for Different Math Questions

P Case Studies for OPRM · Q Step and Outcome ROC Curve · R Few-Shot Prompt in In-Context Learning

This section presents OPRM case studies and ROC analyses of step and outcome discrimination, alongside 4-shot ICL prompt examples for MetaMath Hard and Easy-to-Hard settings.

P Case Studies for OPRM

Q Step and Outcome ROC Curve

  • PRM and OPRM discriminate reasoning steps better than ORM, while OPRM discriminates final outcomes better than both ORM and PRM.The step judgments use balanced samples from PRM800K-test; outcome curves use balanced labeled generations from the 7B policy model on MATH500, and stronger step discrimination does not necessarily make an evaluator more helpful for generation.Source: Q Step and Outcome ROC Curve · p. 38, Table 10 · p. 38, Figure 24 · p. 38

R Few-Shot Prompt in In-Context Learning

Loading 2403.09472v2…