Search papers, labs, and topics across Lattice.
This paper introduces a graph-based online difficulty estimator for reinforcement learning with verifiable rewards (RLVR), addressing the inefficiencies of traditional rollout exploration methods that treat all samples uniformly. By constructing a difficulty-aware sample graph that captures semantic and reasoning similarities, the proposed method allows for continuous updates of difficulty estimates, effectively mitigating issues of cold start and stale feedback without the need for dedicated probing. Experimental results show that this approach enhances exploration efficiency and improves performance across various models and benchmarks.
Sharing rollout feedback across related samples can significantly boost exploration efficiency in RLVR, leading to better reasoning capabilities in large language models.
Reinforcement learning with verifiable rewards (RLVR) improves the reasoning capabilities of large language models but relies on costly rollout exploration. Assigning the same exploration budget to samples with different difficulty levels is inefficient: easy samples may receive redundant rollouts, whereas difficult but learnable samples may receive too little exploration. Existing adaptive schedulers address this mismatch through curriculum-based sample selection or non-uniform rollout allocation based on estimated sample difficulty. However, obtaining reliable online difficulty estimates remains challenging: dedicated probing adds substantial generation overhead, whereas history-based estimators face a cold start with no initial observations and stale feedback, and typically ignore relations among samples. To address these limitations, we propose a plug-and-play graph-based online difficulty estimator that shares rollout feedback across related samples and continuously updates their difficulty estimates, mitigating cold start and staleness without dedicated probing. Specifically, we first construct a difficulty-aware sample graph based on semantic and reasoning similarities. Based on this graph, we introduce latent difficulty states and use a Potts prior to encourage neighboring samples to share the same state. We then employ a state-level Beta-Binomial model to aggregate the rollout outcomes associated with each state. Finally, we use an online mean-field variational algorithm to continuously update the latent-state assignments and state-level difficulty as new feedback arrives. Our framework can be integrated into sample-selection and rollout-allocation schedulers, enabling difficulty-adaptive exploration without dedicated probing. Experiments across multiple base models, RL schedulers, and benchmarks demonstrate that our framework achieves better performance.
