arXiv reaDer
注意深いグループ化による改善された解釈可能なディープメトリック学習に向けて
Towards Improved and Interpretable Deep Metric Learning via Attentive Grouping
グループ化は、さまざまな機能を計算するための詳細なメトリック学習で一般的に使用されています。ただし、現在の方法は過剰適合する傾向があり、解釈可能性に欠けています。この作業では、任意のメトリック学習フレームワークと柔軟に統合される、改善された解釈可能なグループ化方法を提案します。私たちの方法は、各グループの学習可能なクエリを使用した注意メカニズムに基づいています。クエリは完全にトレーニング可能であり、多様性の喪失と組み合わせると、グループ固有の情報を取得できます。私たちの方法の魅力的な特性は、それ自体が自然に解釈可能になることです。学習可能なクエリと各空間位置の間の注意スコアは、その位置の重要性として解釈できます。提案されたグループ化方法が特徴の空間的順列に対して不変であることを正式に示します。畳み込みニューラルネットワークのモジュールとして使用する場合、私たちの方法は並進不変性につながります。私達は私達の方法を評価するために包括的な実験を行います。私たちの定量的な結果は、提案された方法が、さまざまなデータセット、評価メトリック、基本モデル、および損失関数にわたって一貫して大幅に以前の方法を上回っていることを示しています。私たちの知る限りでは初めて、私たちの解釈結果は、提案された方法がグループ間で明確で多様な特徴の学習を可能にすることを明確に示しています。
Grouping has been commonly used in deep metric learning for computing diverse features. However, current methods are prone to overfitting and lack interpretability. In this work, we propose an improved and interpretable grouping method to be integrated flexibly with any metric learning framework. Our method is based on the attention mechanism with a learnable query for each group. The query is fully trainable and can capture group-specific information when combined with the diversity loss. An appealing property of our method is that it naturally lends itself interpretability. The attention scores between the learnable query and each spatial position can be interpreted as the importance of that position. We formally show that our proposed grouping method is invariant to spatial permutations of features. When used as a module in convolutional neural networks, our method leads to translational invariance. We conduct comprehensive experiments to evaluate our method. Our quantitative results indicate that the proposed method outperforms prior methods consistently and significantly across different datasets, evaluation metrics, base models, and loss functions. For the first time to the best of our knowledge, our interpretation results clearly demonstrate that the proposed method enables the learning of distinct and diverse features across groups.
updated: Tue Nov 17 2020 19:08:24 GMT+0000 (UTC)
published: Tue Nov 17 2020 19:08: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アソシエイト