arXiv reaDer
少数ショットセグメンテーションのためのアテンションプロトタイプ推論
Attentional Prototype Inference for Few-Shot Segmentation
この論文は、少数ショットのセグメンテーションに取り組むことを目的としています。既存のプロトタイプベースの手法はかなりの成功を収めていますが、限定されたラベル付きサンプルによって引き起こされる不確実性と曖昧さの問題に悩まされています。この研究では、少数ショットセグメンテーションのための確率的潜在変数フレームワークであるアテンションプロトタイプ推論 (API) を提案します。各オブジェクト カテゴリのプロトタイプを表すグローバル潜在変数を定義し、それを確率分布としてモデル化します。プロトタイプの確率的モデリングは、限られたデータとオブジェクトのクラス内変動によって引き起こされる固有の不確実性を処理することにより、モデルの汎化能力を強化します。モデルをさらに強化するために、各クエリ画像のアテンション マップを表すローカル潜在変数を導入します。これにより、モデルは背景を抑制しながら前景のオブジェクトに注意を向けることができます。提案されたモデルの最適化は、変分ベイジアン推論問題として定式化され、償却推論ネットワークによって確立されます。私たちは 4 つのベンチマークで広範な実験を実施しており、そこで私たちの提案は、最先端のプロトタイプベースの手法よりも少なくとも競争力があり、多くの場合、より優れたパフォーマンスを獲得しています。また、少数ショットセグメンテーションの方法の有効性についての洞察を得るために、包括的な分析とアブレーション研究も提供します。
This paper aims to address few-shot segmentation. While existing prototype-based methods have achieved considerable success, they suffer from uncertainty and ambiguity caused by limited labeled examples. In this work, we propose attentional prototype inference (API), a probabilistic latent variable framework for few-shot segmentation. We define a global latent variable to represent the prototype of each object category, which we model as a probabilistic distribution. The probabilistic modeling of the prototype enhances the model's generalization ability by handling the inherent uncertainty caused by limited data and intra-class variations of objects. To further enhance the model, we introduce a local latent variable to represent the attention map of each query image, which enables the model to attend to foreground objects while suppressing the background. The optimization of the proposed model is formulated as a variational Bayesian inference problem, which is established by amortized inference networks. We conduct extensive experiments on four benchmarks, where our proposal obtains at least competitive and often better performance than state-of-the-art prototype-based methods. We also provide comprehensive analyses and ablation studies to gain insight into the effectiveness of our method for few-shot segmentation.
updated: Mon May 29 2023 08:01:22 GMT+0000 (UTC)
published: Fri May 14 2021 06:58:44 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アソシエイト