arXiv:2410.09754cs.LGcs.AI2024-10ICLR被引 78

通过简单性偏置提升深度强化学习模型参数规模,显著改善采样效率。

SimBa: Simplicity Bias for Scaling Up Parameters in Deep Reinforcement Learning

  • 引入输入归一化、残差前馈和层归一化构建简单性偏置架构
  • 在多个环境上提升SAC等算法的样本效率,性能媲美或超越顶尖方法
  • 适用于多种强化学习算法,适合追求高效训练的开发者

近期计算机视觉与自然语言处理的进步主要源于网络参数的规模化,尽管传统理论认为大模型易过拟合。这些大模型通过引入诱导简单性偏置的组件,引导模型走向简洁且泛化能力强的解。然而,深度强化学习中对网络设计与扩展的研究仍较少。为此,本文提出SimBa架构,通过注入简单性偏置实现深度强化学习模型的参数规模化。SimBa包含三个组件:(i) 使用运行统计量标准化输入的观察归一化层,(ii) 提供从输入到输出线性路径的残差前馈块,(iii) 控制特征幅度的层归一化。通过在多种深度强化学习算法(包括离策略、在线策略及无监督方法)中集成SimBa,其样本效率得到一致提升。仅将SimBa应用于SAC,在DMC、MyoSuite和HumanoidBench等多个基准上均达到或超越当前最优方法,且计算效率高。结果表明SimBa在不同算法与环境间具有广泛适用性与有效性。

原文摘要 · Abstract (English)

Recent advances in CV and NLP have been largely driven by scaling up the number of network parameters, despite traditional theories suggesting that larger networks are prone to overfitting. These large networks avoid overfitting by integrating components that induce a simplicity bias, guiding models toward simple and generalizable solutions. However, in deep RL, designing and scaling up networks have been less explored. Motivated by this opportunity, we present SimBa, an architecture designed to scale up parameters in deep RL by injecting a simplicity bias. SimBa consists of three components: (i) an observation normalization layer that standardizes inputs with running statistics, (ii) a residual feedforward block to provide a linear pathway from the input to output, and (iii) a layer normalization to control feature magnitudes. By scaling up parameters with SimBa, the sample efficiency of various deep RL algorithms-including off-policy, on-policy, and unsupervised methods-is consistently improved. Moreover, solely by integrating SimBa architecture into SAC, it matches or surpasses state-of-the-art deep RL methods with high computational efficiency across DMC, MyoSuite, and HumanoidBench. These results demonstrate SimBa's broad applicability and effectiveness across diverse RL algorithms and environments.

强化学习模型扩展样本效率

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