arXiv:2603.19310cs.LGcs.AI2026-03

用图神经网络给少标签的LLM推理任务自动补奖,效果接近全标签训练。

MemReward: Graph-Based Experience Memory for LLM Reward Prediction with Limited Labels

  • 构建包含思维过程和答案的异构图,通过相似性连接样本节点
  • 仅用20%真实标签,数学/问答/代码任务性能达基准96.6%~97.3%
  • 适合标注成本高、标签稀缺的LLM强化学习场景

强化学习已成为提升大语言模型(LLM)推理能力的重要方法,其依赖对策略采样的轨迹(rollouts)计算奖励信号来优化策略。然而,在数据稀缺场景下,大规模获取真实标签通常需昂贵的人工标注或专家验证。例如,数学证明需专家评审,开放问答无明确答案标准。当真实标签稀少时,强化学习微调效果受限。受半监督学习传播标签思想启发,我们提出MemReward:一种基于图的经验记忆框架,将奖励传播直接融入在线策略优化。该框架将初始LLM策略生成的轨迹(思维过程与最终答案)作为异构图中的节点,通过相似性和结构边连接,并利用图神经网络(GNN)从有标签轨迹向无标签轨迹传播奖励。训练时,先在有标签轨迹上预热GNN,通过查询、思考和答案节点的异构聚合预测奖励。在线强化学习微调中,新无标签轨迹根据查询相似性接入图,由GNN预测奖励,形成结合真实与预测奖励的混合奖励获取策略。在Qwen2.5-1.5B和3B模型上,针对数学、问答、代码生成任务的实验表明,仅使用20%真实标签,即可在1.5B模型上达到96.6%的基准性能,在3B模型上达97.3%,并在跨领域任务上表现接近理想基准(Oracle)。

原文摘要 · Abstract (English)

Reinforcement learning has emerged as a powerful paradigm for improving large language model (LLM) reasoning, where rollouts are sampled from the policy and reward signals computed on those rollouts are used to update the policy. However, in data-scarce scenarios, obtaining ground-truth labels to verify rollouts at scale often requires expensive human annotation or labor-intensive expert verification. For instance, evaluating mathematical proofs demands expert review, and open-ended question answering lacks definitive ground truth. When ground-truth labels are scarce, the effectiveness of reinforcement learning fine-tuning is constrained. Inspired by the success of semi-supervised learning in propagating labels from labeled to unlabeled samples, we propose MemReward, a graph-based experience memory framework that integrates reward propagation directly into online policy optimization. MemReward stores rollouts (thinking processes and final answers) from an initial LLM policy as nodes in a heterogeneous graph connected by similarity and structural edges, over which a GNN propagates rewards from labeled to unlabeled rollouts. To train such a framework, we first warm up the GNN on labeled rollouts to predict rewards via heterogeneous aggregation over query, thinking, and answer nodes. During online RL fine-tuning, unlabeled rollouts are attached to the graph by query similarity, and the GNN predicts their rewards, yielding a hybrid reward acquisition strategy that combines ground-truth and GNN-predicted rewards. Experiments on Qwen2.5-1.5B and 3B in mathematics, question answering, and code generation demonstrate that MemReward, with ground-truth rewards on only 20% of rollouts, achieves 96.6% of Oracle performance on 1.5B and 97.3% on 3B, and closely approaches Oracle on out-of-domain tasks.

LLM推理强化学习少样本图神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。