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

Scheduling Your LLM Reinforcement Learning with Reasoning Trees

About

Using Reinforcement Learning with Verifiable Rewards (RLVR) to optimize Large Language Models (LLMs) can be conceptualized as progressively editing a query's `Reasoning Tree'. This process involves exploring nodes (tokens) and dynamically modifying the model's policy at each node. When combined with data scheduling, this process yields further gains in data efficiency and accuracy. However, existing RLVR data scheduling methods typically rely on path-based metrics to rank queries, overlooking the reasoning tree structures of these queries. In this paper, we introduce a novel metric, namely Reasoning Score (r-score), which measures the query's learning difficulty based on the structure of its reasoning tree. Based on the r-score, we propose the Reasoning Tree Schedule (Re-Schedule), a scheduling algorithm that constructs a curriculum progressing from structurally simple (high r-score) to complex (low r-score) queries. Experiments on six math-reasoning benchmarks show that Re-Schedule significantly improves average accuracy, achieving gains of up to 3.2%. These strong results validate our approach and demonstrate that a structural understanding of the reasoning tree provides a more powerful and principled foundation for RLVR data scheduling.

Hong Wang, Zhezheng Hao, Jian Luo, Chenxing Wei, Yao Shu, Lei Liu, Qiang Lin, Hande Dong, Jiawei Chen• 2025

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMATH 500
Accuracy (Acc)89.8
543
Mathematical ReasoningMATH 500
Top-1 Accuracy82.2
384
Mathematical ReasoningAIME 24
Accuracy27.6
318
Mathematical ReasoningMinerva
Pass@1 Accuracy41.5
289
Mathematical ReasoningMinerva Math
Accuracy42.3
233
Mathematical ReasoningOlympiadBench
Accuracy44.4
213
Mathematical ReasoningAMC23
PASS@1 Accuracy72.4
207
Mathematical ReasoningAIME 25
Pass@1 Accuracy26.9
178
Mathematical ReasoningMinerva
Accuracy (Acc)43.5
146
Mathematical ReasoningOlympiad
Accuracy0.474
134
Showing 10 of 10 rows

Other info

Follow for update