CircuitLens: Reasoning Circuits as Data Selection Signals for Reinforcement Learning with Verifiable Rewards

📅 2026-09-07
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
研究通过引入Circuit Reasoning Score方法,解决了强化学习中基于可验证奖励的数据选择问题,该方法评估数据价值与模型相关,而非仅视作问题的固有属性。
📝 Abstract
Reinforcement learning with verifiable rewards (RLVR) is sensitive to which problems a model trains on, yet existing selection criteria--difficulty filtering, hand-curation, reward-trajectory scoring--assess data value as an intrinsic property of problems, independent of the model that will learn from them. We introduce Circuit Reasoning Score (CRS), a selection signal derived from 46 reasoning-sensitive attention heads identified via contrastive ablation, computed in a single forward pass on the frozen base model without reward labels or rollouts. CRS runs against the intuitive hypothesis that stronger reasoning-circuit engagement produces better training data: on Qwen2.5-Math-7B, the lowest-engagement decile improves over random selection on three medium-difficulty benchmarks (GSM8K +2.0 pp, OlympiadBench +1.6 pp, Minerva +2.9 pp), while the highest-engagement decile gains less and is indistinguishable from the middle decile. The advantage has boundary conditions: on a domain-curated pool no selection method separates from the others; at 1.5B scale the useful direction differs; and the lowest-reward training condition produces the strongest downstream generalization. Within the Qwen2.5-Math settings tested, RLVR data selection appears regime-dependent rather than reducible to a static ranking of problem quality.
Problem

Research questions and friction points this paper is trying to address.

Reinforcement Learning
Verifiable Rewards
Data Selection
Model Training
Circuit Reasoning
Innovation

Methods, ideas, or system contributions that make the work stand out.

Circuit Reasoning Score
contrastive ablation
attention heads
data selection
reinforcement learning with verifiable rewards
Z
Zhuofan Chen
School of Computer Science and Engineering, Beihang University, Beijing, China
Z
Ziqian Jiao
School of Computer Science and Engineering, Beihang University, Beijing, China
Y
Yikai Cui
School of Computer Science and Engineering, Beihang University, Beijing, China
Z
Zhixin Cai
School of Computer Science and Engineering, Beihang University, Beijing, China
Jun Bai
Jun Bai
Assistant professor
Computer aided drug discoveryMedical image analysisAI therapeutic target identification
Wenge Rong
Wenge Rong
Beihang University
Natural Language ProcessingMachine LearningInformation Systems