arXiv reaDer
Predify:脳に触発された予測コーディングダイナミクスでディープニューラルネットワークを強化する
Predify: Augmenting deep neural networks with brain-inspired predictive coding dynamics
ディープニューラルネットワークは画像分類に優れていますが、そのパフォーマンスは人間の知覚よりも入力摂動に対してはるかに堅牢ではありません。この作業では、深い畳み込みネットワークに脳に触発された反復ダイナミクスを組み込むことによって、この欠点に部分的に対処できるかどうかを調査します。神経科学で人気のあるフレームワークである「予測コーディング」からインスピレーションを得ています。階層モデルの各レイヤーで、生成フィードバックは前のレイヤーのアクティビティのパターンを「予測」(つまり、再構築)します。再構成エラーは、タイムステップ全体でネットワークの表現を繰り返し更新し、教師なしトレーニングの形式である自然画像データセットに対するネットワークのフィードバックの重みを最適化するために使用されます。この戦略を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 and adversarial attacks. 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.
updated: Thu Nov 04 2021 09:40:27 GMT+0000 (UTC)
published: Fri Jun 04 2021 22:48:13 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト