Predify: Augmenting deep neural networks with brain-inspired predictive coding dynamics - Archive ouverte HAL Accéder directement au contenu
Communication Dans Un Congrès Année : 2021

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

Résumé

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.
Fichier principal
Vignette du fichier
2106.02749.pdf (3.65 Mo) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03452646 , version 1 (27-11-2021)

Identifiants

Citer

Bhavin Choksi, Milad Mozafari, Callum Biggs O'May, Benjamin Ador, Andrea Alamia, et al.. Predify: Augmenting deep neural networks with brain-inspired predictive coding dynamics. 35th Conference on Neural Information Processing Systems (NeurIPS 2021), 2021, online, Canada. ⟨hal-03452646⟩
71 Consultations
23 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More