arXiv reaDer
深い視覚表現学習のための変分構造化注意ネットワーク
Variational Structured Attention Networks for Deep Visual Representation Learning
畳み込みニューラルネットワークは、視覚表現学習の強力な機能の恩恵を受けて、セマンティックセグメンテーション、深度推定、表面法線予測などのピクセルレベルの予測タスクに対処する上で大きな進歩を可能にしました。通常、最先端のモデルは、改善された深い特徴表現のために注意メカニズムを統合します。最近、いくつかの作品は、深い特徴の洗練のために空間的注意とチャネル的注意の両方を学び、組み合わせることが重要であることを示しています。この論文では、以前のアプローチを効果的に後押しすることを目的とし、結果として生じる注意テンソルを構造化し、これら2つのタイプの注意間の相互作用をモデル化するために、原理的な方法で空間注意マップとチャネル注意ベクトルの両方を共同で学習する統合された深いフレームワークを提案します。具体的には、確率的表現学習フレームワーク内で注意の推定と相互作用を統合し、Variational STructured Attentionネットワーク(VISTA-Net)を導きます。ニューラルネットワーク内に推論規則を実装することで、確率的およびCNNフロントエンドパラメーターのエンドツーエンドの学習が可能になります。高密度の視覚的予測のための6つの大規模データセットに関する広範な経験的評価によって示されるように、VISTA-Netは、複数の連続的および離散的予測タスクで最先端を上回り、共同構造化空間における提案されたアプローチの利点を確認します。 -深い表現学習のためのチャネル注意推定。コードはhttps://github.com/ygjwd12345/VISTA-Netで入手できます。
Convolutional neural networks have enabled major progresses in addressing pixel-level prediction tasks such as semantic segmentation, depth estimation, surface normal prediction and so on, benefiting from their powerful capabilities in visual representation learning. Typically, state of the art models integrate attention mechanisms for improved deep feature representations. Recently, some works have demonstrated the significance of learning and combining both spatial- and channelwise attentions for deep feature refinement. In this paper, weaim at effectively boosting previous approaches and propose a unified deep framework to jointly learn both spatial attention maps and channel attention vectors in a principled manner so as to structure the resulting attention tensors and model interactions between these two types of attentions. Specifically, we integrate the estimation and the interaction of the attentions within a probabilistic representation learning framework, leading to VarIational STructured Attention networks (VISTA-Net). We implement the inference rules within the neural network, thus allowing for end-to-end learning of the probabilistic and the CNN frontend parameters. As demonstrated by our extensive empirical evaluation on six large-scale datasets for dense visual prediction, VISTA-Net outperforms the state-of-the-art in multiple continuous and discrete prediction tasks, thus confirming the benefit of the proposed approach in joint structured spatial-channel attention estimation for deep representation learning. The code is available at https://github.com/ygjwd12345/VISTA-Net.
updated: Wed Dec 15 2021 05:42:23 GMT+0000 (UTC)
published: Fri Mar 05 2021 07:37:24 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アソシエイト