arXiv:2501.16729cs.LGcs.AI2025-01被引 3

稀疏结构能显著提升图像强化学习性能,效果取决于权重是否可训练。

On the Interplay Between Sparsity and Training in Deep Reinforcement Learning

  • 对比多种同容量稀疏结构,发现其对学习效果影响显著。
  • 固定权重时,空间偏置结构表现更优;可学习时,全连接结构更佳。
  • 为不同场景选择合适稀疏结构,可提升强化学习效率。

我们研究了深度强化学习中不同稀疏架构的收益。重点关注图像域,其中空间偏置和全连接架构较为常见。在与多种同容量架构对比后,发现稀疏结构对学习性能有显著影响。此外,针对特定任务选择最优稀疏结构,取决于隐藏层权重是否可学习:固定权重时,空间偏置架构表现更优;若权重可学习,则全连接架构更具优势。

原文摘要 · Abstract (English)

We study the benefits of different sparse architectures for deep reinforcement learning. In particular, we focus on image-based domains where spatially-biased and fully-connected architectures are common. Using these and several other architectures of equal capacity, we show that sparse structure has a significant effect on learning performance. We also observe that choosing the best sparse architecture for a given domain depends on whether the hidden layer weights are fixed or learned.

强化学习稀疏结构神经网络

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