arXiv:2604.14922cs.LGcs.CL2026-04被引 1

通过激活值引导稀疏更新,提升长文本强化学习效果

LongAct: Harnessing Intrinsic Activation Patterns for Long-Context Reinforcement Learning

论文配图:LongAct: Harnessing Intrinsic Activation Patterns for Long-Context Reinforcement Learning
图 1 · 摘自论文原文
  • 基于高激活值筛选关键权重进行稀疏更新
  • 在LongBench v2上提升约8%,RULER上增强泛化能力
  • 适用于多种RL算法,具有强通用性

强化学习(RL)已成为提升大语言模型(LLM)推理能力的关键驱动力。尽管近期研究多聚焦于奖励设计或数据合成,却较少利用模型内在表征特性来指导训练。本文首次观察到,在处理长上下文时,查询和键向量中存在显著的高幅度激活。受模型量化中高激活重要性的启发,并结合长上下文推理天然稀疏的特性,我们提出假设:这些高激活权重是优化过程的关键驱动。据此,我们提出LongAct策略,将更新方式从均匀更新转为基于显著性的稀疏更新。仅对与高激活相关联的权重进行更新,使模型在LongBench v2上性能提升约8%,并在RULER基准上增强泛化能力。此外,该方法表现出优异的普适性,可稳定提升GRPO、DAPO等多种强化学习算法的表现。大量消融实验表明,聚焦显著特征是释放长上下文潜力的关键。

原文摘要 · Abstract (English)

Reinforcement Learning (RL) has emerged as a critical driver for enhancing the reasoning capabilities of Large Language Models (LLMs). While recent advancements have focused on reward engineering or data synthesis, few studies exploit the model's intrinsic representation characteristics to guide the training process. In this paper, we first observe the presence of high-magnitude activations within the query and key vectors when processing long contexts. Drawing inspiration from model quantization -- which establishes the criticality of such high-magnitude activations -- and the insight that long-context reasoning inherently exhibits a sparse structure, we hypothesize that these weights serve as the pivotal drivers for effective model optimization. Based on this insight, we propose LongAct, a strategy that shifts from uniform to saliency-guided sparse updates. By selectively updating only the weights associated with these significant activations, LongAct achieves an approximate 8% improvement on LongBench v2 and enhances generalization on the RULER benchmark. Furthermore, our method exhibits remarkable universality, consistently boosting performance across diverse RL algorithms such as GRPO and DAPO. Extensive ablation studies suggest that focusing on these salient features is key to unlocking long-context potential.

强化学习长文本推理稀疏更新大模型优化

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