arXiv reaDer
マルチラベル画像分類のためのグラフアテンショントランスネットワーク
Graph Attention Transformer Network for Multi-Label Image Classification
マルチラベル分類は、画像から複数のオブジェクトまたは属性を認識することを目的としています。ただし、適切なラベルグラフから学習して、このようなラベル間の相関関係や依存関係を効果的に特徴付けることは困難です。現在の方法では、隣接行列としてトレーニングセットに基づくラベルの共起確率を使用して、この相関をモデル化することがよくあります。これは、データセットによって大幅に制限され、モデルの一般化能力に影響を与えます。この論文では、複雑なラベル間関係を効果的にマイニングできるマルチラベル画像分類の一般的なフレームワークであるグラフアテンショントランスフォーマーネットワーク(GATN)を提案します。まず、豊富な意味情報を表すことができる初期相関行列として、ラベル単語の埋め込みに基づくコサイン類似度を使用します。続いて、現在のドメインに適応するようにこの隣接行列を転送するグラフアテンショントランスフォーマーレイヤーを設計します。私たちの広範な実験は、提案された方法が3つのデータセットで最先端のパフォーマンスを達成できることを示しています。
Multi-label classification aims to recognize multiple objects or attributes from images. However, it is challenging to learn from proper label graphs to effectively characterize such inter-label correlations or dependencies. Current methods often use the co-occurrence probability of labels based on the training set as the adjacency matrix to model this correlation, which is greatly limited by the dataset and affects the model's generalization ability. In this paper, we propose a Graph Attention Transformer Network (GATN), a general framework for multi-label image classification that can effectively mine complex inter-label relationships. First, we use the cosine similarity based on the label word embedding as the initial correlation matrix, which can represent rich semantic information. Subsequently, we design the graph attention transformer layer to transfer this adjacency matrix to adapt to the current domain. Our extensive experiments have demonstrated that our proposed methods can achieve state-of-the-art performance on three datasets.
updated: Mon Jan 15 2024 10:44:57 GMT+0000 (UTC)
published: Tue Mar 08 2022 12:39:05 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アソシエイト