arXiv reaDer
Saccader:視覚に対するハードアテンションモデルの精度の向上
Saccader: Improving Accuracy of Hard Attention Models for Vision
 深い畳み込みニューラルネットワークは、ほぼすべての画像分類タスクにわたって最先端のパフォーマンスを実現しますが、その決定を解釈することは困難です。設計によってある程度の解釈可能性を提供する1つのアプローチは、画像の関連部分のみを使用するハードアテンションです。ただし、クラスラベルの監督のみを使用してハードアテンションモデルをトレーニングすることは難しく、複雑なデータセットに拡張することは難しいことがわかっています。ここでは、Saccaderと呼ばれる新しいハードアテンションモデルを提案します。 Saccaderの鍵は、クラスラベルのみを必要とする事前トレーニングステップであり、ポリシーグラデーションの最適化のための初期の注意場所を提供します。私たちの最高のモデルは、一般的なImageNetベースラインとのギャップを狭めて、画像の3分の1未満に対応しながら75%のトップ1と91%のトップ5を達成します。
Although deep convolutional neural networks achieve state-of-the-art performance across nearly all image classification tasks, their decisions are difficult to interpret. One approach that offers some level of interpretability by design is hard attention, which uses only relevant portions of the image. However, training hard attention models with only class label supervision is challenging, and hard attention has proved difficult to scale to complex datasets. Here, we propose a novel hard attention model, which we term Saccader. Key to Saccader is a pretraining step that requires only class labels and provides initial attention locations for policy gradient optimization. Our best models narrow the gap to common ImageNet baselines, achieving 75% top-1 and 91% top-5 while attending to less than one-third of the image.
updated: Sat Dec 07 2019 00:34:57 GMT+0000 (UTC)
published: Tue Aug 20 2019 23:40:21 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アソシエイト