arXiv:2510.12096cs.LG2025-10被引 1

针对大模型在强化学习中表现下降的问题,提出模块化动态稀疏训练方法。

Rethinking the Role of Dynamic Sparse Training for Scalable Deep Reinforcement Learning

  • 按模块差异设计动态稀疏策略,而非统一方法
  • 在多种算法上实现显著可扩展性提升,无需修改算法
  • 系统比较了不同稀疏训练路径,揭示最优组合

扩大神经网络规模推动了机器学习的突破,但在深度强化学习(DRL)中,更大模型常因独特的优化困境(如可塑性损失)导致性能下降。尽管动态调整训练过程中的网络拓扑可缓解此类问题,现有研究存在三大局限:(1)对编码器、价值函数和策略模块采用统一的动态策略,忽视其学习范式差异;(2)评估局限于基础架构,未厘清动态训练与架构改进之间的相对重要性及交互关系;(3)缺乏对稀疏到稀疏、密集到稀疏、稀疏到密集等不同动态路径的系统比较。通过跨模块与架构的全面分析,我们发现动态稀疏训练能为不同模块带来特定优势,且与架构改进形成互补。最终提炼出模块化动态训练(MST)框架,进一步释放架构改进潜力,在不修改算法的前提下,显著提升多种强化学习算法的可扩展性。

原文摘要 · Abstract (English)

Scaling neural networks has driven breakthrough advances in machine learning, yet this paradigm fails in deep reinforcement learning (DRL), where larger models often degrade performance due to unique optimization pathologies such as plasticity loss. While recent works show that dynamically adapting network topology during training can mitigate these issues, existing studies have three critical limitations: (1) applying uniform dynamic training strategies across all modules despite encoder, critic, and actor following distinct learning paradigms, (2) focusing evaluation on basic architectures without clarifying the relative importance and interaction between dynamic training and architectural improvements, and (3) lacking systematic comparison between different dynamic approaches including sparse-to-sparse, dense-to-sparse, and sparse-to-dense. Through comprehensive investigation across modules and architectures, we reveal that dynamic sparse training strategies provide module-specific benefits that complement the primary scalability foundation established by architectural improvements. We finally distill these insights into Module-Specific Training (MST), a practical framework that further exploits the benefits of architectural improvements and demonstrates substantial scalability gains across diverse RL algorithms without algorithmic modifications.

强化学习稀疏训练可扩展性模块化

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