arXiv reaDer
少数ショットセグメンテーションのための高密度ガウス過程
Dense Gaussian Processes for Few-Shot Segmentation
少数ショット セグメンテーションは、挑戦的な高密度予測タスクであり、小さな注釈付きサポート セットのみが与えられた場合に、新しいクエリ イメージをセグメント化する必要があります。したがって、重要な問題は、サポート セットから詳細な情報を集約する方法を設計することですが、外観とコンテキストの大きな変動に対して堅牢です。この目的のために、密なガウス過程 (GP) 回帰に基づく少数ショット セグメンテーション法を提案します。サポート セットが与えられると、高密度 GP はローカルのディープ イメージ機能からマスク値へのマッピングを学習し、複雑な外観分布をキャプチャできます。さらに、不確実性を捉える原則的な手段を提供します。これは、CNN デコーダーによって取得される最終的なセグメンテーションのもう 1 つの強力な手がかりとして機能します。 1 次元のマスク出力の代わりに、アプローチのエンドツーエンドの学習機能をさらに活用して、GP の高次元出力空間を学習します。私たちのアプローチは、PASCAL-5^i および COCO-20^i ベンチマークで新しい最先端を設定し、COCO-20^i 5 ショット設定で +8.4 mIoU の絶対ゲインを達成します。さらに、堅牢なクロスデータセット転送を実現しながら、サポートセットのサイズを増やすと、アプローチのセグメンテーション品質が適切にスケーリングされます。コードとトレーニング済みモデルは、https://github.com/joakimjohnander/dgpnet で入手できます。
Few-shot segmentation is a challenging dense prediction task, which entails segmenting a novel query image given only a small annotated support set. The key problem is thus to design a method that aggregates detailed information from the support set, while being robust to large variations in appearance and context. To this end, we propose a few-shot segmentation method based on dense Gaussian process (GP) regression. Given the support set, our dense GP learns the mapping from local deep image features to mask values, capable of capturing complex appearance distributions. Furthermore, it provides a principled means of capturing uncertainty, which serves as another powerful cue for the final segmentation, obtained by a CNN decoder. Instead of a one-dimensional mask output, we further exploit the end-to-end learning capabilities of our approach to learn a high-dimensional output space for the GP. Our approach sets a new state-of-the-art on the PASCAL-5^i and COCO-20^i benchmarks, achieving an absolute gain of +8.4 mIoU in the COCO-20^i 5-shot setting. Furthermore, the segmentation quality of our approach scales gracefully when increasing the support set size, while achieving robust cross-dataset transfer. Code and trained models are available at https://github.com/joakimjohnander/dgpnet.
updated: Wed Aug 31 2022 08:55:16 GMT+0000 (UTC)
published: Thu Oct 07 2021 17:57:54 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アソシエイト