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
600
Mathematical ReasoningMATH 500
Top-1 Accuracy82.2
452
Mathematical ReasoningAIME 24
Accuracy27.6
358
Mathematical ReasoningMinerva
Pass@1 Accuracy41.5
289
Mathematical ReasoningMinerva Math
Accuracy42.3
251
Mathematical ReasoningAMC23
PASS@1 Accuracy72.4
216
Mathematical ReasoningOlympiadBench
Accuracy44.4
213
Mathematical ReasoningAIME 25
Pass@1 Accuracy26.9
190
Mathematical ReasoningMinerva
Accuracy (Acc)43.5
146
Mathematical ReasoningOlympiad
Accuracy0.474
136
Showing 10 of 10 rows

Other info

Follow for update