Network Sparsity Unlocks the Scaling Potential of Deep Reinforcement Learning

Guozheng Ma, Lu Li, Zilin Wang, Li Shen, Pierre-Luc Bacon +1 more
6/20/2025
cs.LGcs.AI

No official code found. Best implementation: lilucse/SparseNetwork4DRL (41 stars, last commit 2025-06-05).

Abstract

Effectively scaling up deep reinforcement learning models has proven notoriously difficult due to network pathologies during training, motivating various targeted interventions such as periodic reset and architectural advances such as layer normalization. Instead of pursuing more complex modifications, we show that introducing static network sparsity alone can unlock further scaling potential beyond their dense counterparts with state-of-the-art architectures. This is achieved through simple one-shot random pruning, where a predetermined percentage of network weights are randomly removed once before training. Our analysis reveals that, in contrast to naively scaling up dense DRL networks, such sparse networks achieve both higher parameter efficiency for network expressivity and stronger resistance to optimization challenges like plasticity loss and gradient interference. We further extend our evaluation to visual and streaming RL scenarios, demonstrating the consistent benefits of network sparsity.

View on arXivView PDF

Code Implementations(1)

[ICML 2025 oral] Network Sparsity Unlocks the Scaling Potential of Deep Reinforcement Learning

411PythonJun 5, 20251 years ago
deep-learningdeepmind-control-suitereinforcement-learningsparse-neural-networks

Similar Papers

Cite this paper

@article{ma2025network,
  title  = {Network Sparsity Unlocks the Scaling Potential of Deep Reinforcement Learning},
  author = {Guozheng Ma and Lu Li and Zilin Wang and Li Shen and Pierre-Luc Bacon and Dacheng Tao},
  year   = {2025},
  eprint = {2506.17204},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2506.17204v1},
  journal = {ICML 2025 2025}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2506.17204.svg)](https://codeofpaper.com/paper/2506.17204)

Discussion