Backprop Alternative: Augmented Lagrangian Predictive Coding (pub.sakana.ai)
84 pointsby guld14 hours ago26 comments
lukeinator42 6 hours ago
There is a lot of interesting research into predictive coding as an alternative means to solve the credit assignment problem that might be a more plausible model of what happens in the brain.
I really liked this paper that showed using a predictive coding learning rule leads to the exact same gradients as backprop in arbitrary networks:
Predictive Coding Approximates Backprop Along Arbitrary Computation Graphs https://direct.mit.edu/neco/article/34/6/1329/110646/Predict...
txhwind 2 hours ago
Nice introduction to a simple but useful idea! The Lagrangian works like a time-smoothed optimizing direction state, but it can be placed on any wire, even at non-differentiable boundary! Can it be better than existing training methods for discrete components like argmax, MoE or VQ-VAE? Maybe networks can be composed by a lot of learnable discrete components, or even bits and gates finally.
txhwind 21 minutes ago
other questions:
- can it be used to relax timing order requirement in pipeline parallelism? Each node update lambda on communication, and optimize weights at other time. - given that BP is using SGD, can batches and T share the same timeline in optimization, while keeping the descent direction?
Jeff_Brown 4 hours ago
Could this relate to continual learning? It lets you update without pausing the entire system.
DoctorOetker 3 hours ago
Nothing prevents interleaving inference and gradient descent / RMAD / backpropagation from being interleaved.
Is a man a father or a son? It's a false dilemma, it can be both.
Is a training step pretraining or finetuning? It's the same mathematical operation, with the only difference the intention or reason of applying the training step.
verdverm 2 hours ago
LoRA throws a tiny amount of sand in these gears, but I generally agree there is near zero difference beyond the semantics we mere mortals assign to written tokens
red75prime 17 minutes ago
Maybe it has some interesting properties that could prevent catastrophic forgetting, but so far it looks like a biologically plausible approximation of backpropagation and it might be useful to decrease the training compute requirements.
rao-v 4 hours ago
I wonder if you could take a traditional backprop trained LLM and apply this approach to finetuning it (presumably needs less memory and compute?). It could be another entry in the spectrum between LORA and full fine tuning.
AIorNot 6 hours ago
Oh wow the theoretical implications in neuroscience exite me here - is this a potential model of Fristons Markov Blanket concept
“ Probably the most ambitious and all-encompassing version of the ‘Bayesian turn’ in cognitive science is the free energy principle (FEP). The FEP is a mathematical framework, developed by Karl Friston and colleagues (Friston, Kilner, and Harrison 2006; Friston et al. 2010; Friston 2010; Friston et al. 2017a; Friston 2019), which specifies an objective function that any self-organizing system needs to minimize in order to ensure adaptive exchanges with its environment. One major appeal of the FEP is that it aims for (and seems to deliver) an unprecedented integration of the life sciences (including psychology, neuroscience, and theoretical biology). The difference between the FEP and earlier inferential theories (e.g., Gregory 1980, Grossberg 1980, Rao and Ballard 1999, Lee and Mumford 2003) is that not only perceptual processes, but also other cognitive functions such as learning, attention, and action planning can be subsumed under one single principle: the minimization of free energy through the process of active inference (Friston 2010; Friston et al. 2017). ”
nullbio 3 hours ago
Isn't the FEP basically just loss minimization over KL-divergence? In other words, it's the same thing we already do with ML and already have been doing for years? I've never understood where this differs to the status quo, or why this isn't just a relabelling of techniques/concepts. Although I didn't look too deeply.
AIorNot 2 hours ago
Yeah you are right to basics of the paper and I am extrapolating a bit here:
I think the remarkable result of this paper is that they add a local Lagrange multiplier λ at each layer, which accumulates constraint/prediction error over the inference dynamics.
At equilibrium, in the linear case, those local multipliers converge to exactly the same gradient signal that backpropagation would calculate globally
Now what is Predictive coding: its a network that can minimize prediction errors through local recurrent interactions instead of an explicit global backward pass.
Now I am making the leap to Fristons more philosophical and mathematical work not the paper - so that is me making the allusion
But a light bulb moment for me dawned when I read it
This process (PC-ALM) gives us a concrete example of how globally coherent inference/credit assignment can emerge from purely local dynamical interactions.
PC-ALM lets a recurrent dynamical system relax toward a state in which the backprop gradient is represented locally throughout the network.
That distinction is potentially important for neuroscience.
A brain doesn’t obviously have a central routine saying:
loss.backward()
it certainly has recurrent neural populations whose states continuously influence neighboring populations.
This paper is demonstrating that, at least mathematically, those sorts of local recurrent dynamics can generate the same credit information that backprop obtains through the chain rule. The authors explicitly motivate predictive coding as a biologically plausible local-learning alternative because standard BP requires globally coordinated error variables and update ordering.
Think about it also give plausible evolutionary to chain intelligence through cells coming together and creating nested networks
This has got to be how the neurological intelligence sausage gets made
What it eventually means for ML I’m Not sure but hopeful it opens a door
nullbio 2 hours ago
Thanks for the run down. I definitely see the appeal in pursuing local error correction mechanisms. I can see this opening doors too, or some extension of this. Certainly feels like the right way forward.
txhwind 2 hours ago
Backprop can also be viewed L local signal passing, but it requires global timing coordination. The Lagrangian enables out-of-order update.
guld 14 hours ago
New paper by Sakana.ai [1]
cs702 10 hours ago
~85% accuracy on MNIST. Sigh.
How does it do on CIFAR-10, or even better, ImageNet?
Interesting research, not sure it's a backprop alternative.
===
EDIT: accuracy on MNIST is not ~90%. It's ~85%.
bz_bz_bz 10 hours ago
Their image classification benchmarks include both:
https://pub.sakana.ai/pc-alm/assets/figures/benchmark_accura...
cs702 8 hours ago
~74% on CIFAR-10. Still a far cry from backprop.
I didn't see ImageNet. TinyImageNet is something else.
Lerc 10 hours ago
It might be beneficial while not being optimal on its own.
The obvious example is if it has different behaviour around local minima, it could be an altenate pathway out.
I have often wondered if doing training with radically different aproaches for the first few iterarions would avoid any method specific artifacts before the weights had time to denoise.
dnautics 5 hours ago
You can probably distribute training more easily too
qarl 8 hours ago
They state replacing backprop is not their goal.
Their goal is to understand how distributed systems which cannot do backprop (the brain) can still do learning.
im3w1l 5 hours ago
Personally I think the dirty secret of the brain is that a lot of things are hard coded. And many things that we need to learn are also hard coded except that some parameters need to be tuned.
If we puke, the brain will not do general aversive learning, it will learn to avoid specifically the last thing eaten, because it instinctively knows about food poisoning.
Imprinting is absolutely fascinating. Some newborn animals will run a very simple pattern detector like looking for a red dot or something and use that to bootstrap their conception of their parent.
For fully general learning I have a hunch that it can be done using local history plus a semi-global reward scalar (global neurotransmittor levels).
DoctorOetker 2 hours ago
regardless if the intelligence in the brain is hardcoded or not, to the extent it is, this information must have been compressed in the genome, which runs counter to almost all observations: a child doesn't remember the experience of their ancestors, for example. The only sense in which we do carry mental state without relearning is emotions, instincts, reflexes (some neuronal pathways that connect the eye to the middle ear), hormonal driven behavior (fear adrenalin).
For another, there are about 200k promotor regions (including non-coding) in the human genome.
A promotor region might have say 6 to 15 bits of information.
Can you compress 2025 or even 2024 era LLM intelligence into 3 megabit = ~400 kB ? I think not. I think a lot of compression is still possible, but 400 kB?
So I think we can box up the idea of "dirty secrets of the braing: not learning but hard coding". There is a lot of hard coding in biology, but brains are evolved specifically to enable learning within the individual lifetime instead of only learning by natural selection.
I also don't buy the following argument:
> If we puke, the brain will not do general aversive learning, it will learn to avoid specifically the last thing eaten, because it instinctively knows about food poisoning.
Each time it happens that I end up puking, I do feel aversion and try to avoid puking at all, sometimes I succeed but sometimes is just puke. There must be fundamental puke reflexes (which one fails to avoid) and avertable puke reflexes.
verdverm 2 hours ago
> Can you compress 2025 or even 2024 era LLM intelligence into 3 megabit = ~400 kB ? I think not. I think a lot of compression is still possible, but 400 kB?
There are a few extra levels of interpretation (like protein synthesis) that are more like a transpiler than compression (imo), over a 4-base language that is read in a sliding window and is affected by surrounding conditions, so the same "token" sequence may produce different things depending on external factors. Some biologists I used to collaborate with talked about 7 layers to this process, I have only described one level here
DoctorOetker 10 minutes ago
From an information theory perspective it does not matter how many levels of interpretation are in between, that hardcoded information must pass the genome. for example we can discuss "what about protein synthesis", for example binding affinities, folding helpers etc. they in turn were encoded genetically as well.
> There are a few extra levels of interpretation (like protein synthesis) that are more like a transpiler than compression (imo), over a 4-base language that is read in a sliding window and is affected by surrounding conditions, so the same "token" sequence may produce different things depending on external factors. Some biologists I used to collaborate with talked about 7 layers to this process, I have only described one level here
so the same "token" sequence may produce different things depending on external factors.
yes, non-hereditary learning depends on external factors, thank you for paraphrasing me while shifting attention.
the multi-scale nature (transpilers etc.) doesn't change the theorems in probability and information theory which seriously constrain the maximum amount of information a message can store.
The whole point of a brain is that it is an organ dedicated to storing, retrieving and timely utilising information one can't afford to store in a genome.
imtringued 11 minutes ago
It's really sad that they are only a few years away from making backpropagation completely obsolete.
mikelitoris 5 hours ago
Rolls right off the tongue