用因果推理测试强化学习奖励,发现模型能力是决定效果的关键。
Generalization of RLVR Using Causal Reasoning as a Testbed
- 在因果图上用强化学习优化语言模型的推理能力。
- 大模型(32B)在复杂查询上准确率提升超15%。
- 仅当模型初始能力足够时,强化学习才有效改善推理策略。
强化学习结合可验证奖励(RLVR)已成为大语言模型在复杂推理任务后训练的有前景范式。然而,其泛化能力在何种条件下成立仍不明确。本文以因果图模型中的概率推理为测试场景,从查询类型(关联、干预、反事实)和查询结构复杂度(相关子图规模)两个维度评估通用性。构建包含不同难度的因果图与查询数据集,对Qwen-2.5-Instruct系列(3B–32B)模型采用RLVR或监督微调(SFT)进行训练。结果表明,当模型规模与训练查询层级匹配时,RLVR在同层级及跨层级泛化上均优于SFT;进一步分析显示,其有效性依赖于模型初始推理能力。具备足够基础能力的模型经RLVR后,能改进边缘化策略,降低中间概率计算错误,在复杂查询上取得显著准确率提升,最高达15%以上。研究证明,RLVR可针对性提升因果推理子技能,但前提是模型具备基本推理能力。代码与数据已公开于https://github.com/zhichul/rlcausal。
原文摘要 · Abstract (English)
Reinforcement learning with verifiable rewards (RLVR) has emerged as a promising paradigm for post-training large language models (LLMs) on complex reasoning tasks. Yet, the conditions under which RLVR yields robust generalization remain underexplored. This paper provides an empirical study of RLVR generalization in the setting of probabilistic inference over causal graphical models. This setting offers two natural axes along which to examine generalization: (i) the level of the probabilistic query -- associational, interventional, or counterfactual -- and (ii) the structural complexity of the query, measured by the size of its relevant subgraph. We construct a dataset of causal graphs and queries spanning these difficulty axes and fine-tune Qwen-2.5-Instruct models using RLVR or supervised fine-tuning (SFT). We vary both the model scale (3B-32B) and the query level included in training. We find that RLVR yields stronger within-level and across-level generalization than SFT, but only for specific combinations of model size and training query level. Further analysis shows that RLVR's effectiveness depends on the model's initial reasoning competence. With sufficient initial competence, RLVR improves an LLM's marginalization strategy and reduces errors in intermediate probability calculations, producing substantial accuracy gains, particularly on more complex queries. These results show that RLVR can improve specific causal reasoning subskills, with its benefits emerging only when the model has sufficient initial competence. Our code and data is available at https://github.com/zhichul/rlcausal.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。