arXiv reaDer
対照学習によるマルチラベル画像分類
Multi-Label Image Classification with Contrastive Learning
最近、潜在表現を学習する効果的な方法として、対照学習がますます人気があり、さまざまな分野で成功しています。シングルラベル分類での制約的学習の成功は、この学習フレームワークを活用して、マルチラベル画像分類のパフォーマンスを向上させるための識別性を強化する動機になります。この論文では、対照学習の直接適用はマルチラベルの場合にはほとんど改善できないことを示します。したがって、完全に監視された設定で対照的な学習を伴うマルチラベル分類の新しいフレームワークを提案します。これは、異なるラベルのコンテキストで画像の複数の表現を学習します。これにより、対照学習をモデルにシンプルかつ直感的に適応させて、マルチラベル画像分類のパフォーマンスを向上させることができます。 2つのベンチマークデータセットでの広範な実験は、提案されたフレームワークがマルチラベル分類の高度な方法と比較して最先端のパフォーマンスを達成することを示しています。
Recently, as an effective way of learning latent representations, contrastive learning has been increasingly popular and successful in various domains. The success of constrastive learning in single-label classifications motivates us to leverage this learning framework to enhance distinctiveness for better performance in multi-label image classification. In this paper, we show that a direct application of contrastive learning can hardly improve in multi-label cases. Accordingly, we propose a novel framework for multi-label classification with contrastive learning in a fully supervised setting, which learns multiple representations of an image under the context of different labels. This facilities a simple yet intuitive adaption of contrastive learning into our model to boost its performance in multi-label image classification. Extensive experiments on two benchmark datasets show that the proposed framework achieves state-of-the-art performance in the comparison with the advanced methods in multi-label classification.
updated: Sat Jul 24 2021 15:00:47 GMT+0000 (UTC)
published: Sat Jul 24 2021 15:00:47 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アソシエイト