Predify: Augmenting deep neural networks with brain-inspired predictive coding dynamics

06/04/2021
by   Bhavin Choksi, et al.
0

Deep neural networks excel at image classification, but their performance is far less robust to input perturbations than human perception. In this work we explore whether this shortcoming may be partly addressed by incorporating brain-inspired recurrent dynamics in deep convolutional networks. We take inspiration from a popular framework in neuroscience: 'predictive coding'. At each layer of the hierarchical model, generative feedback 'predicts' (i.e., reconstructs) the pattern of activity in the previous layer. The reconstruction errors are used to iteratively update the network's representations across timesteps, and to optimize the network's feedback weights over the natural image dataset-a form of unsupervised training. We show that implementing this strategy into two popular networks, VGG16 and EfficientNetB0, improves their robustness against various corruptions. We hypothesize that other feedforward networks could similarly benefit from the proposed framework. To promote research in this direction, we provide an open-sourced PyTorch-based package called Predify, which can be used to implement and investigate the impacts of the predictive coding dynamics in any convolutional neural network.

READ FULL TEXT

page 6

page 13

research
02/03/2021

Predictive coding feedback results in perceived illusory contours in a recurrent neural network

Modern feedforward convolutional neural networks (CNNs) can now solve so...
research
06/08/2021

On the role of feedback in visual processing: a predictive coding perspective

Brain-inspired machine learning is gaining increasing consideration, par...
research
07/17/2020

Neural Networks with Recurrent Generative Feedback

Neural networks are vulnerable to input perturbations such as additive n...
research
11/22/2022

Convolutional Neural Generative Coding: Scaling Predictive Coding to Natural Images

In this work, we develop convolutional neural generative coding (Conv-NG...
research
04/12/2023

Mathematical derivation of wave propagation properties in hierarchical neural networks with predictive coding feedback dynamics

Sensory perception (e.g. vision) relies on a hierarchy of cortical areas...
research
05/29/2023

Deep Predictive Coding with Bi-directional Propagation for Classification and Reconstruction

This paper presents a new learning algorithm, termed Deep Bi-directional...
research
09/05/2018

Hierarchical Selective Recruitment in Linear-Threshold Brain Networks - Part I: Intra-Layer Dynamics and Selective Inhibition

Goal-driven selective attention (GDSA) refers to the brain's function of...

Please sign up or login with your details

Forgot password? Click here to reset