arXiv:2502.17496cs.LGcs.AI2025-02被引 2

提出高效可扩展的脉冲强化学习框架,显著提升能效与训练速度。

SpikeRL: A Scalable and Energy-efficient Framework for Deep Spiking Reinforcement Learning

  • 自研深度强化学习-脉冲神经网络组件,支持分布式训练
  • 相比当前最优方法提速4.26倍,能效提升2.25倍
  • 适合需要高能效与可扩展性的真实场景连续控制任务

在人工智能革命时代,大规模数据驱动系统依赖高性能计算,消耗巨大能源。为应对可持续性挑战,脉冲神经网络(SNNs)因其固有的能效优势成为类脑计算的重要方向。然而,在复杂连续控制任务中,SNNs仍缺乏传统人工神经网络的性能优化。为此,本文扩展了先前的SpikeRL框架,构建了一种面向连续控制的可扩展、低功耗深度强化学习-脉冲神经网络系统。新版本采用自研深度强化学习-SNN组件,结合基于人口编码的结构,并使用PyTorch Distributed + NCCL实现分布式训练,同时引入混合精度参数更新。实验表明,该框架比当前最先进的DeepRL-SNN方法快4.26倍,能效提升2.25倍,展现出在真实应用场景中复杂连续控制任务下的真正可扩展与可持续能力。

原文摘要 · Abstract (English)

In this era of AI revolution, massive investments in large-scale data-driven AI systems demand high-performance computing, consuming tremendous energy and resources. This trend raises new challenges in optimizing sustainability without sacrificing scalability or performance. Among the energy-efficient alternatives of the traditional Von Neumann architecture, neuromorphic computing and its Spiking Neural Networks (SNNs) are a promising choice due to their inherent energy efficiency. However, in some real-world application scenarios such as complex continuous control tasks, SNNs often lack the performance optimizations that traditional artificial neural networks have. Researchers have addressed this by combining SNNs with Deep Reinforcement Learning (DeepRL), yet scalability remains unexplored. In this paper, we extend our previous work on SpikeRL, which is a scalable and energy efficient framework for DeepRL-based SNNs for continuous control. In our initial implementation of SpikeRL framework, we depended on the population encoding from the Population-coded Spiking Actor Network (PopSAN) method for our SNN model and implemented distributed training with Message Passing Interface (MPI) through mpi4py. Also, further optimizing our model training by using mixed-precision for parameter updates. In our new SpikeRL framework, we have implemented our own DeepRL-SNN component with population encoding, and distributed training with PyTorch Distributed package with NCCL backend while still optimizing with mixed precision training. Our new SpikeRL implementation is 4.26X faster and 2.25X more energy efficient than state-of-the-art DeepRL-SNN methods. Our proposed SpikeRL framework demonstrates a truly scalable and sustainable solution for complex continuous control tasks in real-world applications.

脉冲神经网络强化学习能效优化分布式训练

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