Learning How Deep to Go: Self-Scaling Deep Reinforcement Learning

Michelangelo Vegliò, Marco Fantozzi, Antonio Di Cecco, Carlo Metta, Flora Angileri, Simone Treccani, Adrienne Chloe Rayos Macazar, Silvia Giulia Galfre', Maurizio Parton, Francesco Morandin

TL;DR: A self-scaling deep RL framework that automatically adapts network depth during training, enabling more efficient, sustainable, and interpretable DRL.

Abstract: Deep Reinforcement Learning (DRL) has achieved remarkable results in complex sequential decision-making tasks, often using very deep neural networks. However, these architectures incur substantial computational and energy costs, and selecting the optimal network depth in advance remains an open challenge. In this paper, we introduce SCALE-RL, a self-scaling DRL framework that dynamically adjusts its architectural depth during training, allowing the network to automatically adapt its depth to the task. Integrated into an AlphaZero-style pipeline for Othello, our approach matches the playing strength of the baseline agent while reducing network depth by 50\%. This process not only translates into substantial savings in computation and energy but also enhances model interpretability through the additive decomposition of decision-making across layers. Our results suggest that enabling DRL models to discover the complexity they require, rather than relying on fixed, over-parameterized architectures, makes it possible to develop more efficient, interpretable, and sustainable DRL agents.

Submission Number: 1249