Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Transferability for General Reasoning: An Automated Curriculum for Multi-Domain RLVR

About

Reinforcement learning with verifiable rewards (RLVR) has been extended from single-domain training to multi-domain reasoning suites spanning mathematics, programming, and science. However, the training curriculum (how often each domain is sampled) is typically fixed or hand-tuned, even though reasoning skills transfer unevenly across domains. Existing learnability-based curricula adapt to where the policy is currently improving, but are blind to whether a gradient step on the selected domain benefits the remaining domains. In this paper, we propose Transfer-Aware Curriculum (TAC), a bandit-style online curriculum that prioritizes domains whose updates broadly benefit the rest of the training suite. TAC repurposes signals already produced by RL training: per-domain advantages capture local learnability, and projected gradients, taken from the GRPO step being computed, estimate cross-domain transferability via gradient-geometry alignment, at negligible cost (<1% wall-clock overhead). Across a six-domain reasoning suite, TAC achieves the best macro-averaged accuracy on both Qwen3-1.7B and Llama3.2-3B, outperforming proportional random sampling, a hand-designed schedule, and a learnability-only bandit, and improving over the last of these by up to 2.8 points (10% relative). Ablations show performance degrades sharply when the transferability term is removed, and TAC remains robust on imbalanced training mixtures where learnability-only curricula over-commit to dominant domains. Our findings establish cross-domain transferability as a key signal for curriculum design in multi-domain RLVR.

Yongjin Yang, Jiarui Liu, Yinghui He, Lechen Zhang, Bernhard Sch\"olkopf, Zhijing Jin• 2026

Related benchmarks

TaskDatasetResultRank
Code GenerationHumanEval (test)
Pass@183.5
701
Code GenerationMBPP (test)
Pass@167.4
411
Code GenerationHumanEval
pass@166.7
329
Mathematical ReasoningAIME
Pass@112.3
32
Code SimulationCRUXEval-I
Pass@161.4
16
STEM ReasoningSuperGPQA
Pass@1 Accuracy32
16
Table ReasoningFinQA
Pass@139.7
16
Table ReasoningHiTab
Pass@170.1
16
Code SimulationCRUXEval-O
Pass@162.1
16
Logical reasoningARC-AGI
Pass@12.6
16
Showing 10 of 19 rows

Other info

GitHub

Follow for update