边训练边剪枝,让强化学习网络更小更快且不丢性能。
Online Training and Pruning of Deep Reinforcement Learning Networks
- 用随机变量控制神经元开关,自动判断是否剪掉无用结构。
- 在MuJoCo上剪掉大量参数,性能损失小于1%。
- 适合追求高效强化学习模型的算法研究者和工程师。
将深度神经网络用于强化学习可提升性能,但带来显著计算与内存开销。尽管监督学习中剪枝已有效缓解此问题,其在强化学习中的应用仍较少。本文提出一种在高级强化学习算法中同步进行训练与剪枝的方法,特别针对引入在线特征提取网络(OFENet)的算法。所提网络XiNet通过求解随机优化问题,联合训练网络权重与变分伯努利分布参数ξ(控制每个单元的0/1开关)。该随机形式引入正则化项,使贡献低的单元对应的参数趋于0,从而永久禁用并剪除。我们设计了一种面向DenseNet架构的、成本感知的稀疏性促进正则化方案,以网络参数数量表示复杂度,并自动匹配正则化项,实现强化学习目标与压缩效果的协同优化。在连续控制基准(MuJoCo)和Soft Actor-Critic代理上评估表明,可大幅剪枝而性能损失极小。结果还显示,训练过程中剪枝大模型比从头训练小模型更高效、表现更好。
原文摘要 · Abstract (English)
Scaling deep neural networks (NN) of reinforcement learning (RL) algorithms has been shown to enhance performance when feature extraction networks are used but the gained performance comes at the significant expense of increased computational and memory complexity. Neural network pruning methods have successfully addressed this challenge in supervised learning. However, their application to RL is underexplored. We propose an approach to integrate simultaneous training and pruning within advanced RL methods, in particular to RL algorithms enhanced by the Online Feature Extractor Network (OFENet). Our networks (XiNet) are trained to solve stochastic optimization problems over the RL networks' weights and the parameters of variational Bernoulli distributions for 0/1 Random Variables $ξ$ scaling each unit in the networks. The stochastic problem formulation induces regularization terms that promote convergence of the variational parameters to 0 when a unit contributes little to the performance. In this case, the corresponding structure is rendered permanently inactive and pruned from its network. We propose a cost-aware, sparsity-promoting regularization scheme, tailored to the DenseNet architecture of OFENets expressing the parameter complexity of involved networks in terms of the parameters of the RVs in these networks. Then, when matching this cost with the regularization terms, the many hyperparameters associated with them are automatically selected, effectively combining the RL objectives and network compression. We evaluate our method on continuous control benchmarks (MuJoCo) and the Soft Actor-Critic RL agent, demonstrating that OFENets can be pruned considerably with minimal loss in performance. Furthermore, our results confirm that pruning large networks during training produces more efficient and higher performing RL agents rather than training smaller networks from scratch.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。