arXiv:2501.11407cs.NEcs.AI2025-01被引 2

实现可通用的稀疏在线梯度突触可塑性,兼顾效率与实时性。

A Truly Sparse and General Implementation of Gradient-Based Synaptic Plasticity

  • 基于自定义自动微分流水线,实现稀疏在线梯度计算
  • 内存占用随网络规模增长,与序列长度无关,实测高效
  • 适用于任意神经元模型,特别适合e-prop等稀疏规则

基于梯度下降的在线突触可塑性规则在多种实际任务中表现优异,但其软件实现常需手工推导梯度或依赖反向传播,牺牲了在线能力。本文提出一种专用于稀疏在线梯度突触可塑性的自动微分(AD)流水线,支持任意神经元模型。该方法结合反向传播式前向AD的编程便利性与内存效率,通过利用在线可塑性的计算和内存优势,对对角张量采用简单的逐元素乘法替代昂贵的张量收缩。例如,资格传播(e-prop)规则恰好具备此性质,因此极大受益于该特性。我们在合成任务上验证了梯度与反向传播一致,且在语音分类基准测试中表现良好。结果表明,内存使用随网络规模增长,但不依赖序列长度,符合前向AD预期。

原文摘要 · Abstract (English)

Online synaptic plasticity rules derived from gradient descent achieve high accuracy on a wide range of practical tasks. However, their software implementation often requires tediously hand-derived gradients or using gradient backpropagation which sacrifices the online capability of the rules. In this work, we present a custom automatic differentiation (AD) pipeline for sparse and online implementation of gradient-based synaptic plasticity rules that generalizes to arbitrary neuron models. Our work combines the programming ease of backpropagation-type methods for forward AD while being memory-efficient. To achieve this, we exploit the advantageous compute and memory scaling of online synaptic plasticity by providing an inherently sparse implementation of AD where expensive tensor contractions are replaced with simple element-wise multiplications if the tensors are diagonal. Gradient-based synaptic plasticity rules such as eligibility propagation (e-prop) have exactly this property and thus profit immensely from this feature. We demonstrate the alignment of our gradients with respect to gradient backpropagation on an synthetic task where e-prop gradients are exact, as well as audio speech classification benchmarks. We demonstrate how memory utilization scales with network size without dependence on the sequence length, as expected from forward AD methods.

突触可塑性自动微分稀疏计算在线学习

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