arXiv reaDer
少数ショットセグメンテーションのためのマイニング潜在クラス
Mining Latent Classes for Few-shot Segmentation
少数ショットセグメンテーション(FSS)は、注釈付きのサンプルがいくつかある場合に、見えないクラスをセグメント化することを目的としています。既存の方法には、機能が損なわれるという問題があります。つまり、潜在的な新規クラスは、トレーニングフェーズ中にバックグラウンドとして扱われます。私たちの方法は、この問題を軽減し、潜在的な小説クラスに埋め込まれている機能を強化することを目的としています。私たちの仕事では、新しい共同トレーニングフレームワークを提案します。サポートとクエリのペアに関する従来のエピソードトレーニングに基づいて、転送可能なサブクラスターを介して潜在的な新規クラスを活用するマイニングブランチを追加し、バックグラウンドとフォアグラウンドの両方のカテゴリで新しい修正手法を追加して、より安定したプロトタイプを実施します。さらに、転送可能なサブクラスターには、ラベルのない追加のデータを活用して機能をさらに強化する機能があります。 2つのFSSベンチマークでの広範な実験により、私たちの方法は、PASCAL-5iで3.7%mIOU、COCO-20iで7.0%mIOUの大幅なマージンで、以前の最先端技術を上回っていますが、パラメーターが74%少なく、2.5倍です。より速い推論速度。ソースコードはhttps://github.com/LiheYoung/MiningFSSで入手できます。
Few-shot segmentation (FSS) aims to segment unseen classes given only a few annotated samples. Existing methods suffer the problem of feature undermining, i.e. potential novel classes are treated as background during training phase. Our method aims to alleviate this problem and enhance the feature embedding on latent novel classes. In our work, we propose a novel joint-training framework. Based on conventional episodic training on support-query pairs, we add an additional mining branch that exploits latent novel classes via transferable sub-clusters, and a new rectification technique on both background and foreground categories to enforce more stable prototypes. Over and above that, our transferable sub-cluster has the ability to leverage extra unlabeled data for further feature enhancement. Extensive experiments on two FSS benchmarks demonstrate that our method outperforms previous state-of-the-art by a large margin of 3.7% mIOU on PASCAL-5i and 7.0% mIOU on COCO-20i at the cost of 74% fewer parameters and 2.5x faster inference speed. The source code is available at https://github.com/LiheYoung/MiningFSS.
updated: Fri Aug 06 2021 18:25:35 GMT+0000 (UTC)
published: Mon Mar 29 2021 07:59:10 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アソシエイト