Error Highways: Scaling Predictive Coding to Very Deep Networks

· Source: Takara TLDR - Daily AI Papers · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, medium

Summary

Error Highways: Scaling Predictive Coding to Very Deep Networks introduces Highway Error Propagation (HEP), a novel scheme designed to overcome the depth limitations of predictive coding networks (PCNs). PCNs offer a biologically-plausible, local-learning alternative to back-propagation but have been largely restricted to shallow architectures due to rapid decay of the learning signal in deeper layers. HEP addresses this by augmenting the PC free energy function with feedback matrices V_{L\to i}, which directly couple selected hidden states to the clamped output error. This linear coupling ensures that correction signals maintain a magnitude independent of network depth, effectively bypassing the Jacobian chain while preserving the local PC synaptic update rule. The method successfully trains Multi-Layer Perceptrons (MLPs) of up to 128 layers on MNIST and Fashion-MNIST datasets, demonstrating robust accuracy across varying depths.

Key takeaway

For Machine Learning Engineers exploring biologically-plausible learning, Highway Error Propagation (HEP) offers a viable path to scale predictive coding networks beyond shallow architectures. You can now train deep MLPs, up to 128 layers, with PCNs on tasks like image classification, achieving robust accuracy. This method provides a local-learning alternative to back-propagation, potentially simplifying model development and reducing computational overhead for certain deep learning applications.

Key insights

Highway Error Propagation enables deep predictive coding networks by ensuring consistent error signal propagation regardless of depth.

Principles

Method

Augment PC's free energy with feedback matrices V_{L\to i} to linearly couple hidden states to output error, bypassing the Jacobian chain for depth-independent correction.

In practice

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Takara TLDR - Daily AI Papers.