arXiv reaDer
プロトタイプベースのインクリメンタル少数ショットセマンティックセグメンテーション
Prototype-based Incremental Few-Shot Semantic Segmentation
セマンティックセグメンテーションモデルには、2つの基本的な弱点があります。i)コストのかかるピクセルレベルの注釈を含む大規模なトレーニングセットが必要であり、ii)トレーニングセットのクラスに制約された静的な出力スペースがあります。両方の問題に対処するために、新しいタスクであるインクリメンタル少数ショットセグメンテーション(iFSS)を導入します。 iFSSの目標は、いくつかの注釈付き画像からの新しいクラスを使用して、古いトレーニングデータにアクセスせずに、事前トレーニングされたセグメンテーションモデルを拡張することです。既存のモデルiniFSSの制限を克服するために、プロトタイプの学習と知識の蒸留を組み合わせたプロトタイプベースのインクリメンタル少数ショットセグメンテーション(PIFS)を提案します。 PIFSは、プロトタイプを利用して新しいクラスの分類子を初期化し、ネットワークを微調整してその機能表現を改良します。過剰適合と忘却を回避するために新旧両方のクラスのプロトタイプのスコアに基づいてプロトタイプベースの蒸留損失を設計し、非iidfewショットデータに対処するためにバッチ繰り込みを設計します。 iFSSの広範なベンチマークを作成し、PIFSがすべてのシナリオでいくつかの数ショットおよび増分学習方法よりも優れていることを示します。
Semantic segmentation models have two fundamental weaknesses: i) they require large training sets with costly pixel-level annotations, and ii) they have a static output space, constrained to the classes of the training set. Toward addressing both problems, we introduce a new task, Incremental Few-Shot Segmentation (iFSS). The goal of iFSS is to extend a pretrained segmentation model with new classes from few annotated images and without access to old training data. To overcome the limitations of existing models iniFSS, we propose Prototype-based Incremental Few-Shot Segmentation (PIFS) that couples prototype learning and knowledge distillation. PIFS exploits prototypes to initialize the classifiers of new classes, fine-tuning the network to refine its features representation. We design a prototype-based distillation loss on the scores of both old and new class prototypes to avoid overfitting and forgetting, and batch-renormalization to cope with non-i.i.d.few-shot data. We create an extensive benchmark for iFSS showing that PIFS outperforms several few-shot and incremental learning methods in all scenarios.
updated: Mon Oct 18 2021 16:51:31 GMT+0000 (UTC)
published: Mon Nov 30 2020 20:45:56 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アソシエイト