用教师信号重构奖励,让模型在长文本处理中每步都能得到精准反馈。
TAMTRL: Teacher-Aligned Reward Reshaping for Multi-Turn Reinforcement Learning in Long-Context Compression
- 用相关文档作教师信号,自监督生成每步的奖励
- 在7个长文本任务上超越强基线,提升记忆更新精度
- 适合需要多轮推理的长文本生成与压缩场景
大语言模型在众多任务中表现优异,但面对超出上下文窗口的长文档时,需分块处理并多轮更新记忆。然而,仅最终结果提供监督,导致各轮次的记忆更新难以评估,引发时间上的信用分配难题。现有方法如LLM作为裁判或过程奖励模型计算开销大且存在估计噪声。为此,我们提出教师对齐奖励重塑(TAMTRL),通过将相关文档与模型每轮输入对齐,利用归一化概率自监督地分配奖励,为每轮记忆更新提供细粒度学习信号,显著提升长文本处理能力。在多个规模不同的模型和七个长文本基准上的实验表明,TAMTRL持续优于强基线,验证了其有效性。代码已公开于https://anonymous.4open.science/r/TAMTRL-F1F8。
原文摘要 · Abstract (English)
The rapid progress of large language models (LLMs) has led to remarkable performance gains across a wide range of tasks. However, when handling long documents that exceed the model's context window limit, the entire context cannot be processed in a single pass, making chunk-wise processing necessary. This requires multiple turns to read different chunks and update memory. However, supervision is typically provided only by the final outcome, which makes it difficult to evaluate the quality of memory updates at each turn in the multi-turn training setting. This introduces a temporal credit assignment challenge. Existing approaches, such as LLM-as-a-judge or process reward models, incur substantial computational overhead and suffer from estimation noise. To better address the credit assignment problem in multi-turn memory training, we propose Teacher-Aligned Reward Reshaping for Multi-Turn Reinforcement Learning (TAMTRL). TAMTRL leverages relevant documents as teacher signals by aligning them with each turn of model input and assigns rewards through normalized probabilities in a self-supervised manner. This provides fine-grained learning signals for each memory update and improves long-context processing. Experiments with multiple models of varying scales across seven long-context benchmarks show that TAMTRL consistently outperforms strong baselines, demonstrating its effectiveness. Our code is available at https://anonymous.4open.science/r/TAMTRL-F1F8.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。