用多巴胺机制设计新优化器,让神经网络更高效地学习。
Dopamine-driven synaptic credit assignment in neural networks
- 基于多巴胺的奖励预测误差,设计无梯度更新的优化方法。
- 在异或任务和混沌时间序列上收敛更快,性能接近梯度法。
- 计算量和内存占用显著降低,更符合生物神经机制。
解决突触信用分配问题(CAP)是生物与人工神经系统的共同核心挑战。理想的突触权重设置需准确分配各神经元对最终输出的影响。当前人工网络依赖梯度法中的反向传播,但存在代价高、需权重传输与更新锁定等问题。为此,本文提出一种受神经强化学习启发的无梯度优化器Dopamine,其基于权重扰动(WP)学习,通过最小化扰动模型预期结果与真实结果间的奖励预测误差(RPE)来调整学习率,类比大脑中多巴胺的作用。实验显示,Dopamine在多层感知机解决异或任务、递归网络预测混沌时间序列中均实现加速收敛,性能优于标准权重扰动法,媲美梯度方法,同时大幅减少计算与内存开销。该方法不仅获得稳健解,且在性能与生物合理性上均优于现有主流优化器。
原文摘要 · Abstract (English)
Solving the synaptic Credit Assignment Problem(CAP) is central to learning in both biological and artificial neural systems. Finding an optimal solution for synaptic CAP means setting the synaptic weights that assign credit to each neuron for influencing the final output and behavior of neural networks or animals. Gradient-based methods solve this problem in artificial neural networks using back-propagation, however, not in the most efficient way. For instance, back-propagation requires a chain of top-down gradient computations. This leads to an expensive optimization process in terms of computing power and memory linked with well-known weight transport and update locking problems. To address these shortcomings, we take a NeuroAI approach and draw inspiration from neural Reinforcement Learning to develop a derivative-free optimizer for training neural networks, Dopamine. Dopamine is developed for Weight Perturbation (WP) learning that exploits stochastic updating of weights towards optima. It achieves this by minimizing the regret, a form of Reward Prediction Error (RPE) between the expected outcome from the perturbed model and the actual outcome from the unperturbed model. We use this RPE to adjust the learning rate in the network (i.e., creating an adaptive learning rate strategy, similar to the role of dopamine in the brain). We tested the Dopamine optimizer for training multi-layered perceptrons for XOR tasks, and recurrent neural networks for chaotic time series forecasting. Dopamine-trained models demonstrate accelerated convergence and outperform standard WP, and give comparable performance to gradient-based algorithms, while consuming significantly less computation and memory. Overall, the Dopamine optimizer not only finds robust solutions and comparable performance to the state-of-the-art Machine Learning optimizers but is also neurobiologically more plausible.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。