Diffusing Blame: Dale-Constrained Learning Without Weight Transport
Sakana AI's 'Diffusing Blame' algorithm trains neural networks without weight transport, moving closer to how biological brains function.
Decoder
- Dale's Principle: The biological hypothesis that a single neuron releases the same type of neurotransmitter at all its synaptic terminals, making the neuron either excitatory or inhibitory.
- Weight Transport Problem: The requirement in standard backpropagation that the backward pass must use the exact transpose of the forward weight matrix, which is considered implausible in biological neural systems.
- PPO: Proximal Policy Optimization, a reinforcement learning algorithm used to train agents by balancing exploration and exploitation.
Original article
Introducing "Diffusing Blame": can a neural network learn competitively while strictly obeying Dale's principle, the rule that real neurons follow? We show it can, across both image classification and reinforcement learning.
Real neurons generally follow Dale’s principle: each neuron is predominantly excitatory or inhibitory. Standard artificial networks usually ignore this constraint, allowing every unit to mix positive and negative outgoing weights. Backprop makes the gap even wider. Its backward pass needs exact transposed copies of the forward weights, the so-called "weight transport problem,” which biology doesn’t seem to have a mechanism for.
So we asked: can a network that strictly enforces Dale's principle still learn well, without weight transport? Our approach builds on Error Diffusion (ED), a local rule that routes a single global error signal directly to every hidden unit, where each layer is split into separate excitatory and inhibitory streams with four non-negative weight matrices, so a synapse's sign comes from fixed population identity rather than a learnable weight.
Our main contribution is to extend ED from binary to multi-class problems via modulo error routing. We then asked whether this routing mechanism could provide useful credit signals in the noisy setting of reinforcement learning. During PPO training on Ant, Humanoid, and HalfCheetah, we compared each local ED update with the corresponding true backpropagation gradient. Among the routing schemes we tested, modulo routing consistently produced the strongest alignment.
Taken together, these results show that Dale-constrained networks can still learn without transporting weights backward, suggesting a potential path toward learning rules that are both effective and more biologically plausible.
Fresh dev news, AI-digested daily. Never miss a trend at devoured.fyi