arXiv reaDer
最終的な完全に接続されたレイヤーの解釈
An interpretation of the final fully connected layer
近年、ニューラルネットワークはさまざまなタスクで最先端の精度を達成していますが、生成された出力の解釈は依然として困難です。この作業では、画像分類モデルの最終的な完全に接続されたレイヤーで学習された重みを理解する方法を提供しようとします。 RLのポリシー勾配目標と教師あり学習目標の間に関係を描くことによって、この方法を動機付けます。一般的に使用されるクロスエントロピーベースの教師あり学習目標は、ポリシー勾配目標の特殊なケースと見なすことができることをお勧めします。この洞察を使用して、画像の最も識別的で紛らわしい部分を見つける方法を提案します。私たちの方法は、ニューラルネットワークの構造について事前の仮定を行わず、計算コストが低くなっています。公開されている事前トレーニング済みモデルにこの方法を適用し、生成された結果を報告します。
In recent years neural networks have achieved state-of-the-art accuracy for various tasks but the the interpretation of the generated outputs still remains difficult. In this work we attempt to provide a method to understand the learnt weights in the final fully connected layer in image classification models. We motivate our method by drawing a connection between the policy gradient objective in RL and supervised learning objective. We suggest that the commonly used cross entropy based supervised learning objective can be regarded as a special case of the policy gradient objective. Using this insight we propose a method to find the most discriminative and confusing parts of an image. Our method does not make any prior assumption about neural network achitecture and has low computational cost. We apply our method on publicly available pre-trained models and report the generated results.
updated: Tue May 24 2022 09:05:19 GMT+0000 (UTC)
published: Tue May 24 2022 09:05:19 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アソシエイト