arXiv reaDer
Contrastive Transformer: Transformer の生来のパッチを使用した対照的な学習スキーム
Contrastive Transformer: Contrastive Learning Scheme with Transformer innate Patches
このホワイト ペーパーでは、Transformer の生来のパッチを使用した対照的な学習スキームである Contrastive Transformer について説明します。 Contrastive Transformer は、画像分類によく使用される既存の対比学習手法を有効にして、セマンティック セグメンテーションなどの高密度のダウンストリーム予測タスクに役立ちます。このスキームは、教師付きパッチレベルの対照学習を実行し、グラウンド トゥルース マスクに基づいてパッチを選択します。その後、ハード ネガティブおよびハード ポジティブ サンプリングに使用されます。この方式は、すべてのビジョン トランスフォーマー アーキテクチャに適用され、実装が容易で、最小限の追加メモリ フットプリントが導入されます。さらに、このスキームでは、各パッチがイメージとして扱われるため、巨大なバッチ サイズが不要になります。低解像度データ、大規模なクラスの不均衡、および同様のセマンティック クラスで知られる航空画像セグメンテーションの場合に、Contrastive Transformer を適用してテストします。 ISPRS ポツダム航空画像セグメンテーション データセットに対するコントラスト トランスフォーマー スキームの有効性を示すために、広範な実験を行います。さらに、複数の本質的に異なる Transformer アーキテクチャに適用することにより、スキームの一般化可能性を示します。最終的に、結果は、すべてのクラスで平均 IoU が一貫して増加していることを示しています。
This paper presents Contrastive Transformer, a contrastive learning scheme using the Transformer innate patches. Contrastive Transformer enables existing contrastive learning techniques, often used for image classification, to benefit dense downstream prediction tasks such as semantic segmentation. The scheme performs supervised patch-level contrastive learning, selecting the patches based on the ground truth mask, subsequently used for hard-negative and hard-positive sampling. The scheme applies to all vision-transformer architectures, is easy to implement, and introduces minimal additional memory footprint. Additionally, the scheme removes the need for huge batch sizes, as each patch is treated as an image. We apply and test Contrastive Transformer for the case of aerial image segmentation, known for low-resolution data, large class imbalance, and similar semantic classes. We perform extensive experiments to show the efficacy of the Contrastive Transformer scheme on the ISPRS Potsdam aerial image segmentation dataset. Additionally, we show the generalizability of our scheme by applying it to multiple inherently different Transformer architectures. Ultimately, the results show a consistent increase in mean IoU across all classes.
updated: Sun Mar 26 2023 20:19:28 GMT+0000 (UTC)
published: Sun Mar 26 2023 20:19:28 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アソシエイト