arXiv reaDer
クラスごとのメモリバンクからのピクセルレベルの対照学習による半教師ありセマンティックセグメンテーション
Semi-Supervised Semantic Segmentation with Pixel-Level Contrastive Learning from a Class-wise Memory Bank
この作品は、半教師ありセマンティックセグメンテーションのための新しいアプローチを提示します。このアプローチの重要な要素は、セグメンテーションネットワークを強制して、データセット全体で同じクラスのサンプルに対して同様のピクセルレベルの特徴表現を生成する対照的な学習モジュールです。これを実現するために、ラベル付けされたデータからの関連性のある高品質の特徴ベクトルで継続的に更新されるメモリバンクを維持します。エンドツーエンドのトレーニングでは、ラベル付きデータとラベルなしデータの両方の機能が、メモリバンクの同じクラスのサンプルと同様になるように最適化されます。私たちのアプローチは、よく知られている公開ベンチマークでの半教師ありセマンティックセグメンテーションと半教師ありドメイン適応の現在の最先端を上回り、最も困難なシナリオ、つまり利用可能なラベル付きデータが少なくなることを大幅に改善します。 https://github.com/Shathe/SemiSeg-対照
This work presents a novel approach for semi-supervised semantic segmentation. The key element of this approach is our contrastive learning module that enforces the segmentation network to yield similar pixel-level feature representations for same-class samples across the whole dataset. To achieve this, we maintain a memory bank continuously updated with relevant and high-quality feature vectors from labeled data. In an end-to-end training, the features from both labeled and unlabeled data are optimized to be similar to same-class samples from the memory bank. Our approach outperforms the current state-of-the-art for semi-supervised semantic segmentation and semi-supervised domain adaptation on well-known public benchmarks, with larger improvements on the most challenging scenarios, i.e., less available labeled data. https://github.com/Shathe/SemiSeg-Contrastive
updated: Fri Aug 06 2021 08:07:18 GMT+0000 (UTC)
published: Tue Apr 27 2021 18:19:33 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アソシエイト