Single-Rollout Hidden-State Dynamics for Training-Free RLVR Data Selection
Not on the map. Why?
Abstract (arXiv)
Reinforcement learning with verifiable rewards (RLVR) can yield large reasoning gains from very few training instances, yet its strong sensitivity to which instances are used makes data selection a central bottleneck. Most existing selection pipelines rely on training-time optimization signals and/or require access to verifiable rewards or ground-truth answers over large candidate pools, which is costly and often infeasible in specialized domains. We study RLVR data selection in a setting where selection must be performed before any RL training and without labels or reward evaluation on the full pool. We propose SHIFT, a one-shot, training-free selector based solely on inference-time hidden-state dynamics. For each candidate instance, SHIFT runs a single deterministic reasoning rollout and computes a reasoning-induced representation shift (RIRS) as the start-to-end hidden-state delta. SHIFT uses the RIRS magnitude as a lightweight proxy for instance utility and enforces coverage via a quality-weighted farthest-first CoreSet procedure in an RIRS-augmented feature space, producing compact subsets that scale to large unlabeled pools. Across mathematical reasoning and medical QA benchmarks under ultra-low budgets, SHIFT consistently outperforms training-free diversity and difficulty/uncertainty baselines, improving both in-domain accuracy and transfer to harder evaluation settings. Ablations show that RIRS-based coverage and quality-weighting contribute complementary gains, and analyses indicate that RIRS is not explained by simple input/output length statistics. Code is available at github.com/JianghaoWu/SHIFT.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 36 of its 51 references that are in the map. First posted on arXiv 27 May 2026.
Builds on
36 of 51 references.
References · 36
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2025Qwen3 Technical Reportcited by 8,195
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,324
- 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
- 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,057
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 948
- 2017Deep Bayesian Active Learning with Image Datacited by 812
- 2023Textbooks Are All You Needcited by 561
- 2020A Survey of Deep Active Learningcited by 547
- 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 525
24 more
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 514
- 2023TinyStories: How Small Can Language Models Be and Still Speak Coherent English?cited by 425
- 2019Learning Loss for Active Learningcited by 335
- 2025Reinforcement Learning for Reasoning in Large Language Models with One Training Examplecited by 288
- 2025Reinforcement Learning with Verifiable Rewards Implicitly Incentivizes Correct Reasoning in Base LLMscited by 276
- 2021GRAD-MATCH: Gradient Matching based Data Subset Selection for Efficient Deep Model Trainingcited by 269
- 2025MedXpertQA: Benchmarking Expert-Level Medical Reasoning and Understandingcited by 212
- 2025LIMR: Less is More for RL Scalingcited by 136
- 2019Consistency-based Semi-supervised Active Learning: Towards Minimizing Labeling Costcited by 114
- 2020Estimating Example Difficulty Using Variance of Gradientscited by 96
- 2020Sequential Graph Convolutional Network for Active Learningcited by 69
- 2022Active Learning by Feature Mixingcited by 63
- 2025A Survey on Large Language Models for Mathematical Reasoningcited by 52
- 2021Semi-Supervised Active Learning with Temporal Output Discrepancycited by 50
- 2025Learning without training: The implicit dynamics of in-context learningcited by 41
- 2023Think Twice Before Selection: Federated Evidential Active Learning for Medical Image Analysis with Domain Shiftscited by 26
- 2023Bi3D: Bi-domain Active Learning for Cross-domain 3D Object Detectioncited by 21
- 2026OPUS: Towards Efficient and Principled Data Selection in Large Language Model Pre-training in Every Iterationcited by 17
- 2025CLUE: Non-parametric Verification from Experience via Hidden-State Clusteringcited by 12
- 2025From Data-Centric to Sample-Centric: Enhancing LLM Reasoning via Progressive Optimizationcited by 10
- 2025Difficulty-Based Preference Data Selection by DPO Implicit Reward Gapcited by 7
- 2025Data Selection for Fine-tuning Vision Language Models via Cross Modal Alignment Trajectoriescited by 7
- 2023Influence Scores at Scale for Efficient Language Data Samplingcited by 5
- 2026One Sample to Rule Them All: Extreme Data Efficiency in Multidiscipline Reasoning with Reinforcement Learningcited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Smart Picks in the Dark: Towards Efficient RLVR for Reasoning via Tracing Metacognitive Pivots
- 2025Towards High Data Efficiency in Reinforcement Learning with Verifiable Reward
- 2026IRDS: Interpretable RLVR Data Selection via Verifier-Coupled Sparse Autoencoder Coverage
- 2026Rate or Fate? RLV^varepsilonR: Reinforcement Learning with Verifiable Noisy Rewards
- 2026Detecting RLVR Training Data via Structural Convergence of Reasoning
- 2025The Path Not Taken: RLVR Provably Learns Off the Principals
- 2026Contextual Rollout Bandits for Reinforcement Learning with Verifiable Rewards
- 2025Data-Efficient RLVR via Off-Policy Influence Guidance