ディープ ニューラル ネットワークは画像分類に優れていますが、そのパフォーマンスは人間の知覚よりも入力の摂動に対してはるかにロバストではありません。この作業では、脳に触発された再帰ダイナミクスを深い畳み込みネットワークに組み込むことで、この欠点に部分的に対処できるかどうかを調べます。私たちは、神経科学の一般的なフレームワークである「予測コーディング」からインスピレーションを得ています。階層モデルの各層では、生成フィードバックが前の層の活動パターンを「予測」 (つまり、再構築) します。再構成エラーは、時間ステップ全体でネットワークの表現を繰り返し更新し、教師なしトレーニングの形式である自然画像データセットに対してネットワークのフィードバック重みを最適化するために使用されます。この戦略を 2 つの一般的なネットワーク、VGG16 と EfficientNetB0 に実装すると、さまざまな破損に対する堅牢性が向上することを示します。提案されたフレームワークから他のフィードフォワード ネットワークが同様に利益を得る可能性があると仮定します。この方向の研究を促進するために、Predify と呼ばれるオープン ソースの PyTorch ベースのパッケージを提供します。このパッケージを使用して、任意の畳み込みニューラル ネットワークにおける予測コーディング ダイナミクスの影響を実装および調査できます。
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.