Neural Network Compression for Reinforcement Learning Tasks

  • 2024-05-13 14:46:02
  • Dmitry A. Ivanov, Denis A. Larionov, Oleg V. Maslennikov, Vladimir V. Voevodin
  • 0

Abstract

In real applications of Reinforcement Learning (RL), such as robotics, lowlatency and energy efficient inference is very desired. The use of sparsity andpruning for optimizing Neural Network inference, and particularly to improveenergy and latency efficiency, is a standard technique. In this work, weperform a systematic investigation of applying these optimization techniquesfor different RL algorithms in different RL environments, yielding up to a400-fold reduction in the size of neural networks.

 

Quick Read (beta)

loading the full paper ...